Commit 8721adcb authored by Andrew Godwin's avatar Andrew Godwin
Browse files

Fix router_honored test to not close connection

parent 3f91238a

tests/commands_sql/tests.py

+5 −2
Original line number Diff line number Diff line
@@ -27,7 +27,7 @@ class SQLCommandsTestCase(TestCase): 


    def test_sql_delete(self):

        app_config = apps.get_app_config('commands_sql')

        output = sql_delete(app_config, no_style(), connections[DEFAULT_DB_ALIAS])

        output = sql_delete(app_config, no_style(), connections[DEFAULT_DB_ALIAS], close_connection=False)

        drop_tables = [o for o in output if o.startswith('DROP TABLE')]

        self.assertEqual(len(drop_tables), 3)

        # Lower so that Oracle's upper case tbl names wont break
@@ -71,6 +71,9 @@ class SQLCommandsRouterTestCase(TestCase): 
    def test_router_honored(self):

        app_config = apps.get_app_config('commands_sql')

        for sql_command in (sql_all, sql_create, sql_delete, sql_indexes, sql_destroy_indexes):

            if sql_command is sql_delete:

                output = sql_command(app_config, no_style(), connections[DEFAULT_DB_ALIAS], close_connection=False)

            else:

                output = sql_command(app_config, no_style(), connections[DEFAULT_DB_ALIAS])

            self.assertEqual(len(output), 0,

                "%s command is not honoring routers" % sql_command.__name__)