Permalink
Browse files

Migrated proxy_models doctests. Thanks to Eric Florenzano.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@13831 bcc190cf-cafb-0310-a4f2-bffc1f526a37
  • Loading branch information...
1 parent b7cd168 commit 17c097b166ea3d16d385eed5c29ae7e2844a2f09 @freakboy3742 freakboy3742 committed Sep 13, 2010
Showing with 315 additions and 229 deletions.
  1. +1 −229 tests/modeltests/proxy_models/models.py
  2. +314 −0 tests/modeltests/proxy_models/tests.py
@@ -161,232 +161,4 @@ class Improvement(Issue):
class ProxyImprovement(Improvement):
class Meta:
- proxy = True
-
-__test__ = {'API_TESTS' : """
-# The MyPerson model should be generating the same database queries as the
-# Person model (when the same manager is used in each case).
->>> from django.db import DEFAULT_DB_ALIAS
->>> MyPerson.other.all().query.get_compiler(DEFAULT_DB_ALIAS).as_sql() == Person.objects.order_by("name").query.get_compiler(DEFAULT_DB_ALIAS).as_sql()
-True
-
-# The StatusPerson models should have its own table (it's using ORM-level
-# inheritance).
->>> StatusPerson.objects.all().query.get_compiler(DEFAULT_DB_ALIAS).as_sql() == Person.objects.all().query.get_compiler(DEFAULT_DB_ALIAS).as_sql()
-False
-
-# Creating a Person makes them accessible through the MyPerson proxy.
->>> _ = Person.objects.create(name="Foo McBar")
->>> len(Person.objects.all())
-1
->>> len(MyPerson.objects.all())
-1
->>> MyPerson.objects.get(name="Foo McBar").id
-1
->>> MyPerson.objects.get(id=1).has_special_name()
-False
-
-# Person is not proxied by StatusPerson subclass, however.
->>> StatusPerson.objects.all()
-[]
-
-# A new MyPerson also shows up as a standard Person
->>> _ = MyPerson.objects.create(name="Bazza del Frob")
->>> len(MyPerson.objects.all())
-2
->>> len(Person.objects.all())
-2
-
->>> _ = LowerStatusPerson.objects.create(status="low", name="homer")
->>> LowerStatusPerson.objects.all()
-[<LowerStatusPerson: homer>]
-
-# Correct type when querying a proxy of proxy
-
->>> MyPersonProxy.objects.all()
-[<MyPersonProxy: Bazza del Frob>, <MyPersonProxy: Foo McBar>, <MyPersonProxy: homer>]
-
-# Proxy models are included in the ancestors for a model's DoesNotExist and MultipleObjectsReturned
->>> try:
-... MyPersonProxy.objects.get(name='Zathras')
-... except Person.DoesNotExist:
-... pass
->>> try:
-... MyPersonProxy.objects.get(id__lt=10)
-... except Person.MultipleObjectsReturned:
-... pass
->>> try:
-... StatusPerson.objects.get(name='Zathras')
-... except Person.DoesNotExist:
-... pass
->>> sp1 = StatusPerson.objects.create(name='Bazza Jr.')
->>> sp2 = StatusPerson.objects.create(name='Foo Jr.')
->>> try:
-... StatusPerson.objects.get(id__lt=10)
-... except Person.MultipleObjectsReturned:
-... pass
-
-# And now for some things that shouldn't work...
-#
-# All base classes must be non-abstract
->>> class NoAbstract(Abstract):
-... class Meta:
-... proxy = True
-Traceback (most recent call last):
- ....
-TypeError: Abstract base class containing model fields not permitted for proxy model 'NoAbstract'.
-
-# The proxy must actually have one concrete base class
->>> class TooManyBases(Person, Abstract):
-... class Meta:
-... proxy = True
-Traceback (most recent call last):
- ....
-TypeError: Abstract base class containing model fields not permitted for proxy model 'TooManyBases'.
-
->>> class NoBaseClasses(models.Model):
-... class Meta:
-... proxy = True
-Traceback (most recent call last):
- ....
-TypeError: Proxy model 'NoBaseClasses' has no non-abstract model base class.
-
-
-# A proxy cannot introduce any new fields
->>> class NoNewFields(Person):
-... newfield = models.BooleanField()
-... class Meta:
-... proxy = True
-Traceback (most recent call last):
- ....
-FieldError: Proxy model 'NoNewFields' contains model fields.
-
-# Manager tests.
-
->>> Person.objects.all().delete()
->>> _ = Person.objects.create(name="fred")
->>> _ = Person.objects.create(name="wilma")
->>> _ = Person.objects.create(name="barney")
-
->>> MyPerson.objects.all()
-[<MyPerson: barney>, <MyPerson: fred>]
->>> MyPerson._default_manager.all()
-[<MyPerson: barney>, <MyPerson: fred>]
-
->>> OtherPerson.objects.all()
-[<OtherPerson: barney>, <OtherPerson: wilma>]
->>> OtherPerson.excluder.all()
-[<OtherPerson: barney>, <OtherPerson: fred>]
->>> OtherPerson._default_manager.all()
-[<OtherPerson: barney>, <OtherPerson: wilma>]
-
-# Test save signals for proxy models
->>> from django.db.models import signals
->>> def make_handler(model, event):
-... def _handler(*args, **kwargs):
-... print u"%s %s save" % (model, event)
-... return _handler
->>> h1 = make_handler('MyPerson', 'pre')
->>> h2 = make_handler('MyPerson', 'post')
->>> h3 = make_handler('Person', 'pre')
->>> h4 = make_handler('Person', 'post')
->>> signals.pre_save.connect(h1, sender=MyPerson)
->>> signals.post_save.connect(h2, sender=MyPerson)
->>> signals.pre_save.connect(h3, sender=Person)
->>> signals.post_save.connect(h4, sender=Person)
->>> dino = MyPerson.objects.create(name=u"dino")
-MyPerson pre save
-MyPerson post save
-
-# Test save signals for proxy proxy models
->>> h5 = make_handler('MyPersonProxy', 'pre')
->>> h6 = make_handler('MyPersonProxy', 'post')
->>> signals.pre_save.connect(h5, sender=MyPersonProxy)
->>> signals.post_save.connect(h6, sender=MyPersonProxy)
->>> dino = MyPersonProxy.objects.create(name=u"pebbles")
-MyPersonProxy pre save
-MyPersonProxy post save
-
->>> signals.pre_save.disconnect(h1, sender=MyPerson)
->>> signals.post_save.disconnect(h2, sender=MyPerson)
->>> signals.pre_save.disconnect(h3, sender=Person)
->>> signals.post_save.disconnect(h4, sender=Person)
->>> signals.pre_save.disconnect(h5, sender=MyPersonProxy)
->>> signals.post_save.disconnect(h6, sender=MyPersonProxy)
-
-# A proxy has the same content type as the model it is proxying for (at the
-# storage level, it is meant to be essentially indistinguishable).
->>> ctype = ContentType.objects.get_for_model
->>> ctype(Person) is ctype(OtherPerson)
-True
-
->>> MyPersonProxy.objects.all()
-[<MyPersonProxy: barney>, <MyPersonProxy: dino>, <MyPersonProxy: fred>, <MyPersonProxy: pebbles>]
-
->>> u = User.objects.create(name='Bruce')
->>> User.objects.all()
-[<User: Bruce>]
->>> UserProxy.objects.all()
-[<UserProxy: Bruce>]
->>> UserProxyProxy.objects.all()
-[<UserProxyProxy: Bruce>]
-
-# Proxy objects can be deleted
->>> u2 = UserProxy.objects.create(name='George')
->>> UserProxy.objects.all()
-[<UserProxy: Bruce>, <UserProxy: George>]
->>> u2.delete()
->>> UserProxy.objects.all()
-[<UserProxy: Bruce>]
-
-
-# We can still use `select_related()` to include related models in our querysets.
->>> country = Country.objects.create(name='Australia')
->>> state = State.objects.create(name='New South Wales', country=country)
-
->>> State.objects.select_related()
-[<State: New South Wales>]
->>> StateProxy.objects.select_related()
-[<StateProxy: New South Wales>]
->>> StateProxy.objects.get(name='New South Wales')
-<StateProxy: New South Wales>
->>> StateProxy.objects.select_related().get(name='New South Wales')
-<StateProxy: New South Wales>
-
->>> contributor = TrackerUser.objects.create(name='Contributor',status='contrib')
->>> someone = BaseUser.objects.create(name='Someone')
->>> _ = Bug.objects.create(summary='fix this', version='1.1beta',
-... assignee=contributor, reporter=someone)
->>> pcontributor = ProxyTrackerUser.objects.create(name='OtherContributor',
-... status='proxy')
->>> _ = Improvement.objects.create(summary='improve that', version='1.1beta',
-... assignee=contributor, reporter=pcontributor,
-... associated_bug=ProxyProxyBug.objects.all()[0])
-
-# Related field filter on proxy
->>> ProxyBug.objects.get(version__icontains='beta')
-<ProxyBug: ProxyBug:fix this>
-
-# Select related + filter on proxy
->>> ProxyBug.objects.select_related().get(version__icontains='beta')
-<ProxyBug: ProxyBug:fix this>
-
-# Proxy of proxy, select_related + filter
->>> ProxyProxyBug.objects.select_related().get(version__icontains='beta')
-<ProxyProxyBug: ProxyProxyBug:fix this>
-
-# Select related + filter on a related proxy field
->>> ProxyImprovement.objects.select_related().get(reporter__name__icontains='butor')
-<ProxyImprovement: ProxyImprovement:improve that>
-
-# Select related + filter on a related proxy of proxy field
->>> ProxyImprovement.objects.select_related().get(associated_bug__summary__icontains='fix')
-<ProxyImprovement: ProxyImprovement:improve that>
-
-Proxy models can be loaded from fixtures (Regression for #11194)
->>> from django.core import management
->>> management.call_command('loaddata', 'mypeople.json', verbosity=0)
->>> MyPerson.objects.get(pk=100)
-<MyPerson: Elvis Presley>
-
-"""}
+ proxy = True
Oops, something went wrong.

0 comments on commit 17c097b

Please sign in to comment.