def test_serialize_list_of_records(self): test_values = { "id": 123, "tagged_vlans": [ { "id": 1, "url": "http://localhost:8000/api/ipam/vlans/1/", "vid": 1, "name": "test1", "display_name": "test1", }, { "id": 2, "url": "http://localhost:8000/api/ipam/vlans/2/", "vid": 2, "name": "test 2", "display_name": "test2", }, ], } test_obj = Record(test_values, Mock(base_url="test"), None) test = test_obj.serialize() self.assertEqual(test["tagged_vlans"], [1, 2])
def test_serialize_list_of_ints(self): test_values = {"id": 123, "units": [12]} test_obj = Record(test_values, None, None) test = test_obj.serialize() self.assertEqual(test["units"], [12])