def test_initial_search(self): gnip_search_client = GnipSearchClient() from_date = gnip_search_client.create_start_time() to_date = gnip_search_client.move_date_forward_by(from_date, 01, 01, 01) from_date = gnip_search_client.convert_date_to_gnip_format(from_date) to_date = gnip_search_client.convert_date_to_gnip_format(to_date) # open context to patch with HTTMock(self.gnip_mock): gnip_search_client.initial_search('Fake_Query', 10, from_date, to_date, 2) saved_json_blob_1 = gnip_search_client.load_json_blob(0) saved_json_blob_2 = gnip_search_client.load_json_blob(1) expected_json_blob_1 = json.loads('{"results": [{"id": 123}],"next": "fake_next_token"}') expected_json_blob_2 = json.loads('{"results": [{"id": 123}],"next": "fake_next_token"}') self.assertEqual(saved_json_blob_1, expected_json_blob_1) self.assertEqual(saved_json_blob_2, expected_json_blob_2) file_to_remove_1 = os.getcwd() + '/Gnip_Client/Gnip_Search_0.json' file_to_remove_2 = os.getcwd() + '/Gnip_Client/Gnip_Search_1.json' os.remove(file_to_remove_1) os.remove(file_to_remove_2) with HTTMock(self.gnip_mock_no_next): gnip_search_client.initial_search('Fake_Query', 10, from_date, to_date, 1) saved_json_blob_1 = gnip_search_client.load_json_blob(2) expected_json_blob_1 = json.loads('{"results": [{"id": 123}]}') self.assertEqual(saved_json_blob_1, expected_json_blob_1) file_to_remove_3 = os.getcwd() + '/Gnip_Client/Gnip_Search_2.json' os.remove(file_to_remove_3)
def test_save_json_blog_to_disk(self): gnip_search_client = GnipSearchClient() content = '{"results": [{"id": 123}]}' json_blob = json.loads(content) gnip_search_client.save_json_blog_to_disk(json_blob, 2) loaded_json_blob = gnip_search_client.load_json_blob(2) self.assertEqual(json_blob, loaded_json_blob) os.remove(gnip_search_client.get_file_path(2))