def test_valid_search(self, mock_search):
dr = dummy_request()
expected_return_code = 200
mock_search.return_value = []
actual_return_code, _ = log_source_search(dr)
assert actual_return_code == expected_return_code
def test_clusters_errors(err_instance, expected_return_code):
dr = dummy_request()
with patch("mycroft.models.aws_connections.TableConnection.get_connection"):
with patch("mycroft.views.clusters.list_all_clusters") as list_all_clusters:
list_all_clusters.side_effect = err_instance
actual_return_code, _ = clusters(dr)
assert actual_return_code == expected_return_code
def test_jobs_errors(err_instance, expected_return_code):
dr = dummy_request()
with patch('mycroft.models.aws_connections.TableConnection.get_connection'):
with patch('mycroft.views.jobs.list_all_jobs') as list_all_jobs:
list_all_jobs.side_effect = err_instance
actual_return_code, _ = jobs(dr)
assert actual_return_code == expected_return_code
def test_schema_name_and_version(request_method):
dr = dummy_request()
dr.method = request_method
expected_return_code = 200
with patch('mycroft.views.schema.s3_bucket_action') as bucket_action:
bucket_action.return_value = {}
actual_return_code, _ = schema_name_and_version(dr)
assert actual_return_code == expected_return_code
def test_jobs_update_job_errors(err_instance, expected_return_code):
dr = dummy_request()
with patch('mycroft.models.aws_connections.TableConnection.get_connection'):
with patch('mycroft.views.jobs.get_scanner_queue'):
with patch('mycroft.views.jobs.put_job') as put_job:
put_job.side_effect = err_instance
actual_return_code, _ = jobs_update_job(dr)
assert actual_return_code == expected_return_code
def test_schema(log_name):
dr = dummy_request()
dr.matchdict["log_name"] = log_name
expected_return_code = 200
with patch('mycroft.views.schema.s3_bucket_action') as bucket_action:
bucket_action.return_value = {}
actual_return_code, _ = schema(dr)
assert actual_return_code == expected_return_code
def test_jobs_filtered_errors(err_instance, expected_return_code):
dr = dummy_request()
dr.matchdict["log_name"] = "dummy_log_name"
with patch('mycroft.models.aws_connections.TableConnection.get_connection'):
with patch('mycroft.views.jobs.list_jobs_by_name') as list_action:
list_action.side_effect = err_instance
actual_return_code, _ = jobs_filtered(dr)
assert actual_return_code == expected_return_code
def test_jobs_errors(err_instance, expected_return_code):
dr = dummy_request()
with patch(
'mycroft.models.aws_connections.TableConnection.get_connection'):
with patch('mycroft.views.jobs.list_all_jobs') as list_all_jobs:
list_all_jobs.side_effect = err_instance
actual_return_code, _ = jobs(dr)
assert actual_return_code == expected_return_code
def test_jobs_filtered_errors(err_instance, expected_return_code):
dr = dummy_request()
dr.matchdict["log_name"] = "dummy_log_name"
with patch(
'mycroft.models.aws_connections.TableConnection.get_connection'):
with patch('mycroft.views.jobs.list_jobs_by_name') as list_action:
list_action.side_effect = err_instance
actual_return_code, _ = jobs_filtered(dr)
assert actual_return_code == expected_return_code
def test_cluster_by_name(cluster_name, parm):
dr = dummy_request()
dr.matchdict["cluster_name"] = cluster_name
expected_return_code = 200
with patch("mycroft.models.aws_connections.TableConnection.get_connection"):
with patch(parm) as list_action:
list_action.return_value = {}
actual_return_code, _ = cluster_by_name(dr)
assert actual_return_code == expected_return_code
def test_clusters(request_method, parm):
dr = dummy_request()
dr.method = request_method
expected_return_code = 200
with patch("mycroft.models.aws_connections.TableConnection.get_connection"):
with patch(parm) as request_method_action:
request_method_action.return_value = {}
actual_return_code, _ = clusters(dr)
assert actual_return_code == expected_return_code
def test_cluster_by_name_errors(err_instance, expected_return_code):
dr = dummy_request()
dr.matchdict["log_name"] = "dummy_log_name"
with patch("mycroft.models.aws_connections.TableConnection.get_connection"):
with patch("mycroft.views.clusters.list_cluster_by_name") as list_action:
list_action.side_effect = err_instance
actual_return_code, _ = cluster_by_name(dr)
assert actual_return_code == expected_return_code
def test_jobs_update_job_errors(err_instance, expected_return_code):
dr = dummy_request()
with patch(
'mycroft.models.aws_connections.TableConnection.get_connection'):
with patch('mycroft.views.jobs.get_scanner_queue'):
with patch('mycroft.views.jobs.put_job') as put_job:
put_job.side_effect = err_instance
actual_return_code, _ = jobs_update_job(dr)
assert actual_return_code == expected_return_code
def test_jobs_update_job(request_method, parm):
dr = dummy_request()
dr.method = request_method
expected_return_code = 200
with patch('mycroft.models.aws_connections.TableConnection.get_connection'):
with patch('mycroft.views.jobs.get_scanner_queue'):
with patch(parm) as list_action:
list_action.return_value = {}
actual_return_code, _ = jobs_update_job(dr)
assert actual_return_code == expected_return_code
def test_runs_filtered(err_type, expected_return_code):
dr = dummy_request()
with patch('mycroft.models.aws_connections.TableConnection.get_connection'):
with patch('mycroft.views.runs.list_runs_by_job_id') as list_runs:
if err_type is not None:
list_runs.side_effect = err_type()
else:
list_runs.returns = {}
actual_return_code, _ = runs_filtered(dr)
assert actual_return_code == expected_return_code
def test_jobs_filtered(log_version, parm):
dr = dummy_request()
dr.matchdict["log_name"] = "dummy_log_name"
dr.matchdict["log_schema_version"] = log_version
expected_return_code = 200
with patch('mycroft.models.aws_connections.TableConnection.get_connection'):
with patch(parm) as list_action:
list_action.return_value = {}
actual_return_code, _ = jobs_filtered(dr)
assert actual_return_code == expected_return_code
def test_jobs_filtered(log_version, parm):
dr = dummy_request()
dr.matchdict["log_name"] = "dummy_log_name"
dr.matchdict["log_schema_version"] = log_version
expected_return_code = 200
with patch(
'mycroft.models.aws_connections.TableConnection.get_connection'):
with patch(parm) as list_action:
list_action.return_value = {}
actual_return_code, _ = jobs_filtered(dr)
assert actual_return_code == expected_return_code
def test_jobs_update_job(request_method, parm):
dr = dummy_request()
dr.method = request_method
expected_return_code = 200
with patch(
'mycroft.models.aws_connections.TableConnection.get_connection'):
with patch('mycroft.views.jobs.get_scanner_queue'):
with patch(parm) as list_action:
list_action.return_value = {}
actual_return_code, _ = jobs_update_job(dr)
assert actual_return_code == expected_return_code
def test_connections(conn, result):
dr = dummy_request()
with patch.dict(
_NAME_TO_FUNC, {
'dynamodb': mock_get_dynamodb_conn,
's3': mock_get_s3_conn,
'sqs': mock_get_sqs_conn,
}):
mock_get_s3_conn.return_value = conn
mock_get_sqs_conn.return_value = conn
mock_get_dynamodb_conn.return_value = conn
res = healthcheck(dr)
assert result in res
def test_connections(conn, result):
dr = dummy_request()
with patch.dict(
_NAME_TO_FUNC,
{
'dynamodb': mock_get_dynamodb_conn,
's3': mock_get_s3_conn,
'sqs': mock_get_sqs_conn,
}
):
mock_get_s3_conn.return_value = conn
mock_get_sqs_conn.return_value = conn
mock_get_dynamodb_conn.return_value = conn
res = healthcheck(dr)
assert result in res
def test_search_errors(self, mock_search, error_instance, expected_return):
dr = dummy_request()
mock_search.side_effect = error_instance
actual_return_code, _ = log_source_search(dr)
assert actual_return_code == expected_return
def test_schema_errors(err_instance, expected_return_code):
dr = dummy_request()
with patch('mycroft.views.schema.s3_bucket_action') as bucket_action:
bucket_action.side_effect = err_instance
actual_return_code, _ = schema(dr)
assert actual_return_code == expected_return_code
