mirror of
https://github.com/django/django.git
synced 2025-10-23 21:59:11 +00:00
Migrated proxy_models doctests. Thanks to Eric Florenzano.
git-svn-id: http://code.djangoproject.com/svn/django/trunk@13831 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
Russell Keith-Magee
@@ -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
|
||||
Reference in New Issue
Block a user