def test_get_query_details_posts_to_expected_url(self, mock_session,
user_context,
successful_post):
alert_client = AlertClient(mock_session, user_context)
alert_ids = ["ALERT_ID_1", "ALERT_ID_2"]
alert_client.get_query_details(alert_ids)
assert mock_session.post.call_args[0][0] == "/svc/api/v1/query-details"
def test_resolve_when_given_single_alert_id_posts_expected_data(
self, mock_session, user_context, successful_post):
alert_client = AlertClient(mock_session, user_context)
alert_client.resolve("ALERT_ID_1")
post_data = json.loads(mock_session.post.call_args[1]["data"])
assert (post_data["tenantId"] == TENANT_ID_FROM_RESPONSE
and post_data["alertIds"][0] == "ALERT_ID_1")
def test_search_posts_to_expected_url(self, mock_session, user_context,
successful_post):
alert_client = AlertClient(mock_session, user_context)
_filter = AlertState.eq("OPEN")
query = AlertQuery(TENANT_ID_FROM_RESPONSE, _filter)
alert_client.search(query)
assert mock_session.post.call_args[0][0] == u"/svc/api/v1/query-alerts"
def test_resolve_when_given_tenant_id_posts_expected_data(
self, mock_session, user_context, successful_post):
alert_client = AlertClient(mock_session, user_context)
alert_ids = ["ALERT_ID_1", "ALERT_ID_2"]
alert_client.resolve(alert_ids, "some-tenant-id")
post_data = json.loads(mock_session.post.call_args[1]["data"])
assert (post_data["tenantId"] == "some-tenant-id"
and post_data["alertIds"][0] == "ALERT_ID_1"
and post_data["alertIds"][1] == "ALERT_ID_2")
def test_reopen_when_not_given_tenant_id_posts_expected_data(
self, mock_session, user_context, successful_post):
alert_client = AlertClient(mock_session, user_context)
alert_ids = ["ALERT_ID_1", "ALERT_ID_2"]
alert_client.reopen(alert_ids)
post_data = json.loads(mock_session.post.call_args[1]["data"])
assert (post_data["tenantId"] == TENANT_ID_FROM_RESPONSE
and post_data["alertIds"][0] == "ALERT_ID_1"
and post_data["alertIds"][1] == "ALERT_ID_2")
def test_get_query_details_when_not_given_tenant_id_posts_expected_data(
self, mock_session, user_context, successful_response):
mock_session.post.return_value = successful_response
alert_client = AlertClient(mock_session, user_context)
alert_ids = ["ALERT_ID_1", "ALERT_ID_2"]
alert_client.get_query_details(alert_ids)
post_data = json.loads(mock_session.post.call_args[1]["data"])
assert (post_data["tenantId"] == TENANT_ID_FROM_RESPONSE
and post_data["alertIds"][0] == "ALERT_ID_1"
and post_data["alertIds"][1] == "ALERT_ID_2")
def test_search_posts_expected_data(self, mock_session, user_context,
successful_post):
alert_client = AlertClient(mock_session, user_context)
_filter = AlertState.eq("OPEN")
query = AlertQuery(TENANT_ID_FROM_RESPONSE, _filter)
alert_client.search(query)
post_data = json.loads(mock_session.post.call_args[1]["data"])
assert (post_data["tenantId"] == TENANT_ID_FROM_RESPONSE
and post_data["groupClause"] == "AND"
and post_data["srtKey"] == "CreatedAt"
and post_data["srtDirection"] == "desc"
and post_data["pgSize"] == 10000 and post_data["pgNum"] == 0
and post_data["groups"][0]["filterClause"] == "AND"
and post_data["groups"][0]["filters"][0]["operator"] == "IS"
and post_data["groups"][0]["filters"][0]["term"] == "state"
and post_data["groups"][0]["filters"][0]["value"] == "OPEN")
def test_reopen_posts_to_expected_url(self, mock_session, user_context,
successful_post):
alert_client = AlertClient(mock_session, user_context)
alert_ids = ["ALERT_ID_1", "ALERT_ID_2"]
alert_client.reopen(alert_ids, "some-tenant-id")
assert mock_session.post.call_args[0][0] == "/svc/api/v1/reopen-alert"