func (s *AssignSuite) assertAssignedUnit(c *gc.C, unit *state.Unit) string { // Get service networks. service, err := unit.Service() c.Assert(err, gc.IsNil) includeNetworks, excludeNetworks, err := service.Networks() c.Assert(err, gc.IsNil) // Check the machine on the unit is set. machineId, err := unit.AssignedMachineId() c.Assert(err, gc.IsNil) // Check that the principal is set on the machine. machine, err := s.State.Machine(machineId) c.Assert(err, gc.IsNil) include, exclude, err := machine.RequestedNetworks() c.Assert(err, gc.IsNil) c.Assert(include, gc.DeepEquals, includeNetworks) c.Assert(exclude, gc.DeepEquals, excludeNetworks) machineUnits, err := machine.Units() c.Assert(err, gc.IsNil) c.Assert(machineUnits, gc.HasLen, 1) // Make sure it is the right unit. c.Assert(machineUnits[0].Name(), gc.Equals, unit.Name()) return machineId }