[1.2.X] Enhanced tests for r14857 added in r15185. Refs #14870. Thanks Harm Geerts.
Backport of [15201] from trunk. git-svn-id: http://code.djangoproject.com/svn/django/branches/releases/1.2.X@15202 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
parent
0ebb02e9dc
commit
88dfbc50fb
|
@ -1698,13 +1698,41 @@ class RouterAttributeErrorTestCase(TestCase):
|
|||
def tearDown(self):
|
||||
router.routers = self.old_routers
|
||||
|
||||
def test_attribute_error(self):
|
||||
def test_attribute_error_read(self):
|
||||
"Check that the AttributeError from AttributeErrorRouter bubbles up"
|
||||
router.routers = [] # Reset routers so we can save a Book instance
|
||||
b = Book.objects.create(title="Pro Django",
|
||||
published=datetime.date(2008, 12, 16))
|
||||
router.routers = [AttributeErrorRouter()] # Install our router
|
||||
self.assertRaises(AttributeError, Book.objects.get, pk=b.pk)
|
||||
|
||||
def test_attribute_error_save(self):
|
||||
"Check that the AttributeError from AttributeErrorRouter bubbles up"
|
||||
dive = Book()
|
||||
dive.title="Dive into Python"
|
||||
dive.published = datetime.date(2009, 5, 4)
|
||||
self.assertRaises(AttributeError, dive.save)
|
||||
|
||||
def test_attribute_error_delete(self):
|
||||
"Check that the AttributeError from AttributeErrorRouter bubbles up"
|
||||
router.routers = [] # Reset routers so we can save our Book, Person instances
|
||||
b = Book.objects.create(title="Pro Django",
|
||||
published=datetime.date(2008, 12, 16))
|
||||
p = Person.objects.create(name="Marty Alchin")
|
||||
b.authors = [p]
|
||||
b.editor = p
|
||||
router.routers = [AttributeErrorRouter()] # Install our router
|
||||
self.assertRaises(AttributeError, b.delete)
|
||||
|
||||
def test_attribute_error_m2m(self):
|
||||
"Check that the AttributeError from AttributeErrorRouter bubbles up"
|
||||
router.routers = [] # Reset routers so we can save our Book, Person instances
|
||||
b = Book.objects.create(title="Pro Django",
|
||||
published=datetime.date(2008, 12, 16))
|
||||
p = Person.objects.create(name="Marty Alchin")
|
||||
router.routers = [AttributeErrorRouter()] # Install our router
|
||||
self.assertRaises(AttributeError, setattr, b, 'authors', [p])
|
||||
|
||||
class ModelMetaRouter(object):
|
||||
"A router to ensure model arguments are real model classes"
|
||||
def db_for_write(self, model, **hints):
|
||||
|
|
Loading…
Reference in New Issue