Added more tests to custom_pk unit-test model

git-svn-id: http://code.djangoproject.com/svn/django/trunk@461 bcc190cf-cafb-0310-a4f2-bffc1f526a37

1 files changed, 24 insertions, 2 deletions

diff --git a/tests/testapp/models/custom_pk.py b/tests/testapp/models/custom_pk.py
index 97750182d7..d7407404f8 100644
--- a/tests/testapp/models/custom_pk.py
+++ b/tests/testapp/models/custom_pk.py
@@ -16,13 +16,35 @@ class Employee(meta.Model):
         meta.CharField('first_name', maxlength=20),
         meta.CharField('last_name', maxlength=20),
     )
+    ordering = ('last_name', 'first_name')
 
     def __repr__(self):
         return "%s %s" % (self.first_name, self.last_name)
 
 API_TESTS = """
->>> e = employees.Employee(employee_code='ABC123', first_name='Dan', last_name='Jones')
->>> e.save()
+>>> dan = employees.Employee(employee_code='ABC123', first_name='Dan', last_name='Jones')
+>>> dan.save()
 >>> employees.get_list()
 [Dan Jones]
+
+>>> fran = employees.Employee(employee_code='XYZ456', first_name='Fran', last_name='Bones')
+>>> fran.save()
+>>> employees.get_list()
+[Fran Bones, Dan Jones]
+
+>>> employees.get_object(pk='ABC123')
+Dan Jones
+>>> employees.get_object(pk='XYZ456')
+Fran Bones
+>>> employees.get_object(pk='foo')
+Traceback (most recent call last):
+    ...
+EmployeeDoesNotExist: Employee does not exist for {'pk': 'foo'}
+
+# Fran got married and changed her last name.
+>>> fran = employees.get_object(pk='XYZ456')
+>>> fran.last_name = 'Jones'
+>>> fran.save()
+>>> employees.get_list(last_name__exact='Jones')
+[Dan Jones, Fran Jones]
 """