← Back to branch summary

~ubuntu-branches/ubuntu/saucy/python-django/saucy-updates

« back to all changes in this revision

Viewing changes to tests/regressiontests/reverse_single_related/models.py

  • Committer: Bazaar Package Importer
  • Author(s): Scott James Remnant
  • Date: 2008-11-15 19:15:33 UTC
  • mto: This revision was merged to the branch mainline in revision 17.
  • Revision ID: james.westby@ubuntu.com-20081115191533-84v2zyjbmp1074ni
Tags: upstream-1.0.1
Import upstream version 1.0.1

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/regressiontests/reverse_single_related/models.py
1
 
"""
2
 
Regression tests for an object that cannot access a single related object due
3
 
to a restrictive default manager.
4
 
"""
5
 
 
6
 
from django.db import models
7
 
 
8
 
 
9
 
class SourceManager(models.Manager):
10
 
    def get_query_set(self):
11
 
        return super(SourceManager, self).get_query_set().filter(is_public=True)
12
 
 
13
 
class Source(models.Model):
14
 
    is_public = models.BooleanField()
15
 
    objects = SourceManager()
16
 
 
17
 
class Item(models.Model):
18
 
    source = models.ForeignKey(Source)
19
 
 
20
 
 
21
 
__test__ = {'API_TESTS':"""
22
 
 
23
 
>>> public_source = Source.objects.create(is_public=True)
24
 
>>> public_item = Item.objects.create(source=public_source)
25
 
 
26
 
>>> private_source = Source.objects.create(is_public=False)
27
 
>>> private_item = Item.objects.create(source=private_source)
28
 
 
29
 
# Only one source is available via all() due to the custom default manager.
30
 
 
31
 
>>> Source.objects.all()
32
 
[<Source: Source object>]
33
 
 
34
 
>>> public_item.source
35
 
<Source: Source object>
36
 
 
37
 
# Make sure that an item can still access its related source even if the default
38
 
# manager doesn't normally allow it.
39
 
 
40
 
>>> private_item.source
41
 
<Source: Source object>
42
 
 
43
 
# If the manager is marked "use_for_related_fields", it'll get used instead
44
 
# of the "bare" queryset. Usually you'd define this as a property on the class,
45
 
# but this approximates that in a way that's easier in tests.
46
 
 
47
 
>>> Source.objects.use_for_related_fields = True
48
 
>>> private_item = Item.objects.get(pk=private_item.pk)
49
 
>>> private_item.source
50
 
Traceback (most recent call last):
51
 
    ...
52
 
DoesNotExist: Source matching query does not exist.
53
 
 
54
 
"""}