schema and migrations test edits

Author: timgraham

tests/migrations/test_base.py

@@ -4,6 +4,8 @@
 from contextlib import contextmanager
 from importlib import import_module
 
+from django_mongodb.fields import ObjectIdAutoField
+
 from django.apps import apps
 from django.db import connection, connections, migrations, models
 from django.db.migrations.migration import Migration
@@ -43,14 +45,16 @@ def assertTableNotExists(self, table, using="default"):
             )
 
     def assertColumnExists(self, table, column, using="default"):
-        self.assertIn(
-            column, [c.name for c in self.get_table_description(table, using=using)]
-        )
+        pass
+        # self.assertIn(
+        #     column, [c.name for c in self.get_table_description(table, using=using)]
+        # )
 
     def assertColumnNotExists(self, table, column, using="default"):
-        self.assertNotIn(
-            column, [c.name for c in self.get_table_description(table, using=using)]
-        )
+        pass
+        # self.assertNotIn(
+        #     column, [c.name for c in self.get_table_description(table, using=using)]
+        # )
 
     def _get_column_allows_null(self, table, column, using):
         return [
@@ -60,10 +64,12 @@ def _get_column_allows_null(self, table, column, using):
         ][0]
 
     def assertColumnNull(self, table, column, using="default"):
-        self.assertTrue(self._get_column_allows_null(table, column, using))
+        pass
+        # self.assertTrue(self._get_column_allows_null(table, column, using))
 
     def assertColumnNotNull(self, table, column, using="default"):
-        self.assertFalse(self._get_column_allows_null(table, column, using))
+        pass
+        # self.assertFalse(self._get_column_allows_null(table, column, using))
 
     def _get_column_collation(self, table, column, using):
         return next(
@@ -223,15 +229,15 @@ def cleanup_test_tables(self):
             frozenset(connection.introspection.table_names())
             - self._initial_table_names
         )
-        with connection.schema_editor() as editor:
-            with connection.constraint_checks_disabled():
-                for table_name in table_names:
-                    editor.execute(
-                        editor.sql_delete_table
-                        % {
-                            "table": editor.quote_name(table_name),
-                        }
-                    )
+        with connection.constraint_checks_disabled():
+            for table_name in table_names:
+                connection.database[table_name].drop()
+                # editor.execute(
+                #     editor.sql_delete_table
+                #     % {
+                #         "table": editor.quote_name(table_name),
+                #     }
+                # )
 
     def apply_operations(self, app_label, project_state, operations, atomic=True):
         migration = Migration("name", app_label)
