Skip to content

Commit 6513233

Browse files
committed
Add tests
1 parent 41ebd6a commit 6513233

File tree

1 file changed

+26
-1
lines changed

1 file changed

+26
-1
lines changed

tests/test_email.py

Lines changed: 26 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,12 @@
11
import pytest
2+
from sqlalchemy.orm import Session
23

3-
from app.internal.email import mail, send_email_invitation, send_email_file
4+
from app.internal.email import mail, send_email_invitation, send_email_file, send
45
from fastapi import BackgroundTasks, status
56

7+
from app.database.models import User
8+
from tests.utils import create_model, delete_instance
9+
610

711
def test_email_send(client, user, event, smtpd):
812
mail.config.SUPPRESS_SEND = 1
@@ -207,3 +211,24 @@ def test_send_mail_bad_file_internal(client,
207211
def test_send_mail_good_file_internal(client, configured_smtpd):
208212
background_task = BackgroundTasks()
209213
assert send_email_file(__file__, "good@mail.com", background_task)
214+
215+
216+
@pytest.fixture
217+
def bad_user(session: Session) -> User:
218+
test_user = create_model(
219+
session, User,
220+
username='test_username',
221+
password='test_password',
222+
email='test.email#gmail.com',
223+
)
224+
yield test_user
225+
delete_instance(session, test_user)
226+
227+
228+
def test_send(session, bad_user, event):
229+
background_task = BackgroundTasks()
230+
assert not send(session=session,
231+
event_used=1,
232+
user_to_send=1,
233+
title="Test",
234+
background_tasks=background_task)

0 commit comments

Comments
 (0)