def test_get_rack_none(self):
serializer = EquipmentSerializer()
rack = {
'rack': None,
'rackunit': None
}
self.assertEqual(rack, serializer.get_rack(None))
def test_get_rack_nonexistent(self):
equipment = Equipment.objects.get(serial='C1234567')
serializer = EquipmentSerializer()
rack = {
'rack': None,
'rackunit': None
}
self.assertEqual(rack, serializer.get_rack(equipment))
def test_get_rack_existent(self):
equipment = Equipment.objects.get(serial='G123456')
serializer = EquipmentSerializer()
rack = {
'rack': equipment.rack.name,
'rackunit': equipment.unit
}
self.assertEqual(rack, serializer.get_rack(equipment))
