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

import datetime

from django.test import TestCase
from django.db import models
from django.contrib.auth.models import User

from lostplaces.models import (
    Taggable,
    Mapable,
    Submittable
)
from lostplaces.tests.models import ModelTestCase

from taggit.managers import TaggableManager


class TaggableTestCase(ModelTestCase):

    model = Taggable

    def test_tags(self):
        self.assertField('tags', TaggableManager)


class MapableTestCase(ModelTestCase):

    model = Mapable

    def test_name(self):
        self.assertCharField(
            field_name='name',
            min_length=10,
            max_length=100
        )

    def test_latitude(self):
        self.assertFloatField(
            field_name='latitude',
            min_value=-90,
            max_value=90
        )

    def test_longitude(self):
        self.assertFloatField(
            field_name='longitude',
            min_value=-180,
            max_value=180
        )
        
class SubmittableTestCase(ModelTestCase):
    model = Submittable

    def test_submitted_when(self):
        self.assertField(
            field_name='submitted_when',
            field_class=models.DateTimeField,
            must_have={'auto_now_add': True}
        )

    def test_submitted_by(self):
        submitted_by = self.assertField(
            field_name='submitted_by',
            field_class=models.ForeignKey
        )
        self.assertEqual(
            submitted_by.remote_field.related_name,
            '%(class)s',
            msg='Expecting the related_name of %s to be \'%%(class)s\', got %s' % (
                str(submitted_by),
                submitted_by.remote_field.related_name
            )
        )
        self.assertTrue(
            submitted_by.null,
            msg='Expecting %s to has null=True' % (
                str(submitted_by)
            )
        )
        self.assertTrue(
            submitted_by.blank,
            msg='Expecting %s to has blank=True' % (
                str(submitted_by)
            )    
        )
        self.assertEqual(
            submitted_by.remote_field.on_delete,
            models.SET_NULL,
            msg='Expecting %s to be null when reference is delete (models.SET_NULL)' % (
                str(submitted_by)
            )
        )