FitTrackee/fittrackee/tests/test_email_template_password_request.py

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