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__())