40 lines
1.6 KiB
Python
40 lines
1.6 KiB
Python
|
from django.test import TestCase
|
||
|
|
from django.db import connection
|
||
|
|
|
||
|
|
|
||
|
|
class MigrationTestBase(TestCase):
|
||
|
|
"""
|
||
|
|
Contains an extended set of asserts for testing migrations and schema operations.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def assertTableExists(self, table):
|
||
|
|
self.assertIn(table, connection.introspection.get_table_list(connection.cursor()))
|
||
|
|
|
||
|
|
def assertTableNotExists(self, table):
|
||
|
|
self.assertNotIn(table, connection.introspection.get_table_list(connection.cursor()))
|
||
|
|
|
||
|
|
def assertColumnExists(self, table, column):
|
||
|
|
self.assertIn(column, [c.name for c in connection.introspection.get_table_description(connection.cursor(), table)])
|
||
|
|
|
||
|
|
def assertColumnNotExists(self, table, column):
|
||
|
|
self.assertNotIn(column, [c.name for c in connection.introspection.get_table_description(connection.cursor(), table)])
|
||
|
|
|
||
|
|
def assertColumnNull(self, table, column):
|
||
|
|
self.assertEqual([c.null_ok for c in connection.introspection.get_table_description(connection.cursor(), table) if c.name == column][0], True)
|
||
|
|
|
||
|
|
def assertColumnNotNull(self, table, column):
|
||
|
|
self.assertEqual([c.null_ok for c in connection.introspection.get_table_description(connection.cursor(), table) if c.name == column][0], False)
|
||
|
|
|
||
|
|
def assertIndexExists(self, table, columns, value=True):
|
||
|
|
self.assertEqual(
|
||
|
|
value,
|
||
|
|
any(
|
||
|
|
c["index"]
|
||
|
|
for c in connection.introspection.get_constraints(connection.cursor(), table).values()
|
||
|
|
if c['columns'] == list(columns)
|
||
|
|
),
|
||
|
|
)
|
||
|
|
|
||
|
|
def assertIndexNotExists(self, table, columns):
|
||
|
|
return self.assertIndexExists(table, columns, False)
|