Commit 88dfbc50 authored by Ramiro Morales's avatar Ramiro Morales
Browse files

[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
parent 0ebb02e9

tests/regressiontests/multiple_database/tests.py

+29 −1
Original line number Diff line number Diff line
@@ -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):