func (s *statePoolSuite) TestKillWorkers(c *gc.C) { // Get some State instances via the pool and extract their // internal workers. st1, err := s.Pool.Get(s.ModelUUID1) c.Assert(err, jc.ErrorIsNil) w1 := state.GetInternalWorkers(st1) workertest.CheckAlive(c, w1) st2, err := s.Pool.Get(s.ModelUUID1) c.Assert(err, jc.ErrorIsNil) w2 := state.GetInternalWorkers(st2) workertest.CheckAlive(c, w2) // Now kill their workers. s.Pool.KillWorkers() // Ensure the internal workers for each State died. c.Check(workertest.CheckKilled(c, w1), jc.ErrorIsNil) c.Check(workertest.CheckKilled(c, w2), jc.ErrorIsNil) }
func assertClosed(c *gc.C, st *state.State) { w := state.GetInternalWorkers(st) c.Check(workertest.CheckKilled(c, w), jc.ErrorIsNil) }