def test_get_tasklist(self):
expected_tasklist = TaskList(entity_id="1",
title="Test List Title", updated_date=datetime(2012, 3, 10, 3, 30, 6))
mock_service_proxy = mock()
mock_get_request = mock()
when(mock_service_proxy).get(tasklist=expected_tasklist.entity_id).thenReturn(mock_get_request)
when(mock_get_request).execute().thenReturn(expected_tasklist.to_str_dict())
tasklist_service = TaskListService(mock_service_proxy)
result_tasklist = tasklist_service.get_tasklist(expected_tasklist.entity_id)
self.assertEqual(expected_tasklist, result_tasklist)
def test_add_list(self):
mock_service_proxy = mock()
mock_insert_request = mock()
tasklist = TaskList(entity_id="abcdfakekey", title="Test List 1",
updated_date=datetime(2012, 3, 22, 13, 50, 00))
when(mock_service_proxy).insert(body=any(dict)).thenReturn(mock_insert_request)
when(mock_insert_request).execute().thenReturn(tasklist.to_str_dict())
tasklist_service = TaskListService(mock_service_proxy)
result_tasklist = tasklist_service.add_tasklist(tasklist)
verify(mock_insert_request).execute()
self.assertIsNotNone(result_tasklist)
self.assertIsNotNone(result_tasklist.entity_id)
def test_get_all_tasklists(self):
mock_service_proxy = mock()
mock_list_request = mock()
tasklist_items_dict = {coggrinder.utilities.GoogleKeywords.ITEMS:[]}
expected_tasklists = dict()
for count in range(1, 4):
tasklist = TaskList(entity_id=str(count),
title="Test List " + str(count), updated_date=datetime(2012, 3, 10, 3, 30, 6))
expected_tasklists[tasklist.entity_id] = tasklist
tasklist_items_dict.get(coggrinder.utilities.GoogleKeywords.ITEMS).append(tasklist.to_str_dict())
when(mock_service_proxy).list().thenReturn(mock_list_request)
when(mock_list_request).execute().thenReturn(tasklist_items_dict)
tasklist_service = TaskListService(mock_service_proxy)
result_tasklists = tasklist_service.get_all_tasklists()
self.assertEqual(expected_tasklists, result_tasklists)
