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

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from django.test import TestCase, RequestFactory, Client
from django.urls import reverse_lazy
from django.db import models
from django.contrib.auth.models import User, AnonymousUser
from django.contrib.messages.storage.fallback import FallbackStorage
from django.utils import timezone
from django.utils.translation import gettext as _
from django.shortcuts import render

from lostplaces.models import Place
from lostplaces.views import (
    IsAuthenticatedMixin,
    PlaceAssetCreateView
)
from lostplaces.tests.views import ViewTestCase
from lostplaces.models import DummyAsset

class TestIsAuthenticated(ViewTestCase):
    view = IsAuthenticatedMixin

    @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)
        # Expecting a 405 because IsAuthenticatedMixin has no 'get' method
        self.assertHttpMethodNotAllowed(response)

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

        response = IsAuthenticatedMixin.as_view()(request)
        self.assertHttpRedirect(response, '?'.join([str(reverse_lazy('login')), 'next=/']))
        
        self.assertTrue(len(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=timezone.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 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)
    
class TestPlaceAssetCreateView(ViewTestCase):
    view = PlaceAssetCreateView
        
    class DummyAssetCreateView(PlaceAssetCreateView):
        model = DummyAsset
        fields = ['name']
        template_name = 'flat_form.html'
        
    @classmethod
    def setUpTestData(cls):
        user = User.objects.create_user(
            username='testpeter',
            password='Develop123'
        )
        
        place = Place.objects.create(
            name='Im a place',
            submitted_when=timezone.now(),
            submitted_by=user.explorer,
            location='Testtown',
            latitude=50.5,
            longitude=7.0,
            description='This is just a test, do not worry'
        )
        
    def test_create_asset(self):
        request = RequestFactory().post(
            '/',
            {
                'name': 'im a place'
            }
        )
        request.user = User.objects.get(id=1)
        
        response = TestPlaceAssetCreateView.DummyAssetCreateView.as_view()(request, place_id=1)
        
        self.assertTrue(
            DummyAsset.objects.all().count() >= 1,
            msg='Expecting a dummy place asset to be created by the view'
        )
        
        self.assertEqual(
            DummyAsset.objects.get(id=1).name,
            'im a place',
            msg='Expecting the dummy place asset to have the correct data'
        )