def test_get_dc_name_nonexistent(self): equipment = Equipment.objects.get(serial='C1234567') serializer = EquipmentSerializer() name = None self.assertEqual(name, serializer.get_dc_name(equipment))
def test_get_dc_name_none(self): serializer = EquipmentSerializer() name = None self.assertEqual(name, serializer.get_dc_name(None))
def test_get_dc_name_existent(self): equipment = Equipment.objects.get(serial='G123456') serializer = EquipmentSerializer() name = equipment.dc.name self.assertEqual(name, serializer.get_dc_name(equipment))