Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
Browse files

Enhanced tests for r14857 added in r15185. Refs #14870. Thanks Harm G…

…eerts.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@15201 bcc190cf-cafb-0310-a4f2-bffc1f526a37
  • Loading branch information...
commit 7916c754aa91f83a96b2b8b3c54a2f32bdeca740 1 parent b7ce6b0
@ramiro ramiro authored
Showing with 29 additions and 1 deletion.
  1. +29 −1 tests/regressiontests/multiple_database/tests.py
View
30 tests/regressiontests/multiple_database/tests.py
@@ -1810,13 +1810,41 @@ def setUp(self):
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):
Please sign in to comment.
Something went wrong with that request. Please try again.