Skip to content

Commit

Permalink
added test
Browse files Browse the repository at this point in the history
  • Loading branch information
KlemenSpruk committed Oct 18, 2023
1 parent 20a549a commit 1f66b22
Showing 1 changed file with 22 additions and 0 deletions.
22 changes: 22 additions & 0 deletions django_project_base/notifications/tests/api/test_create_mail.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,11 @@
import socket

import swapper
from django.urls import reverse
from rest_framework import status
from rest_framework.response import Response

from django_project_base.notifications.base.channels.mail_channel import MailChannel
from django_project_base.notifications.email_notification import EMailNotification
from django_project_base.notifications.models import DjangoProjectBaseMessage, DjangoProjectBaseNotification
from django_project_base.notifications.tests.notifications_transaction_test_case import NotificationsTransactionTestCase
Expand Down Expand Up @@ -34,3 +40,19 @@ def test_send_mail(self):
).send()
)
self.assertEqual(DjangoProjectBaseNotification.objects.all().count(), 1)

def test_api_create_mail(self):
self._login_to_api_client_with_test_user()
response: Response = self.api_client.put(
reverse("notification-detail", kwargs=dict(pk="new")),
format="json",
data={
"message_body": "demo",
"message_subject": "demo",
"message_to": [self.test_user.pk],
"send_on_channels": [MailChannel.name],
},
HTTP_HOST=socket.gethostname().lower(),
HTTP_CURRENT_PROJECT=swapper.load_model("django_project_base", "Project").objects.first().slug,
)
self.assertEqual(response.status_code, status.HTTP_201_CREATED)

0 comments on commit 1f66b22

Please sign in to comment.