fix issue #9

Author: michiya

django_pyodbc/creation.py

@@ -52,10 +52,20 @@ class DatabaseCreation(BaseDatabaseCreation):
     def _destroy_test_db(self, test_database_name, verbosity):
         "Internal implementation - remove the test db tables."
         cursor = self.connection.cursor()
-        self.set_autocommit()
+        self.connection.connection.autocommit = True
         #time.sleep(1) # To avoid "database is being accessed by other users" errors.
         cursor.execute("ALTER DATABASE %s SET SINGLE_USER WITH ROLLBACK IMMEDIATE " % \
                 self.connection.ops.quote_name(test_database_name))
         cursor.execute("DROP DATABASE %s" % \
                 self.connection.ops.quote_name(test_database_name))
         self.connection.close()
+
+    def _prepare_for_test_db_ddl(self):
+        self.connection.connection.rollback()
+        self.connection.connection.autocommit = True
+
+    def sql_table_creation_suffix(self):
+        suffix = []
+        if self.connection.settings_dict['TEST_COLLATION']:
+            suffix.append('COLLATE %s' % self.connection.settings_dict['TEST_COLLATION'])
+        return ' '.join(suffix)

django_pyodbc/operations.py

@@ -179,8 +179,8 @@ def sql_flush(self, style, tables, sequences):
         if tables:
             # Cannot use TRUNCATE on tables that are referenced by a FOREIGN KEY
             # So must use the much slower DELETE
-            from django.db import connection
-            cursor = connection.cursor()
+            from django.db import connections
+            cursor = connections[self.connection.alias].cursor()
             # Try to minimize the risks of the braindeaded inconsistency in
             # DBCC CHEKIDENT(table, RESEED, n) behavior.
             seqs = []