def test_alert_rules_modules_raises_invalid_rule_type_error_when_adding_to_system_rule(
self,
mocker,
mock_microservice_client_factory,
mock_alert_rules_client,
mock_alerts_client_system_rule,
):
def add(*args, **kwargs):
base_err = mocker.MagicMock(spec=HTTPError)
base_err.response = mocker.MagicMock(spec=Response)
raise Py42InternalServerError(base_err)
mock_alert_rules_client.add_user.side_effect = add
mock_microservice_client_factory.get_alert_rules_client.return_value = (
mock_alert_rules_client
)
mock_microservice_client_factory.get_alerts_client.return_value = (
mock_alerts_client_system_rule
)
alert_rules_module = AlertRulesModule(mock_microservice_client_factory)
with pytest.raises(Py42InvalidRuleOperationError) as err:
alert_rules_module.add_user(self._rule_id, self._rule_id)
assert (
"Only alert rules with a source of 'Alerting' can be targeted by this command."
in str(err.value)
)
def test_alert_rules_module_calls_add_user_with_expected_value(
self, mock_microservice_client_factory, mock_alert_rules_client
):
mock_microservice_client_factory.get_alert_rules_client.return_value = (
mock_alert_rules_client
)
alert_rules_module = AlertRulesModule(mock_microservice_client_factory)
alert_rules_module.add_user(self._rule_id, self._rule_id)
mock_alert_rules_client.add_user.assert_called_once_with(
self._rule_id, self._rule_id
)