def test_alerts_client_get_all_alert_details_does_not_call_get_details_when_no_alerts_found(
self, mock_alerts_service_with_no_alerts,
mock_alert_rules_service):
alert_client = AlertsClient(mock_alerts_service_with_no_alerts,
mock_alert_rules_service)
query = AlertQuery()
list(alert_client.get_all_alert_details(query))
assert mock_alerts_service_with_no_alerts.get_details.call_count == 0
def test_alerts_client_get_all_alert_details_returns_empty_generator_when_no_alerts_found(
self, mock_alerts_service_with_no_alerts,
mock_alert_rules_service):
alert_client = AlertsClient(mock_alerts_service_with_no_alerts,
mock_alert_rules_service)
query = AlertQuery()
results = list(alert_client.get_all_alert_details(query))
assert results == []
def test_alerts_client_get_all_alert_details_calls_get_details_for_each_page(
self, mock_alerts_service_with_pages, mock_alert_rules_service):
mock_alerts_service = mock_alerts_service_with_pages(ascending=True)
alert_client = AlertsClient(mock_alerts_service,
mock_alert_rules_service)
query = AlertQuery()
list(alert_client.get_all_alert_details(query))
assert mock_alerts_service.get_details.call_args_list[0][0][0] == [
"A",
"B",
"C",
]
assert mock_alerts_service.get_details.call_args_list[1][0][0] == [
"D",
"E",
"F",
]
def test_alerts_client_get_all_alert_details_sorts_results_descending_when_specified(
self,
mock_alerts_service_with_pages,
mock_alert_rules_service,
mock_details,
sort_key,
):
mock_alerts_service = mock_alerts_service_with_pages(ascending=False)
mock_alerts_service.get_details = mock_details
alert_client = AlertsClient(mock_alerts_service,
mock_alert_rules_service)
query = AlertQuery()
query.sort_direction = "desc"
query.sort_key = sort_key
results = list(alert_client.get_all_alert_details(query))
assert results == [
ALERT_F, ALERT_E, ALERT_D, ALERT_C, ALERT_B, ALERT_A
]
def test_alerts_client_get_all_alert_details_sorts_results_according_to_query_sort_key(
self,
mock_alerts_service_with_pages,
mock_alert_rules_service,
mock_details,
sort_key,
):
mock_alerts_service = mock_alerts_service_with_pages(ascending=True)
mock_alerts_service.get_details = mock_details
alert_client = AlertsClient(mock_alerts_service,
mock_alert_rules_service)
query = AlertQuery()
query.sort_direction = "asc"
query.sort_key = sort_key
results = list(alert_client.get_all_alert_details(query))
assert results == [
ALERT_A, ALERT_B, ALERT_C, ALERT_D, ALERT_E, ALERT_F
]