Misago/misago/legal/tests/test_api.py

import json

from django.urls import reverse

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


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

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

        self.api_link = reverse(
            'misago:api:submit-agreement', kwargs={'pk': self.agreement.pk})

    def post_json(self, data):
        return self.client.post(
            self.api_link, json.dumps(data), content_type='application/json')

    def test_anonymous(self):
        self.logout_user()

        response = self.client.post(self.api_link)
        self.assertEqual(response.status_code, 403)
        self.assertEqual(response.json(), {
            'detail': "This action is not available to guests.",
        })

    def test_get_request(self):
        response = self.client.get(self.api_link)
        self.assertEqual(response.status_code, 405)
        self.assertEqual(response.json(), {
            'detail': 'Method "GET" not allowed.',
        })

    def test_invalid_agreement_id(self):
        api_link = reverse(
            'misago:api:submit-agreement', kwargs={'pk': self.agreement.pk + 1})

        response = self.client.post(api_link)
        self.assertEqual(response.status_code, 404)
        self.assertEqual(response.json(), {
            'detail': "Not found.",
        })

    def test_agreement_already_accepted(self):
        self.user.agreements.append(self.agreement.id)
        self.user.save()

        response = self.client.post(self.api_link)
        self.assertEqual(response.status_code, 403)
        self.assertEqual(response.json(), {
            'detail': "You have already accepted this agreement.",
        })

    def test_no_accept_sent(self):
        response = self.client.post(self.api_link)
        self.assertEqual(response.status_code, 403)
        self.assertEqual(response.json(), {
            'detail': "You need to submit a valid choice.",
        })

    def test_invalid_accept_sent(self):
        response = self.post_json({'accept': 1})
        self.assertEqual(response.status_code, 403)
        self.assertEqual(response.json(), {
            'detail': "You need to submit a valid choice.",
        })

    def test_accept_false(self):
        response = self.post_json({'accept': False})
        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.json(), {'detail': 'ok'})

        self.user.refresh_from_db()
        self.assertTrue(self.user.is_deleting_account)
        self.assertFalse(self.user.is_active)

    def test_accept_false_staff(self):
        self.user.is_staff = True
        self.user.save()

        response = self.post_json({'accept': False})
        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.json(), {'detail': 'ok'})

        self.user.refresh_from_db()
        self.assertFalse(self.user.is_deleting_account)
        self.assertTrue(self.user.is_active)

    def test_accept_true(self):
        response = self.post_json({'accept': True})
        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.json(), {'detail': 'ok'})

        self.user.refresh_from_db()
        self.assertEqual(self.user.agreements, [self.agreement.id])
        self.assertFalse(self.user.is_deleting_account)
        self.assertTrue(self.user.is_active)