~ubuntu-branches/ubuntu/quantal/python-django/quantal

« back to all changes in this revision

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

  • Committer: Bazaar Package Importer
  • Author(s): Scott James Remnant, Eddy Mulyono
  • Date: 2008-09-16 12:18:47 UTC
  • mfrom: (1.1.5 upstream) (4.1.1 lenny)
  • Revision ID: james.westby@ubuntu.com-20080916121847-mg225rg5mnsdqzr0
Tags: 1.0-1ubuntu1
* Merge from Debian (LP: #264191), remaining changes:
  - Run test suite on build.

[Eddy Mulyono]
* Update patch to workaround network test case failures.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# coding: utf-8
 
2
from django.db import models
 
3
 
 
4
CHOICES = (
 
5
    (1, 'first'),
 
6
    (2, 'second'),
 
7
)
 
8
 
 
9
class Article(models.Model):
 
10
    headline = models.CharField(max_length=100, default='Default headline')
 
11
    pub_date = models.DateTimeField()
 
12
    status = models.IntegerField(blank=True, null=True, choices=CHOICES)
 
13
    misc_data = models.CharField(max_length=100, blank=True)
 
14
    article_text = models.TextField()
 
15
 
 
16
    class Meta:
 
17
        ordering = ('pub_date','headline')
 
18
        # A utf-8 verbose name (Ångström's Articles) to test they are valid.
 
19
        verbose_name = "\xc3\x85ngstr\xc3\xb6m's Articles"
 
20
 
 
21
    def __unicode__(self):
 
22
        return self.headline
 
23
 
 
24
class Movie(models.Model):
 
25
    #5218: Test models with non-default primary keys / AutoFields
 
26
    movie_id = models.AutoField(primary_key=True)
 
27
    name = models.CharField(max_length=60)
 
28
 
 
29
class Party(models.Model):
 
30
    when = models.DateField()
 
31
 
 
32
class Event(models.Model):
 
33
    when = models.DateTimeField()
 
34
 
 
35
class Department(models.Model):
 
36
    id = models.PositiveIntegerField(primary_key=True)
 
37
    name = models.CharField(max_length=200)
 
38
 
 
39
    def __unicode__(self):
 
40
        return self.name
 
41
 
 
42
class Worker(models.Model):
 
43
    department = models.ForeignKey(Department)
 
44
    name = models.CharField(max_length=200)
 
45
 
 
46
    def __unicode__(self):
 
47
        return self.name
 
48
 
 
49
__test__ = {'API_TESTS': """
 
50
(NOTE: Part of the regression test here is merely parsing the model
 
51
declaration. The verbose_name, in particular, did not always work.)
 
52
 
 
53
An empty choice field should return None for the display name.
 
54
 
 
55
>>> from datetime import datetime
 
56
>>> a = Article(headline="Look at me!", pub_date=datetime.now())
 
57
>>> a.save()
 
58
>>> a.get_status_display() is None
 
59
True
 
60
 
 
61
Empty strings should be returned as Unicode
 
62
>>> a2 = Article.objects.get(pk=a.id)
 
63
>>> a2.misc_data
 
64
u''
 
65
 
 
66
# TextFields can hold more than 4000 characters (this was broken in Oracle).
 
67
>>> a3 = Article(headline="Really, really big", pub_date=datetime.now())
 
68
>>> a3.article_text = "ABCDE" * 1000
 
69
>>> a3.save()
 
70
>>> a4 = Article.objects.get(pk=a3.id)
 
71
>>> len(a4.article_text)
 
72
5000
 
73
 
 
74
# Regression test for #659
 
75
>>> import datetime
 
76
>>> p = Party.objects.create(when = datetime.datetime(1999, 12, 31))
 
77
>>> p = Party.objects.create(when = datetime.datetime(1998, 12, 31))
 
78
>>> p = Party.objects.create(when = datetime.datetime(1999, 1, 1))
 
79
>>> [p.when for p in Party.objects.filter(when__month=2)]
 
80
[]
 
81
>>> [p.when for p in Party.objects.filter(when__month=1)]
 
82
[datetime.date(1999, 1, 1)]
 
83
>>> [p.when for p in Party.objects.filter(when__month=12)]
 
84
[datetime.date(1999, 12, 31), datetime.date(1998, 12, 31)]
 
85
>>> [p.when for p in Party.objects.filter(when__year=1998)]
 
86
[datetime.date(1998, 12, 31)]
 
87
 
 
88
# Regression test for #8510
 
89
>>> [p.when for p in Party.objects.filter(when__day='31')]
 
90
[datetime.date(1999, 12, 31), datetime.date(1998, 12, 31)]
 
91
>>> [p.when for p in Party.objects.filter(when__month='12')]
 
92
[datetime.date(1999, 12, 31), datetime.date(1998, 12, 31)]
 
93
>>> [p.when for p in Party.objects.filter(when__year='1998')]
 
94
[datetime.date(1998, 12, 31)]
 
95
 
 
96
# Check that get_next_by_FIELD and get_previous_by_FIELD don't crash when we
 
97
# have usecs values stored on the database
 
98
#
 
99
# [It crashed after the Field.get_db_prep_* refactor, because on most backends
 
100
#  DateTimeFields supports usecs, but DateTimeField.to_python didn't recognize
 
101
#  them. (Note that Model._get_next_or_previous_by_FIELD coerces values to
 
102
#  strings)]
 
103
#
 
104
>>> e = Event.objects.create(when = datetime.datetime(2000, 1, 1, 16, 0, 0))
 
105
>>> e = Event.objects.create(when = datetime.datetime(2000, 1, 1, 6, 1, 1))
 
106
>>> e = Event.objects.create(when = datetime.datetime(2000, 1, 1, 13, 1, 1))
 
107
>>> e = Event.objects.create(when = datetime.datetime(2000, 1, 1, 12, 0, 20, 24))
 
108
>>> e.get_next_by_when().when
 
109
datetime.datetime(2000, 1, 1, 13, 1, 1)
 
110
>>> e.get_previous_by_when().when
 
111
datetime.datetime(2000, 1, 1, 6, 1, 1)
 
112
 
 
113
# Check Department and Worker
 
114
>>> d = Department(id=10, name='IT')
 
115
>>> d.save()
 
116
>>> w = Worker(department=d, name='Full-time')
 
117
>>> w.save()
 
118
>>> w
 
119
<Worker: Full-time>
 
120
 
 
121
"""
 
122
}