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) }