view email_list/tests/test_models.py @ 131:9c74cd8668eb

For Django 1.6: take advantage of new database connection persistence.
author Brian Neal <bgneal@gmail.com>
date Tue, 24 Dec 2013 16:50:13 -0600
parents 3062c547bb90
children 1b5b9be17764
line wrap: on
line source
"""
Model tests for the email_list application.

"""
import datetime

from django.test import TestCase

from email_list.models import Subscriber


class SubscriberTestCase(TestCase):

    def test_auto_save(self):

        sub = Subscriber(name='', location='', email='test@example.com')
        sub.save()

        now = datetime.datetime.now()
        self.assertTrue(now - sub.status_date < datetime.timedelta(seconds=2))

        self.assertTrue(sub.status == 'A')
        self.assertTrue(sub.is_active())
        self.failIf(sub.is_pending())
        self.failIf(sub.is_leaving())

    def test_set_pending(self):

        sub = Subscriber(name='', location='', email='test@example.com')
        sub.set_pending()

        now = datetime.datetime.now()
        self.assertTrue(now - sub.status_date < datetime.timedelta(seconds=2))

        self.assertTrue(sub.status == 'P')
        self.failIf(sub.is_active())
        self.assertTrue(sub.is_pending())
        self.failIf(sub.is_leaving())

        self.assertTrue(len(sub.key) == sub.key_length)

    def test_set_active(self):

        sub = Subscriber(name='', location='', email='test@example.com')
        sub.set_active()

        now = datetime.datetime.now()
        self.assertTrue(now - sub.status_date < datetime.timedelta(seconds=2))

        self.assertTrue(sub.status == 'A')
        self.assertTrue(sub.is_active())
        self.failIf(sub.is_pending())
        self.failIf(sub.is_leaving())

    def test_set_leaving(self):

        sub = Subscriber(name='', location='', email='test@example.com')
        sub.set_leaving()

        now = datetime.datetime.now()
        self.assertTrue(now - sub.status_date < datetime.timedelta(seconds=2))

        self.assertTrue(sub.status == 'L')
        self.failIf(sub.is_active())
        self.failIf(sub.is_pending())
        self.assertTrue(sub.is_leaving())

        self.assertTrue(len(sub.key) == sub.key_length)

    def test_gen_key(self):

        sub = Subscriber(name='', location='', email='test@example.com')
        sub.status_date = datetime.datetime.now()
        sub.gen_key()
        self.assertTrue(len(sub.key) == sub.key_length)