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_create_start_time(self):
gnip_search_client = GnipSearchClient()
expected = datetime(2016, 04, 14, 02, 30)
self.assertEqual(expected, gnip_search_client.create_start_time())
