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))