func (s *openSuite) TestOpenStateNormal(c *C) {
conf := agent.Conf{
StateInfo: s.StateInfo(c),
}
conf.OldPassword = "******"
st, changed, err := conf.OpenState()
c.Assert(err, IsNil)
defer st.Close()
c.Assert(changed, Equals, false)
c.Assert(st, NotNil)
}
func (s *openSuite) TestOpenStateNoPassword(c *C) {
conf := agent.Conf{
StateInfo: s.StateInfo(c),
}
conf.OldPassword = conf.StateInfo.Password
conf.StateInfo.Password = ""
st, changed, err := conf.OpenState()
c.Assert(err, IsNil)
defer st.Close()
c.Assert(changed, Equals, true)
c.Assert(st, NotNil)
p, err := trivial.RandomPassword()
c.Assert(err, IsNil)
c.Assert(conf.StateInfo.Password, HasLen, len(p))
c.Assert(conf.OldPassword, Equals, s.StateInfo(c).Password)
}
func runOnce(c *agent.Conf, a Agent) error {
st, passwordChanged, err := c.OpenState()
if err != nil {
return err
}
defer st.Close()
entity, err := a.Entity(st)
if state.IsNotFound(err) || err == nil && entity.Life() == state.Dead {
return worker.ErrDead
}
if err != nil {
return err
}
if passwordChanged {
if err := c.Write(); err != nil {
return err
}
if err := entity.SetMongoPassword(c.StateInfo.Password); err != nil {
return err
}
}
return a.RunOnce(st, entity)
}