def test_perform_email(initialized_db):
repository = Repository("buynlarge", "orgrepo")
notification = Notification(
uuid="fake",
event_name="repo_push",
method_name="email",
event_config_dict={},
method_config_dict={"email": "*****@*****.**"},
repository=repository,
)
event_handler = NotificationEvent.get_event("repo_push")
sample_data = event_handler.get_sample_data(repository.namespace_name,
repository.name, {})
mock = Mock()
def get_mock(*args, **kwargs):
return mock
with patch("notifications.notificationmethod.Message", get_mock):
method = EmailMethod()
method.perform(notification, event_handler, {
"event_data": sample_data,
"performer_data": {}
})
mock.send.assert_called_once()
def test_perform_http_call(method, method_config, netloc, initialized_db):
repository = Repository("buynlarge", "orgrepo")
notification = Notification(
uuid="fake",
event_name="repo_push",
method_name=method.method_name(),
event_config_dict={},
method_config_dict=method_config,
repository=repository,
)
event_handler = NotificationEvent.get_event("repo_push")
sample_data = event_handler.get_sample_data(repository.namespace_name,
repository.name, {})
url_hit = [False]
@urlmatch(netloc=netloc)
def url_handler(_, __):
url_hit[0] = True
return ""
with HTTMock(url_handler):
method().perform(notification, event_handler, {
"event_data": sample_data,
"performer_data": {}
})
assert url_hit[0]
def process_queue_item(self, job_details):
notification = model.get_enabled_notification(
job_details['notification_uuid'])
if notification is None:
return
event_name = notification.event_name
method_name = notification.method_name
try:
event_handler = NotificationEvent.get_event(event_name)
method_handler = NotificationMethod.get_method(method_name)
except InvalidNotificationMethodException as ex:
logger.exception('Cannot find notification method: %s', ex.message)
raise JobException('Cannot find notification method: %s' %
ex.message)
except InvalidNotificationEventException as ex:
logger.exception('Cannot find notification event: %s', ex.message)
raise JobException('Cannot find notification event: %s' %
ex.message)
if event_handler.should_perform(job_details['event_data'],
notification):
try:
method_handler.perform(notification, event_handler,
job_details)
model.reset_number_of_failures_to_zero(notification)
except (JobException, KeyError) as exc:
model.increment_notification_failure_count(notification)
raise exc
def test_perform_quay_notification(target, expected_users, initialized_db):
repository = Repository("buynlarge", "orgrepo")
notification = Notification(
uuid="fake",
event_name="repo_push",
method_name="quay",
event_config_dict={},
method_config_dict={"target": target},
repository=repository,
)
event_handler = NotificationEvent.get_event("repo_push")
sample_data = event_handler.get_sample_data(repository.namespace_name,
repository.name, {})
method = QuayNotificationMethod()
method.perform(notification, event_handler, {"event_data": sample_data})
# Ensure that the notification was written for all the expected users.
if target["kind"] != "team":
user = model.user.get_namespace_user(target["name"])
assert len(
model.notification.list_notifications(user,
kind_name="repo_push")) > 0
def test_build_notification(event_name, initialized_db):
# Create the notification event.
found = NotificationEvent.get_event(event_name)
sample_data = found.get_sample_data('foo', 'bar', {'level': 'low'})
# Make sure all calls succeed.
notification_data = {
'performer_data': {},
}
found.get_level(sample_data, notification_data)
found.get_summary(sample_data, notification_data)
found.get_message(sample_data, notification_data)
def test_build_notification(event_name, initialized_db):
# Create the notification event.
found = NotificationEvent.get_event(event_name)
sample_data = found.get_sample_data("foo", "bar", {"level": "low"})
# Make sure all calls succeed.
notification_data = {
"performer_data": {},
}
found.get_level(sample_data, notification_data)
found.get_summary(sample_data, notification_data)
found.get_message(sample_data, notification_data)
def test_perform_quay_notification(target, expected_users, initialized_db):
repository = Repository('buynlarge', 'orgrepo')
notification = Notification(uuid='fake', event_name='repo_push', method_name='quay',
event_config_dict={}, method_config_dict={'target': target},
repository=repository)
event_handler = NotificationEvent.get_event('repo_push')
sample_data = event_handler.get_sample_data(repository.namespace_name, repository.name, {})
method = QuayNotificationMethod()
method.perform(notification, event_handler, {'event_data': sample_data})
# Ensure that the notification was written for all the expected users.
if target['kind'] != 'team':
user = model.user.get_namespace_user(target['name'])
assert len(model.notification.list_notifications(user, kind_name='repo_push')) > 0
def test_perform_email(initialized_db):
repository = Repository('buynlarge', 'orgrepo')
notification = Notification(uuid='fake', event_name='repo_push', method_name='email',
event_config_dict={}, method_config_dict={'email': '*****@*****.**'},
repository=repository)
event_handler = NotificationEvent.get_event('repo_push')
sample_data = event_handler.get_sample_data(repository.namespace_name, repository.name, {})
mock = Mock()
def get_mock(*args, **kwargs):
return mock
with patch('notifications.notificationmethod.Message', get_mock):
method = EmailMethod()
method.perform(notification, event_handler, {'event_data': sample_data, 'performer_data': {}})
mock.send.assert_called_once()
def test_perform_http_call(method, method_config, netloc, initialized_db):
repository = Repository('buynlarge', 'orgrepo')
notification = Notification(uuid='fake', event_name='repo_push', method_name=method.method_name(),
event_config_dict={}, method_config_dict=method_config,
repository=repository)
event_handler = NotificationEvent.get_event('repo_push')
sample_data = event_handler.get_sample_data(repository.namespace_name, repository.name, {})
url_hit = [False]
@urlmatch(netloc=netloc)
def url_handler(_, __):
url_hit[0] = True
return ''
with HTTMock(url_handler):
method().perform(notification, event_handler, {'event_data': sample_data, 'performer_data': {}})
assert url_hit[0]
def queue_test_notification(self, uuid):
try:
notification = model.notification.get_repo_notification(uuid)
except InvalidNotificationException:
return None
event_config = json.loads(notification.event_config_json or "{}")
event_info = NotificationEvent.get_event(notification.event.name)
sample_data = event_info.get_sample_data(
notification.repository.namespace_user.username,
notification.repository.name,
event_config,
)
notification_data = build_notification_data(notification, sample_data)
notification_queue.put(
[
notification.repository.namespace_user.username,
notification.uuid,
notification.event.name,
],
json.dumps(notification_data),
)
return self._notification(notification)
def test_create_notifications(event_kind):
assert NotificationEvent.get_event(event_kind) is not None