140 lines
4.9 KiB
Python
140 lines
4.9 KiB
Python
import pytest
|
|
|
|
from infrasynth.notifications.models import (
|
|
ChannelConfig,
|
|
NotificationDispatch,
|
|
NotificationTemplate,
|
|
)
|
|
|
|
pytestmark = pytest.mark.django_db
|
|
|
|
TEMPLATES_URL = "/api/notifications/templates/"
|
|
DISPATCHES_URL = "/api/notifications/dispatches/"
|
|
CHANNELS_URL = "/api/notifications/channels/"
|
|
|
|
|
|
@pytest.fixture
|
|
def template():
|
|
return NotificationTemplate.objects.create(
|
|
slug="welcome",
|
|
name="Welcome",
|
|
channel="email",
|
|
subject_template="Hi",
|
|
body_template="Hello",
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def dispatch(template, user):
|
|
return NotificationDispatch.objects.create(
|
|
template=template,
|
|
recipient="a@b.com",
|
|
channel="email",
|
|
subject="Hi",
|
|
body="Hello",
|
|
status=NotificationDispatch.Status.SENT,
|
|
)
|
|
|
|
|
|
class TestNotificationTemplateViewSet:
|
|
def test_list_templates(self, authenticated_client, template):
|
|
response = authenticated_client.get(TEMPLATES_URL)
|
|
assert response.status_code == 200
|
|
assert response.data["count"] == 1
|
|
assert response.data["results"][0]["slug"] == "welcome"
|
|
|
|
def test_requires_auth(self, api_client, template):
|
|
assert api_client.get(TEMPLATES_URL).status_code == 401
|
|
|
|
def test_create_template(self, authenticated_client):
|
|
response = authenticated_client.post(
|
|
TEMPLATES_URL,
|
|
{
|
|
"slug": "order_shipped",
|
|
"name": "Order shipped",
|
|
"channel": "email",
|
|
"subject_template": "Your order shipped",
|
|
"body_template": "Tracking: {{ tracking }}",
|
|
},
|
|
format="json",
|
|
)
|
|
assert response.status_code == 201
|
|
assert NotificationTemplate.objects.filter(slug="order_shipped").exists()
|
|
|
|
def test_retrieve_and_update_template(self, authenticated_client, template):
|
|
response = authenticated_client.get(f"{TEMPLATES_URL}{template.id}/")
|
|
assert response.status_code == 200
|
|
response = authenticated_client.patch(
|
|
f"{TEMPLATES_URL}{template.id}/",
|
|
{"name": "Renamed"},
|
|
format="json",
|
|
)
|
|
assert response.status_code == 200
|
|
template.refresh_from_db()
|
|
assert template.name == "Renamed"
|
|
|
|
def test_delete_template(self, authenticated_client, template):
|
|
response = authenticated_client.delete(f"{TEMPLATES_URL}{template.id}/")
|
|
assert response.status_code == 204
|
|
assert not NotificationTemplate.objects.filter(pk=template.pk).exists()
|
|
|
|
|
|
class TestNotificationDispatchViewSet:
|
|
def test_list_dispatches(self, authenticated_client, dispatch):
|
|
response = authenticated_client.get(DISPATCHES_URL)
|
|
assert response.status_code == 200
|
|
assert response.data["count"] == 1
|
|
assert response.data["results"][0]["recipient"] == "a@b.com"
|
|
|
|
def test_retrieve_dispatch(self, authenticated_client, dispatch):
|
|
response = authenticated_client.get(f"{DISPATCHES_URL}{dispatch.id}/")
|
|
assert response.status_code == 200
|
|
assert response.data["template_info"]["slug"] == "welcome"
|
|
|
|
def test_list_does_not_allow_create(self, authenticated_client):
|
|
response = authenticated_client.post(
|
|
DISPATCHES_URL,
|
|
{"recipient": "x@y.com", "channel": "email", "subject": "S", "body": "B"},
|
|
format="json",
|
|
)
|
|
assert response.status_code == 405
|
|
|
|
def test_requires_auth(self, api_client, dispatch):
|
|
assert api_client.get(DISPATCHES_URL).status_code == 401
|
|
|
|
def test_filter_by_status(self, authenticated_client, dispatch):
|
|
response = authenticated_client.get(DISPATCHES_URL, {"status": "sent"})
|
|
assert response.data["count"] == 1
|
|
response = authenticated_client.get(DISPATCHES_URL, {"status": "failed"})
|
|
assert response.data["count"] == 0
|
|
|
|
|
|
class TestChannelConfigViewSet:
|
|
def test_list_channels(self, authenticated_client):
|
|
ChannelConfig.objects.create(
|
|
slug="email",
|
|
channel_type="email",
|
|
display_name="Email",
|
|
config={"host": "smtp.x.com"},
|
|
)
|
|
response = authenticated_client.get(CHANNELS_URL)
|
|
assert response.status_code == 200
|
|
assert response.data["count"] == 1
|
|
assert response.data["results"][0]["display_name"] == "Email"
|
|
|
|
def test_create_channel(self, authenticated_client):
|
|
response = authenticated_client.post(
|
|
CHANNELS_URL,
|
|
{
|
|
"slug": "sms",
|
|
"channel_type": "sms",
|
|
"display_name": "SMS",
|
|
"config": {"account_sid": "sid"},
|
|
},
|
|
format="json",
|
|
)
|
|
assert response.status_code == 201
|
|
assert ChannelConfig.objects.filter(slug="sms").exists()
|
|
|
|
def test_requires_auth(self, api_client):
|
|
assert api_client.get(CHANNELS_URL).status_code == 401
|