From 1f66b222802e95ed29684f331e29e407c2030e8b Mon Sep 17 00:00:00 2001 From: KlemenSpruk Date: Wed, 18 Oct 2023 09:13:43 +0200 Subject: [PATCH] added test --- .../tests/api/test_create_mail.py | 22 +++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/django_project_base/notifications/tests/api/test_create_mail.py b/django_project_base/notifications/tests/api/test_create_mail.py index 1b812f2d..429f81ff 100644 --- a/django_project_base/notifications/tests/api/test_create_mail.py +++ b/django_project_base/notifications/tests/api/test_create_mail.py @@ -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 @@ -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)