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)