155 lines
4.4 KiB
Python
155 lines
4.4 KiB
Python
#!/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()
|
|
)
|