@@ -289,14 +295,14 @@ def set_up_test_model(
             migrations.CreateModel(
                 "Pony",
                 [
-                    ("id", models.AutoField(primary_key=True)),
+                    ("id", ObjectIdAutoField(primary_key=True)),
                     ("pink", models.IntegerField(default=3)),
                     ("weight", models.FloatField()),
                     ("green", models.IntegerField(null=True)),
                     (
                         "yellow",
                         models.CharField(
-                            blank=True, null=True, db_default="Yellow", max_length=20
+                            blank=True, null=True, default="Yellow", max_length=20
                         ),
                     ),
                 ],
@@ -328,7 +334,7 @@ def set_up_test_model(
                 migrations.CreateModel(
                     "Stable",
                     [
-                        ("id", models.AutoField(primary_key=True)),
+                        ("id", ObjectIdAutoField(primary_key=True)),
                     ],
                 )
             )
@@ -337,7 +343,7 @@ def set_up_test_model(
                 migrations.CreateModel(
                     "Van",
                     [
-                        ("id", models.AutoField(primary_key=True)),
+                        ("id", ObjectIdAutoField(primary_key=True)),
                     ],
                 )
             )
@@ -346,7 +352,7 @@ def set_up_test_model(
                 migrations.CreateModel(
                     "Rider",
                     [
-                        ("id", models.AutoField(primary_key=True)),
+                        ("id", ObjectIdAutoField(primary_key=True)),
                         ("pony", models.ForeignKey("Pony", models.CASCADE)),
                         (
                             "friend",
@@ -393,7 +399,7 @@ def set_up_test_model(
                 migrations.CreateModel(
                     "Food",
                     fields=[
-                        ("id", models.AutoField(primary_key=True)),
+                        ("id", ObjectIdAutoField(primary_key=True)),
                     ],
                     managers=[
                         ("food_qs", FoodQuerySet.as_manager()),

tests/migrations/test_commands.py

@@ -870,10 +870,7 @@ def test_sqlmigrate_forwards(self):
                 "--",
             ],
         )
-        self.assertIn(
-            "create table %s" % connection.ops.quote_name("migrations_author").lower(),
-            lines[3].lower(),
-        )
+        self.assertIn("db.create_collection('migrations_author')", lines[3].lower())
         pos = lines.index("--", 3)
         self.assertEqual(
             lines[pos : pos + 3],
@@ -884,8 +881,7 @@ def test_sqlmigrate_forwards(self):
             ],
         )
         self.assertIn(
-            "create table %s" % connection.ops.quote_name("migrations_tribble").lower(),
-            lines[pos + 3].lower(),
+            "db.create_collection('migrations_tribble')", lines[pos + 3].lower()
         )
         pos = lines.index("--", pos + 3)
         self.assertEqual(
@@ -918,6 +914,7 @@ def test_sqlmigrate_backwards(self):
         call_command("sqlmigrate", "migrations", "0001", stdout=out, backwards=True)
 
         lines = out.getvalue().splitlines()
+
         try:
             if connection.features.can_rollback_ddl:
                 self.assertEqual(lines[0], connection.ops.start_transaction_sql())
@@ -951,10 +948,7 @@ def test_sqlmigrate_backwards(self):
                 ],
             )
             next_pos = lines.index("--", pos + 3)
-            drop_table_sql = (
-                "drop table %s"
-                % connection.ops.quote_name("migrations_tribble").lower()
-            )
+            drop_table_sql = "db.migrations_tribble.drop()"
             for line in lines[pos + 3 : next_pos]:
                 if drop_table_sql in line.lower():
                     break
@@ -969,9 +963,7 @@ def test_sqlmigrate_backwards(self):
                     "--",
                 ],
             )
-            drop_table_sql = (
-                "drop table %s" % connection.ops.quote_name("migrations_author").lower()
-            )
+            drop_table_sql = "db.migrations_author.drop()"
             for line in lines[pos + 3 :]:
                 if drop_table_sql in line.lower():
                     break

tests/migrations/test_executor.py

@@ -466,16 +466,16 @@ def test_detect_soft_applied_add_field_manytomanyfield(self):
 
         # Leave the tables for 0001 except the many-to-many table. That missing
         # table should cause detect_soft_applied() to return False.
-        with connection.schema_editor() as editor:
-            for table in tables[2:]:
-                editor.execute(editor.sql_delete_table % {"table": table})
+        for table in tables[2:]:
+            connection.database[table].drop()
+            # editor.execute(editor.sql_delete_table % {"table": table})
         migration = executor.loader.get_migration("migrations", "0001_initial")
         self.assertIs(executor.detect_soft_applied(None, migration)[0], False)
 
         # Cleanup by removing the remaining tables.
-        with connection.schema_editor() as editor:
-            for table in tables[:2]:
-                editor.execute(editor.sql_delete_table % {"table": table})
+        for table in tables[:2]:
+            connection.database[table].drop()
+            # editor.execute(editor.sql_delete_table % {"table": table})
         for table in tables:
             self.assertTableNotExists(table)
 
@@ -689,11 +689,13 @@ def test_alter_id_type_with_fk(self):
             # Rebuild the graph to reflect the new DB state
             executor.loader.build_graph()
         finally:
+            connection.database["book_app_book"].drop()
+            connection.database["author_app_author"].drop()
             # We can't simply unapply the migrations here because there is no
             # implicit cast from VARCHAR to INT on the database level.
-            with connection.schema_editor() as editor:
-                editor.execute(editor.sql_delete_table % {"table": "book_app_book"})
-                editor.execute(editor.sql_delete_table % {"table": "author_app_author"})
+            # with connection.schema_editor() as editor:
+            #   editor.execute(editor.sql_delete_table % {"table": "book_app_book"})
+            #   editor.execute(editor.sql_delete_table % {"table": "author_app_author"})
             self.assertTableNotExists("author_app_author")
             self.assertTableNotExists("book_app_book")
             executor.migrate([("author_app", None)], fake=True)

tests/migrations/test_migrations_no_changes/0001_initial.py

@@ -1,3 +1,5 @@
+from django_mongodb.fields import ObjectIdAutoField
+
 from django.db import migrations, models
 
 
@@ -6,7 +8,7 @@ class Migration(migrations.Migration):
         migrations.CreateModel(
             "Author",
             [
-                ("id", models.AutoField(primary_key=True)),
+                ("id", ObjectIdAutoField(primary_key=True)),
                 ("name", models.CharField(max_length=255)),
                 ("slug", models.SlugField(null=True)),
                 ("age", models.IntegerField(default=0)),
@@ -16,7 +18,7 @@ class Migration(migrations.Migration):
         migrations.CreateModel(
             "Tribble",
             [
-                ("id", models.AutoField(primary_key=True)),
+                ("id", ObjectIdAutoField(primary_key=True)),
                 ("fluffy", models.BooleanField(default=True)),
             ],
         ),

tests/migrations/test_migrations_no_changes/0002_second.py

@@ -1,3 +1,5 @@
+from django_mongodb.fields import ObjectIdAutoField
+
 from django.db import migrations, models
 
 
@@ -13,7 +15,7 @@ class Migration(migrations.Migration):
         migrations.CreateModel(
             "Book",
             [
-                ("id", models.AutoField(primary_key=True)),
+                ("id", ObjectIdAutoField(primary_key=True)),
                 (
                     "author",
                     models.ForeignKey("migrations.Author", models.SET_NULL, null=True),

tests/migrations/test_migrations_no_changes/0003_third.py

@@ -1,3 +1,5 @@
+from django_mongodb.fields import ObjectIdAutoField
+
 from django.db import migrations, models
 
 
@@ -12,7 +14,7 @@ class Migration(migrations.Migration):
             fields=[
                 (
                     "id",
-                    models.AutoField(
+                    ObjectIdAutoField(
                         verbose_name="ID",
                         serialize=False,
                         auto_created=True,
@@ -28,7 +30,7 @@ class Migration(migrations.Migration):
             fields=[
                 (
                     "id",
-                    models.AutoField(
+                    ObjectIdAutoField(
                         verbose_name="ID",
                         serialize=False,
                         auto_created=True,

tests/migrations/test_migrations_no_default/0001_initial.py

@@ -1,3 +1,5 @@
+from django_mongodb.fields import ObjectIdAutoField
+
 from django.db import migrations, models
 
 
@@ -10,7 +12,7 @@ class Migration(migrations.Migration):
             fields=[
                 (
                     "id",
-                    models.AutoField(
+                    ObjectIdAutoField(
                         verbose_name="ID",
                         serialize=False,
                         auto_created=True,