#!/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' )