168 lines
9.3 KiB
Python
168 lines
9.3 KiB
Python
import unittest
|
|
|
|
from django_migration_linter.sql_analyser import analyse_sql_statements
|
|
|
|
|
|
class SqlAnalyserTestCase(unittest.TestCase):
|
|
database_vendor = "default"
|
|
|
|
def analyse_sql(self, sql):
|
|
if isinstance(sql, str):
|
|
sql = sql.splitlines()
|
|
return analyse_sql_statements(
|
|
sql_statements=sql, database_vendor=self.database_vendor
|
|
)
|
|
|
|
def assertValidSql(self, sql):
|
|
errors, _ = self.analyse_sql(sql)
|
|
self.assertEqual(0, len(errors), "Found errors in sql: {}".format(errors))
|
|
|
|
def assertBackwardIncompatibleSql(self, sql):
|
|
errors, _ = self.analyse_sql(sql)
|
|
self.assertNotEqual(0, len(errors), "Found no errors in sql")
|
|
|
|
|
|
class MySqlAnalyserTestCase(SqlAnalyserTestCase):
|
|
database_vendor = "mysql"
|
|
|
|
def test_alter_column(self):
|
|
sql = "ALTER TABLE `app_alter_column_a` MODIFY `field` varchar(10) NULL;"
|
|
self.assertValidSql(sql)
|
|
|
|
def test_drop_not_null(self):
|
|
sql = "ALTER TABLE `app_alter_column_drop_not_null_a` MODIFY `not_null_field` integer NULL;"
|
|
self.assertValidSql(sql)
|
|
|
|
def test_add_not_null(self):
|
|
sql = [
|
|
"ALTER TABLE `app_add_not_null_column_a` ADD COLUMN `new_not_null_field` integer DEFAULT 1 NOT NULL;",
|
|
"ALTER TABLE `app_add_not_null_column_a` ALTER COLUMN `new_not_null_field` DROP DEFAULT;",
|
|
]
|
|
self.assertBackwardIncompatibleSql(sql)
|
|
|
|
def test_add_not_null_followed_by_default(self):
|
|
sql = [
|
|
"ALTER TABLE `app_add_not_null_column_followed_by_default_a` ADD COLUMN `new_not_null_field` integer DEFAULT 1 NOT NULL;",
|
|
"ALTER TABLE `app_add_not_null_column_followed_by_default_a` ALTER COLUMN `new_not_null_field` DROP DEFAULT;",
|
|
"ALTER TABLE `app_add_not_null_column_followed_by_default_a` ALTER COLUMN `new_not_null_field` SET DEFAULT '1';",
|
|
]
|
|
self.assertValidSql(sql)
|
|
|
|
def test_unique_together(self):
|
|
sql = "ALTER TABLE `app_unique_together_a` ADD CONSTRAINT `app_unique_together_a_int_field_char_field_979ac7d8_uniq` UNIQUE (`int_field`, `char_field`);"
|
|
self.assertBackwardIncompatibleSql(sql)
|
|
|
|
sql = "ALTER TABLE `app_unique_together_a` DROP INDEX `app_unique_together_a_int_field_char_field_979ac7d8_uniq`;"
|
|
self.assertValidSql(sql)
|
|
|
|
def test_add_many_to_many_field(self):
|
|
sql = [
|
|
"CREATE TABLE `app_add_manytomany_field_b_many_to_many`(`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, `b_id` integer NOT NULL, `a_id` integer NOT NULL);",
|
|
"ALTER TABLE `app_add_manytomany_field_b_many_to_many` ADD CONSTRAINT `app_add_manytomany_f_b_id_953b185b_fk_app_add_m` FOREIGN KEY(`b_id`) REFERENCES `app_add_manytomany_field_b`(`id`);",
|
|
"ALTER TABLE `app_add_manytomany_field_b_many_to_many` ADD CONSTRAINT `app_add_manytomany_f_a_id_4b44832a_fk_app_add_m` FOREIGN KEY(`a_id`) REFERENCES `app_add_manytomany_field_a`(`id`);",
|
|
"ALTER TABLE `app_add_manytomany_field_b_many_to_many` ADD CONSTRAINT `app_add_manytomany_field_b_many_to_many_b_id_a_id_3e15251d_uniq` UNIQUE(`b_id`, `a_id`);",
|
|
]
|
|
self.assertValidSql(sql)
|
|
|
|
|
|
class SqliteAnalyserTestCase(SqlAnalyserTestCase):
|
|
database_vendor = "sqlite"
|
|
|
|
def test_drop_not_null(self):
|
|
sql = [
|
|
'ALTER TABLE "app_alter_column_drop_not_null_a" RENAME TO "app_alter_column_drop_not_null_a__old";',
|
|
'CREATE TABLE "app_alter_column_drop_not_null_a" ("id" integer NOT NULL PRIMARY KEY AUTOINCREMENT, "not_null_field" integer NULL);',
|
|
'INSERT INTO "app_alter_column_drop_not_null_a" ("id", "not_null_field") SELECT "id", "not_null_field" FROM "app_alter_column_drop_not_null_a__old";',
|
|
'DROP TABLE "app_alter_column_drop_not_null_a__old";',
|
|
]
|
|
self.assertValidSql(sql)
|
|
|
|
def test_add_not_null(self):
|
|
sql = [
|
|
'ALTER TABLE "app_add_not_null_column_a" RENAME TO "app_add_not_null_column_a__old";',
|
|
'CREATE TABLE "app_add_not_null_column_a" ("id" integer NOT NULL PRIMARY KEY AUTOINCREMENT, "new_not_null_field" integer NOT NULL, "null_field" integer NULL);',
|
|
'INSERT INTO "app_add_not_null_column_a" ("id", "null_field", "new_not_null_field") SELECT "id", "null_field", 1 FROM "app_add_not_null_column_a__old";',
|
|
'DROP TABLE "app_add_not_null_column_a__old";',
|
|
]
|
|
self.assertBackwardIncompatibleSql(sql)
|
|
|
|
def test_create_table_with_not_null(self):
|
|
sql = 'CREATE TABLE "app_create_table_with_not_null_column_a" ("id" integer NOT NULL PRIMARY KEY AUTOINCREMENT, "field" varchar(150) NOT NULL);'
|
|
self.assertValidSql(sql)
|
|
|
|
def test_rename_table(self):
|
|
sql = 'ALTER TABLE "app_rename_table_a" RENAME TO "app_rename_table_b";'
|
|
self.assertBackwardIncompatibleSql(sql)
|
|
|
|
def test_alter_column(self):
|
|
sql = [
|
|
'ALTER TABLE "app_alter_column_a" RENAME TO "app_alter_column_a__old";',
|
|
'CREATE TABLE "app_alter_column_a" ("id" integer NOT NULL PRIMARY KEY AUTOINCREMENT, "field" varchar(10) NULL);',
|
|
'INSERT INTO "app_alter_column_a" ("id", "field") SELECT "id", "field" FROM "app_alter_column_a__old";',
|
|
'DROP TABLE "app_alter_column_a__old";',
|
|
]
|
|
self.assertValidSql(sql)
|
|
|
|
def test_alter_column_after_django22(self):
|
|
sql = [
|
|
'CREATE TABLE "new__app_alter_column_a" ("id" integer NOT NULL PRIMARY KEY AUTOINCREMENT, "field" varchar(10) NULL);',
|
|
'INSERT INTO "new__app_alter_column_a" ("id", "field") SELECT "id", "field" FROM "app_alter_column_a";',
|
|
'DROP TABLE "app_alter_column_a";',
|
|
'ALTER TABLE "new__app_alter_column_a" RENAME TO "app_alter_column_a";',
|
|
]
|
|
self.assertValidSql(sql)
|
|
|
|
def test_unique_together(self):
|
|
sql = 'CREATE UNIQUE INDEX "app_unique_together_a_int_field_char_field_979ac7d8_uniq" ON "app_unique_together_a" ("int_field", "char_field");'
|
|
self.assertBackwardIncompatibleSql(sql)
|
|
|
|
sql = 'DROP INDEX "app_unique_together_a_int_field_char_field_979ac7d8_uniq";'
|
|
self.assertValidSql(sql)
|
|
|
|
def test_add_many_to_many_field(self):
|
|
sql = [
|
|
'CREATE TABLE "app_add_manytomany_field_b_many_to_many"("id" integer NOT NULL PRIMARY KEY AUTOINCREMENT, "b_id" integer NOT NULL REFERENCES "app_add_manytomany_field_b"("id") DEFERRABLE INITIALLY DEFERRED, "a_id" integer NOT NULL REFERENCES "app_add_manytomany_field_a"("id") DEFERRABLE INITIALLY DEFERRED);',
|
|
'CREATE UNIQUE INDEX "app_add_manytomany_field_b_many_to_many_b_id_a_id_3e15251d_uniq" ON "app_add_manytomany_field_b_many_to_many"("b_id", "a_id");',
|
|
'CREATE INDEX "app_add_manytomany_field_b_many_to_many_b_id_953b185b" ON "app_add_manytomany_field_b_many_to_many"("b_id");',
|
|
'CREATE INDEX "app_add_manytomany_field_b_many_to_many_a_id_4b44832a" ON "app_add_manytomany_field_b_many_to_many"("a_id");',
|
|
]
|
|
self.assertValidSql(sql)
|
|
|
|
|
|
class PostgresqlAnalyserTestCase(SqlAnalyserTestCase):
|
|
database_vendor = "postgresql"
|
|
|
|
def test_alter_column(self):
|
|
sql = 'ALTER TABLE "app_alter_column_a" ALTER COLUMN "field" TYPE varchar(10) USING "field"::varchar(10);'
|
|
self.assertBackwardIncompatibleSql(sql)
|
|
|
|
def test_not_null_followed_by_default(self):
|
|
sql = [
|
|
'ALTER TABLE "app_add_not_null_column_followed_by_default_a" ADD COLUMN "not_null_field" integer DEFAULT 1 NOT NULL;',
|
|
'ALTER TABLE "app_add_not_null_column_followed_by_default_a" ALTER COLUMN "not_null_field" DROP DEFAULT;',
|
|
'ALTER TABLE "app_add_not_null_column_followed_by_default_a" ALTER COLUMN "not_null_field" SET DEFAULT \'1\';',
|
|
]
|
|
self.assertValidSql(sql)
|
|
|
|
def test_drop_not_null(self):
|
|
sql = 'ALTER TABLE "app_alter_column_drop_not_null_a" ALTER COLUMN "not_null_field" DROP NOT NULL;'
|
|
self.assertValidSql(sql)
|
|
|
|
def test_unique_together(self):
|
|
sql = 'ALTER TABLE "app_unique_together_a" ADD CONSTRAINT "app_unique_together_a_int_field_char_field_979ac7d8_uniq" UNIQUE ("int_field", "char_field");'
|
|
self.assertBackwardIncompatibleSql(sql)
|
|
|
|
sql = 'ALTER TABLE "app_unique_together_a" DROP CONSTRAINT "app_unique_together_a_int_field_char_field_979ac7d8_uniq";'
|
|
self.assertValidSql(sql)
|
|
|
|
def test_add_many_to_many_field(self):
|
|
sql = [
|
|
'CREATE TABLE "app_add_manytomany_field_b_many_to_many"("id" serial NOT NULL PRIMARY KEY, "b_id" integer NOT NULL, "a_id" integer NOT NULL);',
|
|
'ALTER TABLE "app_add_manytomany_field_b_many_to_many" ADD CONSTRAINT "app_add_manytomany_f_b_id_953b185b_fk_app_add_m" FOREIGN KEY("b_id") REFERENCES "app_add_manytomany_field_b"("id") DEFERRABLE INITIALLY DEFERRED;',
|
|
'ALTER TABLE "app_add_manytomany_field_b_many_to_many" ADD CONSTRAINT "app_add_manytomany_f_a_id_4b44832a_fk_app_add_m" FOREIGN KEY("a_id") REFERENCES "app_add_manytomany_field_a"("id") DEFERRABLE INITIALLY DEFERRED;',
|
|
'ALTER TABLE "app_add_manytomany_field_b_many_to_many" ADD CONSTRAINT "app_add_manytomany_field_b_many_to_many_b_id_a_id_3e15251d_uniq" UNIQUE("b_id", "a_id");',
|
|
'CREATE INDEX "app_add_manytomany_field_b_many_to_many_b_id_953b185b" ON "app_add_manytomany_field_b_many_to_many"("b_id");',
|
|
'CREATE INDEX "app_add_manytomany_field_b_many_to_many_a_id_4b44832a" ON "app_add_manytomany_field_b_many_to_many"("a_id");',
|
|
]
|
|
self.assertValidSql(sql)
|