def test_add_cloud_alias_limit_raises_custom_error_on_limit(
self,
mock_connection,
user_context,
mock_user_client_error_on_adding_cloud_aliases,
):
detection_list_user_client = DetectionListUserService(
mock_connection,
user_context,
mock_user_client_error_on_adding_cloud_aliases,
)
with pytest.raises(Py42CloudAliasLimitExceededError) as err:
detection_list_user_client.add_cloud_alias("942897397520289999", "Test")
assert "Cloud alias limit exceeded." in str(err.value)
def test_add_cloud_alias_posts_expected_data(self, mock_connection,
user_context,
mock_user_client):
detection_list_user_client = DetectionListUserService(
mock_connection, user_context, mock_user_client)
detection_list_user_client.add_cloud_alias("942897397520289999",
u"Test")
posted_data = mock_connection.post.call_args[1]["json"]
assert mock_connection.post.call_count == 1
assert mock_connection.post.call_args[0][
0] == "v2/user/addcloudusernames"
assert (posted_data["tenantId"]
== user_context.get_current_tenant_id()
and posted_data["userId"] == "942897397520289999"
and posted_data["cloudUsernames"] == ["Test"])
