lostplaces-backend/django_lostplaces/lostplaces/tests/models/test_place_model.py

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

from django.test import TestCase
from django.db import models

from django.contrib.auth.models import User
from django.utils import timezone

from lostplaces.models import Place
from lostplaces.tests.models import ModelTestCase

class PlaceTestCase(ModelTestCase):
    model = Place
    related_name = 'places'
    nullable = True

    @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.place = Place.objects.get(id=1)

    def test_location(self):
        self.assertCharField(
            field_name='location',
            min_length=10,
            max_length=100
        )

    def test_description(self):
        self.assertField('description', models.TextField)

    def test_average_latlon(self):
        '''
        Tests the average latitude/longitude calculation of a list
        of 10 places
        '''
        place_list = []
        for i in range(10):
            place = Place.objects.get(id=1)
            place.id = None
            place.latitude = i+1
            place.longitude = i+10
            place.save()
            place_list.append(place)

        avg_latlon = Place.average_latlon(place_list)
        
        self.assertTrue('latitude' in avg_latlon,
            msg='Expecting avg_latlon dict to have an \'latitude\' key'
        )
        self.assertTrue('longitude' in avg_latlon,
            msg='Expecting avg_latlon dict to have an \'longitude\' key'
        )
        
        self.assertEqual(avg_latlon['latitude'], 5.5,
            msg='%s: average latitude missmatch' % (
                self.model.__name__
            )
        )
        self.assertEqual(avg_latlon['longitude'], 14.5,
            msg='%s: average longitude missmatch' % (
                self.model.__name__
            )
        )

    def test_average_latlon_one_place(self):
        '''
        Tests the average latitude/longitude calculation of a list
        of one place
        '''
        place = Place.objects.get(id=1)
        avg_latlon = Place.average_latlon([place])
        self.assertEqual(avg_latlon['latitude'], place.latitude,
            msg='%s:(one place) average latitude missmatch' % (
                self.model.__name__
            )
        )
        self.assertEqual(avg_latlon['longitude'], place.longitude,
            msg='%s: (one place) average longitude missmatch' % (
                self.model.__name__
            )
        )

    def test_average_latlon_no_places(self):
        '''
        Tests the average latitude/longitude calculation of 
        an empty list
        '''
        avg_latlon = Place.average_latlon([])
        self.assertEqual(avg_latlon['latitude'], 0,
            msg='%s: (no places) average latitude missmatch' % (
                self.model.__name__
            )
        )
        self.assertEqual(avg_latlon['longitude'], 0,
            msg='%s: (no places) average longitude missmatch' % (
                self.model.__name__
            )
        )

    def test_str(self):
        place = self.place
        self.assertTrue(place.name.lower() in str(place).lower(),
            msg='Expecting %s.__str__ to contain the name' % (
                self.model.__name__
            )
        )

    def test_level_calculation(self):
        place = self.place
        explorer = place.submitted_by

        PlaceVoting.objects.create(
            submitted_by=explorer,
            place=place,
            vote=5,
            expires_when=timezone.now()+self.delta
        )
        PlaceVoting.objects.create(
            submitted_by=explorer,
            place=place,
            vote=2,
            expires_when=timezone.now()+self.delta
        )
        PlaceVoting.objects.create(
            submitted_by=explorer,
            place=place,
            vote=4,
            expires_when=timezone.now()+self.delta
        )

        self.assertEqual(
            4,
            place.calculate_place_level()
        )