from typing import Any from unittest.mock import Mock, patch from fittrackee import email_service from fittrackee.email.email import EmailMessage from flask import Flask from ..template_results.password_reset_request import expected_en_text_body class TestEmailMessage: def test_it_generate_email_data(self) -> None: message = EmailMessage( sender='fittrackee@example.com', recipient='test@test.com', subject='Fittrackee - test email', html="""\

Hello !

""", text='Hello !', ) message_data = message.generate_message() assert message_data.get('From') == 'fittrackee@example.com' assert message_data.get('To') == 'test@test.com' assert message_data.get('Subject') == 'Fittrackee - test email' message_string = message_data.as_string() assert 'Hello !' in message_string class TestEmailSending: email_data = { 'expiration_delay': '3 seconds', 'username': 'test', 'password_reset_url': 'http://localhost/password-reset?token=xxx', 'operating_system': 'Linux', 'browser_name': 'Firefox', } @staticmethod def get_args(call_args: Any) -> Any: if len(call_args) == 2: args, _ = call_args else: _, args, _ = call_args return args def assert_smtp(self, smtp: Mock) -> None: assert smtp.sendmail.call_count == 1 call_args = self.get_args(smtp.sendmail.call_args) assert call_args[0] == 'fittrackee@example.com' assert call_args[1] == 'test@test.com' assert expected_en_text_body in call_args[2] @patch('smtplib.SMTP_SSL') @patch('smtplib.SMTP') def test_it_sends_message( self, mock_smtp: Mock, mock_smtp_ssl: Mock, app: Flask ) -> None: email_service.send( template='password_reset_request', lang='en', recipient='test@test.com', data=self.email_data, ) smtp = mock_smtp.return_value.__enter__.return_value assert smtp.starttls.not_called self.assert_smtp(smtp) @patch('smtplib.SMTP_SSL') @patch('smtplib.SMTP') def test_it_sends_message_with_ssl( self, mock_smtp: Mock, mock_smtp_ssl: Mock, app_ssl: Flask ) -> None: email_service.send( template='password_reset_request', lang='en', recipient='test@test.com', data=self.email_data, ) smtp = mock_smtp_ssl.return_value.__enter__.return_value assert smtp.starttls.not_called self.assert_smtp(smtp) @patch('smtplib.SMTP_SSL') @patch('smtplib.SMTP') def test_it_sends_message_with_tls( self, mock_smtp: Mock, mock_smtp_ssl: Mock, app_tls: Flask ) -> None: email_service.send( template='password_reset_request', lang='en', recipient='test@test.com', data=self.email_data, ) smtp = mock_smtp.return_value.__enter__.return_value assert smtp.starttls.call_count == 1 self.assert_smtp(smtp)