def test_serialize_obj(self):
data = ModelDataSet(Apple)
data.resource_name = "apple"
apple = Apple(name="test")
apple.save()
serialized = data.serialize_obj(apple)
self.assertEqual(serialized, {"id": 1, "name": "test", "resource_uri": "/v1/apple/1/"})
serialized = data.serialize_obj(apple, ["name"])
self.assertEqual(serialized, {"id": 1, "name": "test", "resource_uri": "/v1/apple/1/"})
def test_serialize_list(self):
data = ModelDataSet(Apple)
data.resource_name = "apple"
self.assertEqual(len(data.serialize_list()), 0)
apple = Apple(name="test")
apple.save()
self.assertEqual(len(data.serialize_list()), 1)
self.assertEqual(sorted(data.serialize_list()[0].keys()), ["id", "name", "resource_uri"])
self.assertEqual(len(data.serialize_list(["name"])), 1)
self.assertEqual(sorted(data.serialize_list(["name"])[0].keys()), ["id", "name", "resource_uri"])
