func (s *assignCleanSuite) assignUnit(unit *state.Unit) (*state.Machine, error) { if s.policy == state.AssignCleanEmpty { return unit.AssignToCleanEmptyMachine() } return unit.AssignToCleanMachine() }