func (s *ReadAllStateDirsSuite) TestReadAllStateDirs(c *C) {
basedir := c.MkDir()
relsdir := setUpDir(c, basedir, "relations", map[string]string{
"ignored": "blah",
"foo-bar-123": "gibberish",
})
setUpDir(c, relsdir, "123", map[string]string{
"foo-0": "change-version: 1\n",
"foo-1": "change-version: 2\nchanged-pending: true\n",
"gibberish": "gibberish",
})
setUpDir(c, relsdir, "456", map[string]string{
"bar-0": "change-version: 3\n",
"bar-1": "change-version: 4\n",
})
setUpDir(c, relsdir, "789", nil)
setUpDir(c, relsdir, "onethousand", map[string]string{
"baz-0": "change-version: 3\n",
"baz-1": "change-version: 4\n",
})
dirs, err := relation.ReadAllStateDirs(relsdir)
c.Assert(err, IsNil)
for id, dir := range dirs {
c.Logf("%d: %#v", id, dir)
}
assertState(c, dirs[123], relsdir, 123, msi{"foo/0": 1, "foo/1": 2}, "foo/1", false)
assertState(c, dirs[456], relsdir, 456, msi{"bar/0": 3, "bar/1": 4}, "", false)
assertState(c, dirs[789], relsdir, 789, msi{}, "", false)
c.Assert(dirs, HasLen, 3)
}
// restoreRelations reconciles the supplied relation state dirs with the
// remote state of the corresponding relations.
func (u *Uniter) restoreRelations() error {
dirs, err := relation.ReadAllStateDirs(u.relationsDir)
if err != nil {
return err
}
for id, dir := range dirs {
remove := false
rel, err := u.st.Relation(id)
if state.IsNotFound(err) {
remove = true
} else if err != nil {
return err
}
if err = u.addRelation(rel, dir); err == state.ErrCannotEnterScope {
remove = true
} else if err != nil {
return err
}
if remove {
// If the previous execution was interrupted in the process of
// joining or departing the relation, the directory will be empty
// and the state is sane.
if err := dir.Remove(); err != nil {
return fmt.Errorf("cannot synchronize relation state: %v", err)
}
}
}
return nil
}
func (s *ReadAllStateDirsSuite) TestBadStateDir(c *C) {
basedir := c.MkDir()
relsdir := setUpDir(c, basedir, "relations", nil)
setUpDir(c, relsdir, "123", map[string]string{
"bad-0": "blah: blah\n",
})
_, err := relation.ReadAllStateDirs(relsdir)
c.Assert(err, ErrorMatches, `cannot load relations state from .*: cannot load relation state from .*: invalid unit file "bad-0": "changed-version" not set`)
}
func (s *ReadAllStateDirsSuite) TestNoDir(c *C) {
basedir := c.MkDir()
relsdir := filepath.Join(basedir, "relations")
dirs, err := relation.ReadAllStateDirs(relsdir)
c.Assert(err, IsNil)
c.Assert(dirs, HasLen, 0)
_, err = os.Stat(relsdir)
c.Assert(os.IsNotExist(err), Equals, true)
}