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