pbp_com/apps/catalog/tests/model_tests.py

from decimal import Decimal
from django.contrib.auth.models import User
from django.test import TestCase
from catalog.constants import INSTATE, REGULAR
from catalog.models import Book, Movie, Map, Subscription


class ItemFixture(TestCase):

    @property
    def data(self):
        # basic item data
        return {
            'title':'mytitle',
            'description': 'mydescription',
            'synopsis': 'mysynopsis',
            # todo image
            'price': Decimal(5.00),
            'shipping_cost': Decimal(5.00),
            'shipping_cost_multiple': 1,
            'created_by': self.user,
            'last_updated_by': self.user}

    @property
    def user(self):
        return User.objects.get_or_create(username='test_user')[0]


class BookFixture(ItemFixture):

    @property
    def book_data(self):
        data = self.data
        data.update({
            'authors': 'john',
            'illustrator': 'john',
            'publisher': 'john',
            'printer': 'john',
            'publish_date': '2007',
            'edition': 'first',
            'features': 'my features',
            'dimensions': '46"x25"',
            'pages': 433,
            'isbn': '123456789',
            'library_of_congress_number': '987654321',
            'genre': 'horror'})
        return data


class IsValidBook(BookFixture):

    def setUp(self):
        self.result = Book.objects.get_or_create(defaults=self.book_data)[0]

    def test(self):
        self.assertIsNotNone(self.result)


class MovieFixture(ItemFixture):

    @property
    def movie_data(self):
        data = self.data
        data.update({
            'producer': 'john',
            'music': 'john',
            'runtime_minutes': 120,
            'format': 'john',
            'region': 'US',
            'sound': 'yes',
            'language': 'english',
            'captioning': 'english',
            'genre': 'horror'})
        return data


class IsValidMovie(MovieFixture):

    def setUp(self):
        self.result = Movie.objects.get_or_create(defaults=self.movie_data)

    def test(self):
        self.assertIsNotNone(self.result)


class MapFixture(ItemFixture):

    @property
    def map_data(self):
        data = self.data
        data.update({'dimensions': '46"x25"'})
        return data


class IsValidMap(MapFixture):

    def setUp(self):
        self.result = Map.objects.get_or_create(defaults=self.map_data)

    def test(self):
        self.assertIsNotNone(self.result)


class SubscriptionFixture(ItemFixture):

    @property
    def subscription_data(self):
        data = self.data
        data.update({
            'shipping_type': INSTATE,
            'shipping_method': REGULAR,
            'duration': 9})
        return data

class IsValidSubscription(SubscriptionFixture):

    def setUp(self):
        self.result = Subscription.objects.get_or_create(
            defaults=self.subscription_data)[0]

    def test(self):
        self.assertIsNotNone(self.result)