Browse Source

Fix router_honored test to not close connection

Andrew Godwin 10 years ago
parent
commit
8721adcbfb
1 changed files with 5 additions and 2 deletions
  1. 5 2
      tests/commands_sql/tests.py

+ 5 - 2
tests/commands_sql/tests.py

@@ -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):
-            output = sql_command(app_config, no_style(), connections[DEFAULT_DB_ALIAS])
+            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__)