def test_get_allocation_nonexistent(self):
equipment = Equipment.objects.get(serial='R123457')
serializer = EquipmentSerializer()
self.assertEqual(None, serializer.get_allocation(equipment))
def test_get_allocation_none(self):
serializer = EquipmentSerializer()
self.assertEqual(None, serializer.get_allocation(None))
def test_get_allocation_existent(self):
equipment = Equipment.objects.get(serial='G123456')
allocation = equipment.allocation
serializer = EquipmentSerializer()
self.assertEqual(allocation, serializer.get_allocation(equipment))
