def test_get_attrs_existent(self): equipment = Equipment.objects.get(serial='G123456') serializer = EquipmentSerializer() attrs = { 'key': 'value' } self.assertEqual(attrs, serializer.get_attrs(equipment))
def test_get_attrs_none(self): serializer = EquipmentSerializer() attrs = {} self.assertEqual(attrs, serializer.get_attrs(None))
def test_get_attrs_nonexistent(self): equipment = Equipment.objects.get(serial='C1234567') serializer = EquipmentSerializer() attrs = {} self.assertEqual(attrs, serializer.get_attrs(equipment))