def test_load_objects(self):
"""Load an object from the JSON database."""
task_1 = Task("Test1")
task_2 = Task("Test2")
task_3 = Task("Test3")
self.json_handler.write(task_1.to_json())
self.json_handler.write(task_2.to_json())
self.json_handler.write(task_3.to_json())
# Check the first instance
json_data = self.json_handler.load_obj("Test1")
self.assertEquals(json_data, task_1.__dict__())
# Check the second instance
json_data = self.json_handler.load_obj("Test2")
self.assertEquals(json_data, task_2.__dict__())
# Check the third instance
json_data = self.json_handler.load_obj("Test3")
self.assertEquals(json_data, task_3.__dict__())
class TaskTests(TestCase):
def setUp(self):
self.task = Task()
self.pickler = Pickler()
self.unpickler = Unpickler()
def test_get_json_representation(self):
"""Get the json representation for the __dict__ object of the Task object..."""
task_dict = self.task.__dict__()
json_dict = self.unpickler.restore(self.task.to_json())
self.assertEquals(json_dict, task_dict)
def test_load_object(self):
task_1 = Task("Test1")
self.json_handler.write(task_1.to_json())
json_data = self.json_handler.load_obj("Test1")
# Assert that the data is the same.
self.assertEquals(json_data, task_1.__dict__())
