lostplaces-backend/lostplaces/lostplaces_app/tests/views/test_base_views.py

import datetime

from django.test import TestCase, RequestFactory, Client
from django.urls import reverse_lazy
from django.contrib.auth.models import User, AnonymousUser
from django.contrib.messages.storage.fallback import FallbackStorage

from lostplaces_app.models import Place
from lostplaces_app.views import IsAuthenticatedMixin

class TestIsAuthenticatedMixin(RedirectTestCase):
        
    @classmethod
    def setUpTestData(cls):
        user = User.objects.create_user(
            username='testpeter',
            password='Develop123'
        )

    def test_logged_in(self):
        request = RequestFactory().get('/')
        request.user = User.objects.get(id=1)

        response = IsAuthenticatedMixin.as_view()(request)
        self.assertEqual(response.status_code, 200)

    def test_not_logged_in(self):
        request = RequestFactory().get('/someurl1234')
        request.user = AnonymousUser()
        request.session = 'session'
        messages = FallbackStorage(request)
        request._messages = messages

        response = IsAuthenticatedMixin.as_view()(request)
        self.assertRedirectsTo(response, '?'.join([str(reverse_lazy('login')), 'next=/someurl1234']))
      
        self.assertTrue(response.context['messages'])
        self.assertTrue(len(response.context['messages']) > 0)

class TestIsPlaceSubmitterMixin(TestCase):

    @classmethod
    def setUpTestData(cls):
        user = User.objects.create_user(
            username='testpeter',
            password='Develop123'
        )
        
        place = Place.objects.create(
            name='Im a place',
            submitted_when=datetime.datetime.now(),
            submitted_by=user.explorer,
            location='Testtown',
            latitude=50.5,
            longitude=7.0,
            description='This is just a test, do not worry'
        )
        place.tags.add('I a tag', 'testlocation')
        place.save()

    def setUp(self):
        self. client = Client()

    def test_is_submitter(self):
        self.client.login(username='testpeter', password='Develop123')
        response = self.client.get(reverse_lazy('place_edit', kwargs={'pk': 1}))
        self.assertEqual(response.status_code, 200)

    def test_is_no_submitter(self):
        User.objects.create_user(
            username='manfred',
            password='Develop123'
        )
        self.client.login(username='manfred', password='Develop123')
        response = self.client.get(reverse_lazy('place_edit', kwargs={'pk': 1}))
        self.assertEqual(response.status_code, 403)
        self.assertTrue(response.context['messages'])
        self.assertTrue(len(response.context['messages']) > 0)