Misago/misago/legal/tests/test_required_agreement.py

from django.urls import reverse

from misago.legal.models import Agreement
from misago.users.testutils import AuthenticatedUserTestCase


class RequiredAgreementTests(AuthenticatedUserTestCase):
    def setUp(self):
        super(RequiredAgreementTests, self).setUp()

        self.test_link = reverse('misago:index')

        Agreement.objects.invalidate_cache()

    def tearDown(self):
        Agreement.objects.invalidate_cache()

    def test_tos_link(self):
        Agreement.objects.create(
            type=Agreement.TYPE_TOS,
            link='https://test-agreement.com',
            is_active=True,
        )

        response = self.client.get(self.test_link)
        self.assertEqual(response.status_code, 200)

    def test_tos_text(self):
        Agreement.objects.create(
            type=Agreement.TYPE_TOS,
            text='Lorem ipsum',
            is_active=True,
        )

        response = self.client.get(self.test_link)
        self.assertEqual(response.status_code, 200)

    def test_tos_text_and_link(self):
        Agreement.objects.create(
            type=Agreement.TYPE_TOS,
            link='https://test-agreement.com',
            text='Lorem ipsum',
            is_active=True,
        )

        response = self.client.get(self.test_link)
        self.assertEqual(response.status_code, 200)

    def test_privacy_link(self):
        Agreement.objects.create(
            type=Agreement.TYPE_PRIVACY,
            link='https://test-agreement.com',
            is_active=True,
        )

        response = self.client.get(self.test_link)
        self.assertEqual(response.status_code, 200)

    def test_privacy_text(self):
        Agreement.objects.create(
            type=Agreement.TYPE_PRIVACY,
            text='Lorem ipsum',
            is_active=True,
        )

        response = self.client.get(self.test_link)
        self.assertEqual(response.status_code, 200)

    def test_privacy_text_and_link(self):
        Agreement.objects.create(
            type=Agreement.TYPE_PRIVACY,
            link='https://test-agreement.com',
            text='Lorem ipsum',
            is_active=True,
        )

        response = self.client.get(self.test_link)
        self.assertEqual(response.status_code, 200)

    def test_both(self):
        Agreement.objects.create(
            type=Agreement.TYPE_TOS,
            link='https://test-agreement.com',
            text='Lorem ipsum',
            is_active=True,
        )

        Agreement.objects.create(
            type=Agreement.TYPE_PRIVACY,
            link='https://test-agreement.com',
            text='Lorem ipsum',
            is_active=True,
        )

        response = self.client.get(self.test_link)
        self.assertEqual(response.status_code, 200)