func (s *StateFileSuite) TestStates(c *gc.C) {
for i, t := range stateTests {
c.Logf("test %d", i)
path := filepath.Join(c.MkDir(), "uniter")
file := uniter.NewStateFile(path)
_, err := file.Read()
c.Assert(err, gc.Equals, uniter.ErrNoStateFile)
write := func() {
err := file.Write(t.st.Started, t.st.Op, t.st.OpStep, t.st.Hook, t.st.CharmURL)
c.Assert(err, gc.IsNil)
}
if t.err != "" {
c.Assert(write, gc.PanicMatches, "invalid uniter state: "+t.err)
err := utils.WriteYaml(path, &t.st)
c.Assert(err, gc.IsNil)
_, err = file.Read()
c.Assert(err, gc.ErrorMatches, "cannot read charm state at .*: invalid uniter state: "+t.err)
continue
}
write()
st, err := file.Read()
c.Assert(err, gc.IsNil)
c.Assert(*st, gc.DeepEquals, t.st)
}
}
// storeManifest stores, into dataPath, the supplied manifest for the supplied charm.
func (d *manifestDeployer) storeManifest(url *charm.URL, manifest set.Strings) error {
if err := os.MkdirAll(d.DataPath(manifestsDataPath), 0755); err != nil {
return err
}
name := charm.Quote(url.String())
path := filepath.Join(d.DataPath(manifestsDataPath), name)
return utils.WriteYaml(path, manifest.SortedValues())
}
// Write stores the supplied state to the file.
func (f *StateFile) Write(started bool, op Op, step OpStep, hi *uhook.Info, url *charm.URL) error {
st := &State{
Started: started,
Op: op,
OpStep: step,
Hook: hi,
CharmURL: url,
}
if err := st.validate(); err != nil {
panic(err)
}
return utils.WriteYaml(f.path, st)
}