def test_query_response(self, mock_query_response, mock_api_client_logon,
mock_api_client):
mock_api_client_logon.return_value = None
mock_api_client.return_value = None
mocked_return_value = '108cb8b0-0744-4dd9-8e35-ea8311cd6211'
mock_query_response.return_value = mocked_return_value
config = {
"auth": {
"bxapikey": "placeholder"
},
"client_info": "placeholder"
}
connection = {
"instance_crn": "placeholder",
"target_cos": "placeholder"
}
query = '{"query":"SELECT target.id from cos://us-geo/at-data/rest.1*.json STORED AS JSON c"}'
entry_point = EntryPoint(connection, config)
query_response = entry_point.create_query_connection(query)
assert query_response is not None
assert 'search_id' in query_response
assert query_response[
'search_id'] == "108cb8b0-0744-4dd9-8e35-ea8311cd6211"
def test_status_response(self, mock_status_response, mock_api_client_logon,
mock_api_client):
mock_api_client_logon.return_value = None
mock_api_client.return_value = None
mocked_return_value = json.loads(
'{"status": "completed", "end_time": "2018-08-28T15:51:24.899Z", "submit_time": "2018-08-28T15:51:19.899Z"}'
)
mock_status_response.return_value = mocked_return_value
config = {
"auth": {
"bxapikey": "placeholder"
},
"client_info": "placeholder"
}
connection = {
"instance_crn": "placeholder",
"target_cos": "placeholder"
}
search_id = "108cb8b0-0744-4dd9-8e35-ea8311cd6211"
entry_point = EntryPoint(connection, config)
status_response = entry_point.create_status_connection(search_id)
assert status_response is not None
assert status_response['success']
assert 'status' in status_response
assert status_response['status'] == Status.COMPLETED.value
def test_ping_endpoint(self, mock_ping_response, mock_api_client_logon,
mock_api_client):
mock_api_client_logon.return_value = None
mock_api_client.return_value = None
mocked_return_value = '[{"job_id": "placeholder", "status": "placeholder",\
"user_id": "placeholder", "statement": "placeholder",\
"resultset_location": "placeholder", "submit_time": "placeholder",\
"end_time": "placeholder", "error": "placeholder", error_message": "placeholder"}]'
mock_ping_response.return_value = mocked_return_value
config = {
"auth": {
"bxapikey": "placeholder"
},
"client_info": "placeholder"
}
connection = {
"instance_crn": "placeholder",
"target_cos": "placeholder"
}
entry_point = EntryPoint(connection, config)
ping_response = entry_point.ping_connection()
assert ping_response is not None
assert ping_response['success']
def test_delete_response(self, mock_delete_response, mock_api_client_logon,
mock_api_client):
mock_api_client_logon.return_value = None
mock_api_client.return_value = None
mocked_return_value = pd.DataFrame(columns=['Deleted Object'])
mocked_return_value = mocked_return_value.append([{
'Deleted Object':
'result/jobid=9b0f77b1-74e6-4953-84df-e0571a398ef7/part-00000-17db0efc-f563-45c2-9d12-560933cd01b6-c000-attempt_20181016200828_0024_m_000000_0.csv'
}],
ignore_index=True)
mock_delete_response.return_value = mocked_return_value
config = {
"auth": {
"bxapikey": "placeholder"
},
"client_info": "placeholder"
}
connection = {
"instance_crn": "placeholder",
"target_cos": "placeholder"
}
search_id = "108cb8b0-0744-4dd9-8e35-ea8311cd6211"
entry_point = EntryPoint(connection, config)
delete_response = entry_point.delete_query_connection(search_id)
assert delete_response is not None
assert delete_response['success']
def test_results_response(self, mock_results_response,
mock_api_client_logon, mock_api_client):
mock_api_client_logon.return_value = None
mock_api_client.return_value = None
mocked_return_value = pd.DataFrame(columns=['id'])
mocked_return_value = mocked_return_value.append(
[{
'id': 'crn:v1:bluemix:public:iam-identity::a/::apikey:1234'
}],
ignore_index=True)
mock_results_response.return_value = mocked_return_value
config = {
"auth": {
"bxapikey": "placeholder"
},
"client_info": "placeholder"
}
connection = {
"instance_crn": "placeholder",
"target_cos": "placeholder"
}
search_id = "108cb8b0-0744-4dd9-8e35-ea8311cd6211"
offset = 0
length = 1
entry_point = EntryPoint(connection, config)
results_response = entry_point.create_results_connection(
search_id, offset, length)
assert results_response is not None
assert results_response['success']
assert 'data' in results_response
assert len(results_response['data']) > 0
def test_query_flow(self, mock_results_response, mock_status_response,
mock_query_response, mock_api_client_logon,
mock_api_client):
mock_api_client_logon.return_value = None
mock_api_client.return_value = None
results_mock = pd.DataFrame(columns=['id'])
results_mock = results_mock.append(
[{
'id': 'crn:v1:bluemix:public:iam-identity::a/::apikey:1234'
}],
ignore_index=True)
query_mock = '108cb8b0-0744-4dd9-8e35-ea8311cd6211'
status_mock = json.loads(
'{"status": "completed", "end_time": "2018-08-28T15:51:24.899Z", "submit_time": "2018-08-28T15:51:19.899Z"}'
)
mock_results_response.return_value = results_mock
mock_status_response.return_value = status_mock
mock_query_response.return_value = query_mock
config = {
"auth": {
"bxapikey": "placeholder"
},
"client_info": "placeholder"
}
connection = {
"instance_crn": "placeholder",
"target_cos": "placeholder"
}
query = '{"query":"SELECT target.id from cos://us-geo/at-data/rest.1*.json STORED AS JSON c"}'
entry_point = EntryPoint(connection, config)
query_response = entry_point.create_query_connection(query)
assert query_response is not None
assert 'search_id' in query_response
assert query_response[
'search_id'] == "108cb8b0-0744-4dd9-8e35-ea8311cd6211"
search_id = "108cb8b0-0744-4dd9-8e35-ea8311cd6211"
status_response = entry_point.create_status_connection(search_id)
assert status_response is not None
assert 'status' in status_response
assert status_response['status'] == Status.COMPLETED.value
offset = 0
length = 1
results_response = entry_point.create_results_connection(
search_id, offset, length)
assert results_response is not None
assert 'data' in results_response
assert len(results_response['data']) > 0
def test_is_async(self, mock_api_client_logon, mock_api_client):
mock_api_client_logon.return_value = None
mock_api_client.return_value = None
config = {
"auth": {
"bxapikey": "placeholder"
},
"client_info": "placeholder"
}
connection = {
"instance_crn": "placeholder",
"target_cos": "placeholder"
}
entry_point = EntryPoint(connection, config)
check_async = entry_point.is_async()
assert check_async