mirror of
https://github.com/quay/quay.git
synced 2026-01-27 18:42:52 +03:00
5f63b3a7bb
* chore: drop deprecated tables and remove unused code * isort imports * migration: check for table existence before drop
87 lines
2.8 KiB
Python
87 lines
2.8 KiB
Python
from test.fixtures import *
|
|
|
|
import pytest
|
|
from httmock import HTTMock, urlmatch
|
|
from mock import Mock, patch
|
|
|
|
from notifications.models_interface import Repository
|
|
from notifications.notificationevent import RepoPushEvent
|
|
from notifications.notificationmethod import (
|
|
CannotValidateNotificationMethodException,
|
|
EmailMethod,
|
|
FlowdockMethod,
|
|
HipchatMethod,
|
|
QuayNotificationMethod,
|
|
SlackMethod,
|
|
WebhookMethod,
|
|
)
|
|
from workers.notificationworker.models_pre_oci import pre_oci_model as model
|
|
from workers.notificationworker.notificationworker import NotificationWorker
|
|
|
|
|
|
def test_basic_notification_endtoend(initialized_db):
|
|
# Ensure the public user doesn't have any notifications.
|
|
assert not model.user_has_local_notifications("public")
|
|
|
|
# Add a basic build notification.
|
|
notification_uuid = model.create_notification_for_testing("public")
|
|
event_data = {}
|
|
|
|
# Fire off the queue processing.
|
|
worker = NotificationWorker(None)
|
|
worker.process_queue_item(
|
|
{
|
|
"notification_uuid": notification_uuid,
|
|
"event_data": event_data,
|
|
}
|
|
)
|
|
|
|
# Ensure the notification was handled.
|
|
assert model.user_has_local_notifications("public")
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"method,method_config,netloc",
|
|
[
|
|
(QuayNotificationMethod, {"target": {"name": "devtable", "kind": "user"}}, None),
|
|
(EmailMethod, {"email": "jschorr@devtable.com"}, None),
|
|
(WebhookMethod, {"url": "http://example.com"}, "example.com"),
|
|
(FlowdockMethod, {"flow_api_token": "sometoken"}, "api.flowdock.com"),
|
|
(HipchatMethod, {"notification_token": "token", "room_id": "foo"}, "api.hipchat.com"),
|
|
(SlackMethod, {"url": "http://example.com"}, "example.com"),
|
|
],
|
|
)
|
|
def test_notifications(method, method_config, netloc, initialized_db):
|
|
url_hit = [False]
|
|
|
|
@urlmatch(netloc=netloc)
|
|
def url_handler(_, __):
|
|
url_hit[0] = True
|
|
return ""
|
|
|
|
mock = Mock()
|
|
|
|
def get_mock(*args, **kwargs):
|
|
return mock
|
|
|
|
with patch("notifications.notificationmethod.Message", get_mock):
|
|
with HTTMock(url_handler):
|
|
# Add a basic build notification.
|
|
notification_uuid = model.create_notification_for_testing(
|
|
"public", method_name=method.method_name(), method_config=method_config
|
|
)
|
|
event_data = RepoPushEvent().get_sample_data("devtable", "simple", {})
|
|
|
|
# Fire off the queue processing.
|
|
worker = NotificationWorker(None)
|
|
worker.process_queue_item(
|
|
{
|
|
"notification_uuid": notification_uuid,
|
|
"event_data": event_data,
|
|
"performer_data": {},
|
|
}
|
|
)
|
|
|
|
if netloc is not None:
|
|
assert url_hit[0]
|