Tests - minor refactor

2020-12-30 12:09:34 +01:00
parent 806c333bf5
commit 6f627ef6c9
24 changed files with 2 additions and 2 deletions
--- a/fittrackee/tests/email/init.py
+++ b/fittrackee/tests/email/init.py
--- a/fittrackee/tests/email/test_email.py
+++ b/fittrackee/tests/email/test_email.py
@@ -0,0 +1,102 @@
+from unittest.mock import patch
+
+from fittrackee import email_service
+from fittrackee.email.email import EmailMessage
+
+from ..template_results.password_reset_request import expected_en_text_body
+
+
+class TestEmailMessage:
+    def test_it_generate_email_data(self):
+        message = EmailMessage(
+            sender='fittrackee@example.com',
+            recipient='test@test.com',
+            subject='Fittrackee - test email',
+            html="""\
+<html>
+  <body>
+    <p>Hello !</p>
+  </body>
+</html>
+            """,
+            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):
+        if len(call_args) == 2:
+            args, _ = call_args
+        else:
+            _, args, _ = call_args
+        return args
+
+    def assert_smtp(self, smtp):
+        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_smtp_ssl, app):
+
+        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_smtp_ssl, app_ssl
+    ):
+        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_smtp_ssl, app_tls
+    ):
+        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)
--- a/fittrackee/tests/email/test_email_template_password_request.py
+++ b/fittrackee/tests/email/test_email_template_password_request.py
@@ -0,0 +1,79 @@
+import pytest
+from fittrackee.email.email import EmailTemplate
+
+from ..template_results.password_reset_request import (
+    expected_en_html_body,
+    expected_en_text_body,
+    expected_fr_html_body,
+    expected_fr_text_body,
+)
+
+
+class TestEmailTemplateForPasswordRequest:
+    @pytest.mark.parametrize(
+        'lang, expected_subject',
+        [
+            ('en', 'FitTrackee - Password reset request'),
+            ('fr', 'FitTrackee - Réinitialiser votre mot de passe'),
+        ],
+    )
+    def test_it_gets_subject(self, app, lang, expected_subject):
+        email_template = EmailTemplate(app.config.get('TEMPLATES_FOLDER'))
+
+        subject = email_template.get_content(
+            'password_reset_request', lang, 'subject.txt', {}
+        )
+
+        assert subject == expected_subject
+
+    @pytest.mark.parametrize(
+        'lang, expected_text_body',
+        [('en', expected_en_text_body), ('fr', expected_fr_text_body)],
+    )
+    def test_it_gets_text_body(self, app, lang, expected_text_body):
+        email_template = EmailTemplate(app.config.get('TEMPLATES_FOLDER'))
+        email_data = {
+            'expiration_delay': '3 seconds' if lang == 'en' else '3 secondes',
+            'username': 'test',
+            'password_reset_url': 'http://localhost/password-reset?token=xxx',
+            'operating_system': 'Linux',
+            'browser_name': 'Firefox',
+        }
+
+        text_body = email_template.get_content(
+            'password_reset_request', lang, 'body.txt', email_data
+        )
+
+        assert text_body == expected_text_body
+
+    def test_it_gets_en_html_body(self, app):
+        email_template = EmailTemplate(app.config.get('TEMPLATES_FOLDER'))
+        email_data = {
+            'expiration_delay': '3 seconds',
+            'username': 'test',
+            'password_reset_url': 'http://localhost/password-reset?token=xxx',
+            'operating_system': 'Linux',
+            'browser_name': 'Firefox',
+        }
+
+        text_body = email_template.get_content(
+            'password_reset_request', 'en', 'body.html', email_data
+        )
+
+        assert expected_en_html_body in text_body
+
+    def test_it_gets_fr_html_body(self, app):
+        email_template = EmailTemplate(app.config.get('TEMPLATES_FOLDER'))
+        email_data = {
+            'expiration_delay': '3 secondes',
+            'username': 'test',
+            'password_reset_url': 'http://localhost/password-reset?token=xxx',
+            'operating_system': 'Linux',
+            'browser_name': 'Firefox',
+        }
+
+        text_body = email_template.get_content(
+            'password_reset_request', 'fr', 'body.html', email_data
+        )
+
+        assert expected_fr_html_body in text_body
--- a/fittrackee/tests/email/test_email_utils.py
+++ b/fittrackee/tests/email/test_email_utils.py
@@ -0,0 +1,39 @@
+import pytest
+from fittrackee.email.utils_email import InvalidEmailUrlScheme, parse_email_url
+
+
+class TestEmailUrlParser:
+    def test_it_raises_error_if_url_scheme_is_invalid(self):
+        url = 'stmp://username:password@localhost:587'
+        with pytest.raises(InvalidEmailUrlScheme):
+            parse_email_url(url)
+
+    def test_it_parses_email_url(self):
+        url = 'smtp://test@example.com:12345678@localhost:25'
+        parsed_email = parse_email_url(url)
+        assert parsed_email['username'] == 'test@example.com'
+        assert parsed_email['password'] == '12345678'
+        assert parsed_email['host'] == 'localhost'
+        assert parsed_email['port'] == 25
+        assert parsed_email['use_tls'] is False
+        assert parsed_email['use_ssl'] is False
+
+    def test_it_parses_email_url_with_tls(self):
+        url = 'smtp://test@example.com:12345678@localhost:587?tls=True'
+        parsed_email = parse_email_url(url)
+        assert parsed_email['username'] == 'test@example.com'
+        assert parsed_email['password'] == '12345678'
+        assert parsed_email['host'] == 'localhost'
+        assert parsed_email['port'] == 587
+        assert parsed_email['use_tls'] is True
+        assert parsed_email['use_ssl'] is False
+
+    def test_it_parses_email_url_with_ssl(self):
+        url = 'smtp://test@example.com:12345678@localhost:465?ssl=True'
+        parsed_email = parse_email_url(url)
+        assert parsed_email['username'] == 'test@example.com'
+        assert parsed_email['password'] == '12345678'
+        assert parsed_email['host'] == 'localhost'
+        assert parsed_email['port'] == 465
+        assert parsed_email['use_tls'] is False
+        assert parsed_email['use_ssl'] is True