mirror of
https://github.com/Hopiu/django-model-utils.git
synced 2026-03-17 04:10:24 +00:00
02b2082e90
- remove __getitem__, no need for it anymore - remove implicit lowercasing (principle of least surprise: strings are usually case sensitive)
265 lines
9 KiB
Python
265 lines
9 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
from django.test import TestCase
|
|
from django.conf import settings
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.db.models.fields import FieldDoesNotExist
|
|
|
|
from model_utils import ChoiceEnum, Choices
|
|
from model_utils.fields import get_excerpt
|
|
from model_utils.tests.models import InheritParent, InheritChild, TimeStamp, \
|
|
Post, Article, Status, Status2, TimeFrame
|
|
|
|
|
|
class GetExcerptTests(TestCase):
|
|
def test_split(self):
|
|
e = get_excerpt("some content\n\n<!-- split -->\n\nsome more")
|
|
self.assertEquals(e, 'some content\n')
|
|
|
|
def test_auto_split(self):
|
|
e = get_excerpt("para one\n\npara two\n\npara three")
|
|
self.assertEquals(e, 'para one\n\npara two')
|
|
|
|
def test_middle_of_para(self):
|
|
e = get_excerpt("some text\n<!-- split -->\nmore text")
|
|
self.assertEquals(e, 'some text')
|
|
|
|
def test_middle_of_line(self):
|
|
e = get_excerpt("some text <!-- split --> more text")
|
|
self.assertEquals(e, "some text <!-- split --> more text")
|
|
|
|
class SplitFieldTests(TestCase):
|
|
full_text = u'summary\n\n<!-- split -->\n\nmore'
|
|
excerpt = u'summary\n'
|
|
|
|
def setUp(self):
|
|
self.post = Article.objects.create(
|
|
title='example post', body=self.full_text)
|
|
|
|
def test_unicode_content(self):
|
|
self.assertEquals(unicode(self.post.body), self.full_text)
|
|
|
|
def test_excerpt(self):
|
|
self.assertEquals(self.post.body.excerpt, self.excerpt)
|
|
|
|
def test_content(self):
|
|
self.assertEquals(self.post.body.content, self.full_text)
|
|
|
|
def test_has_more(self):
|
|
self.failUnless(self.post.body.has_more)
|
|
|
|
def test_not_has_more(self):
|
|
post = Article.objects.create(title='example 2',
|
|
body='some text\n\nsome more\n')
|
|
self.failIf(post.body.has_more)
|
|
|
|
def test_load_back(self):
|
|
post = Article.objects.get(pk=self.post.pk)
|
|
self.assertEquals(post.body.content, self.post.body.content)
|
|
self.assertEquals(post.body.excerpt, self.post.body.excerpt)
|
|
|
|
def test_assign_to_body(self):
|
|
new_text = u'different\n\n<!-- split -->\n\nother'
|
|
self.post.body = new_text
|
|
self.post.save()
|
|
self.assertEquals(unicode(self.post.body), new_text)
|
|
|
|
def test_assign_to_content(self):
|
|
new_text = u'different\n\n<!-- split -->\n\nother'
|
|
self.post.body.content = new_text
|
|
self.post.save()
|
|
self.assertEquals(unicode(self.post.body), new_text)
|
|
|
|
def test_assign_to_excerpt(self):
|
|
def _invalid_assignment():
|
|
self.post.body.excerpt = 'this should fail'
|
|
self.assertRaises(AttributeError, _invalid_assignment)
|
|
|
|
|
|
class ChoiceEnumTests(TestCase):
|
|
def setUp(self):
|
|
self.STATUS = ChoiceEnum('DRAFT', 'PUBLISHED')
|
|
|
|
def test_getattr(self):
|
|
self.assertEquals(self.STATUS.DRAFT, 0)
|
|
|
|
def test_getitem(self):
|
|
self.assertEquals(self.STATUS[1], 'PUBLISHED')
|
|
|
|
def test_iteration(self):
|
|
self.assertEquals(tuple(self.STATUS), ((0, 'DRAFT'), (1, 'PUBLISHED')))
|
|
|
|
|
|
class ChoicesTests(TestCase):
|
|
def setUp(self):
|
|
self.STATUS = Choices('DRAFT', 'PUBLISHED')
|
|
|
|
def test_getattr(self):
|
|
self.assertEquals(self.STATUS.DRAFT, 'DRAFT')
|
|
|
|
def test_iteration(self):
|
|
self.assertEquals(tuple(self.STATUS), (('DRAFT', 'DRAFT'), ('PUBLISHED', 'PUBLISHED')))
|
|
|
|
class LabelChoicesTests(ChoicesTests):
|
|
def setUp(self):
|
|
self.STATUS = Choices(
|
|
('DRAFT', 'draft'),
|
|
('PUBLISHED', 'published'),
|
|
'DELETED',
|
|
)
|
|
|
|
def test_iteration(self):
|
|
self.assertEquals(tuple(self.STATUS), (
|
|
('DRAFT', 'draft'),
|
|
('PUBLISHED', 'published'),
|
|
('DELETED', 'DELETED'))
|
|
)
|
|
|
|
def test_display(self):
|
|
self.assertEquals(self.STATUS.DELETED, 'DELETED')
|
|
|
|
|
|
class InheritanceCastModelTests(TestCase):
|
|
def setUp(self):
|
|
self.parent = InheritParent.objects.create()
|
|
self.child = InheritChild.objects.create()
|
|
|
|
def testParentRealType(self):
|
|
self.assertEquals(self.parent.real_type,
|
|
ContentType.objects.get_for_model(InheritParent))
|
|
|
|
def testChildRealType(self):
|
|
self.assertEquals(self.child.real_type,
|
|
ContentType.objects.get_for_model(InheritChild))
|
|
|
|
def testCast(self):
|
|
obj = InheritParent.objects.get(pk=self.child.pk).cast()
|
|
self.assertEquals(obj.__class__, InheritChild)
|
|
|
|
|
|
class TimeStampedModelTests(TestCase):
|
|
def testCreated(self):
|
|
t1 = TimeStamp.objects.create()
|
|
t2 = TimeStamp.objects.create()
|
|
self.assert_(t2.created > t1.created)
|
|
|
|
def testModified(self):
|
|
t1 = TimeStamp.objects.create()
|
|
t2 = TimeStamp.objects.create()
|
|
t1.save()
|
|
self.assert_(t2.modified < t1.modified)
|
|
|
|
|
|
class TimeFramedModelTests(TestCase):
|
|
|
|
def testCreated(self):
|
|
now = datetime.now()
|
|
# objects are out of the timeframe
|
|
TimeFrame.objects.create(start=now+timedelta(days=2))
|
|
TimeFrame.objects.create(end=now-timedelta(days=1))
|
|
self.assertEquals(TimeFrame.timeframed.count(), 0)
|
|
|
|
# objects in the timeframe for various reasons
|
|
TimeFrame.objects.create(start=now-timedelta(days=10))
|
|
TimeFrame.objects.create(end=now+timedelta(days=2))
|
|
TimeFrame.objects.create(start=now-timedelta(days=1), end=now+timedelta(days=1))
|
|
self.assertEquals(TimeFrame.timeframed.count(), 3)
|
|
|
|
|
|
class StatusModelTests(TestCase):
|
|
def setUp(self):
|
|
self.model = Status
|
|
|
|
def testCreated(self):
|
|
c1 = self.model.objects.create()
|
|
c2 = self.model.objects.create()
|
|
self.assert_(c2.status_date > c1.status_date)
|
|
self.assertEquals(self.model.active.count(), 2)
|
|
self.assertEquals(self.model.deleted.count(), 0)
|
|
|
|
def testModification(self):
|
|
t1 = self.model.objects.create()
|
|
date_created = t1.status_date
|
|
t1.status = t1.STATUS.on_hold
|
|
t1.save()
|
|
self.assertEquals(self.model.active.count(), 0)
|
|
self.assertEquals(self.model.on_hold.count(), 1)
|
|
self.assert_(t1.status_date > date_created)
|
|
date_changed = t1.status_date
|
|
t1.save()
|
|
self.assertEquals(t1.status_date, date_changed)
|
|
date_active_again = t1.status_date
|
|
t1.status = t1.STATUS.active
|
|
t1.save()
|
|
self.assert_(t1.status_date > date_active_again)
|
|
|
|
def testPreviousConditon(self):
|
|
status = self.model.objects.create()
|
|
self.assertEquals(status.previous_status, None)
|
|
status.status = status.STATUS.on_hold
|
|
status.save()
|
|
self.assertEquals(status.previous_status, status.STATUS.active)
|
|
|
|
class Status2ModelTests(StatusModelTests):
|
|
def setUp(self):
|
|
self.model = Status2
|
|
|
|
def testModification(self):
|
|
t1 = self.model.objects.create()
|
|
date_created = t1.status_date
|
|
t1.status = t1.STATUS[2][0] # boring on_hold status
|
|
t1.save()
|
|
self.assertEquals(self.model.active.count(), 0)
|
|
self.assertEquals(self.model.on_hold.count(), 1)
|
|
self.assert_(t1.status_date > date_created)
|
|
date_changed = t1.status_date
|
|
t1.save()
|
|
self.assertEquals(t1.status_date, date_changed)
|
|
date_active_again = t1.status_date
|
|
t1.status = t1.STATUS[0][0] # boring active status
|
|
t1.save()
|
|
self.assert_(t1.status_date > date_active_again)
|
|
|
|
def testPreviousConditon(self):
|
|
status = self.model.objects.create()
|
|
self.assertEquals(status.previous_status, None)
|
|
status.status = status.STATUS[2][0]
|
|
status.save()
|
|
self.assertEquals(status.previous_status, status.STATUS[0][0])
|
|
|
|
class QueryManagerTests(TestCase):
|
|
def setUp(self):
|
|
data = ((True, True, 0),
|
|
(True, False, 4),
|
|
(False, False, 2),
|
|
(False, True, 3),
|
|
(True, True, 1),
|
|
(True, False, 5))
|
|
for p, c, o in data:
|
|
Post.objects.create(published=p, confirmed=c, order=o)
|
|
|
|
def testPassingKwargs(self):
|
|
qs = Post.public.all()
|
|
self.assertEquals([p.order for p in qs], [0, 1, 4, 5])
|
|
|
|
def testPassingQ(self):
|
|
qs = Post.public_confirmed.all()
|
|
self.assertEquals([p.order for p in qs], [0, 1])
|
|
|
|
def testOrdering(self):
|
|
qs = Post.public_reversed.all()
|
|
self.assertEquals([p.order for p in qs], [5, 4, 1, 0])
|
|
|
|
if 'south' in settings.INSTALLED_APPS:
|
|
class SouthFreezingTests(TestCase):
|
|
def test_introspector_adds_no_excerpt_field(self):
|
|
from south.modelsinspector import introspector
|
|
mf = Article._meta.get_field('body')
|
|
args, kwargs = introspector(mf)
|
|
self.assertEquals(kwargs['no_excerpt_field'], 'True')
|
|
|
|
def test_no_excerpt_field_works(self):
|
|
from models import NoRendered
|
|
self.assertRaises(FieldDoesNotExist,
|
|
NoRendered._meta.get_field,
|
|
'_body_excerpt')
|