108 lines
4.4 KiB
Python
108 lines
4.4 KiB
Python
# Copyright 2018 3YOURMIND GmbH
|
|
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import unittest
|
|
|
|
from django_migration_linter import MigrationLinter
|
|
from tests import fixtures
|
|
|
|
|
|
class TestLinterFunctions(unittest.TestCase):
|
|
def test_get_sql(self):
|
|
project_path = fixtures.ADD_NOT_NULL_COLUMN_PROJECT
|
|
linter = MigrationLinter(project_path)
|
|
sql_statements = linter.get_sql('test_app', '0001')
|
|
self.assertEqual(len(sql_statements), 6)
|
|
self.assertEqual(sql_statements[0], 'BEGIN;')
|
|
self.assertEqual(sql_statements[-1], 'COMMIT;')
|
|
|
|
def test_has_errors(self):
|
|
project_path = fixtures.MULTI_COMMIT_PROJECT
|
|
linter = MigrationLinter(project_path)
|
|
self.assertFalse(linter.has_errors)
|
|
|
|
linter.lint_migration('test_app', '0001')
|
|
self.assertFalse(linter.has_errors)
|
|
|
|
linter.lint_migration('test_app', '0002')
|
|
self.assertTrue(linter.has_errors)
|
|
|
|
linter.lint_migration('test_app', '0001')
|
|
self.assertTrue(linter.has_errors)
|
|
|
|
def test_linter_creation(self):
|
|
with self.assertRaises(ValueError):
|
|
MigrationLinter('foo/')
|
|
with self.assertRaises(ValueError):
|
|
MigrationLinter('/dev/null')
|
|
with self.assertRaises(ValueError):
|
|
MigrationLinter(fixtures.NOT_DJANGO_GIT_PROJECT)
|
|
|
|
def test_split_migration_long_path(self):
|
|
input_path = 'apps/the_app/migrations/0001_stuff.py'
|
|
app, mig = MigrationLinter._split_migration_path(input_path)
|
|
self.assertEqual(app, 'the_app')
|
|
self.assertEqual(mig, '0001_stuff')
|
|
|
|
def test_split_migration_path(self):
|
|
input_path = 'the_app/migrations/0001_stuff.py'
|
|
app, mig = MigrationLinter._split_migration_path(input_path)
|
|
self.assertEqual(app, 'the_app')
|
|
self.assertEqual(mig, '0001_stuff')
|
|
|
|
def test_split_migration_full_path(self):
|
|
input_path = '/home/user/djangostuff/apps/the_app/migrations/0001_stuff.py'
|
|
app, mig = MigrationLinter._split_migration_path(input_path)
|
|
self.assertEqual(app, 'the_app')
|
|
self.assertEqual(mig, '0001_stuff')
|
|
|
|
def test_ignore_migration_include_apps(self):
|
|
linter = MigrationLinter(
|
|
fixtures.ADD_NOT_NULL_COLUMN_PROJECT,
|
|
include_apps=('test_app2',))
|
|
self.assertTrue(linter.should_ignore_migration('test_app1', '0001'))
|
|
self.assertFalse(linter.should_ignore_migration('test_app2', '0001'))
|
|
|
|
def test_ignore_migration_exclude_apps(self):
|
|
linter = MigrationLinter(
|
|
fixtures.ADD_NOT_NULL_COLUMN_PROJECT,
|
|
exclude_apps=('test_app1',))
|
|
self.assertTrue(linter.should_ignore_migration('test_app1', '0001'))
|
|
self.assertFalse(linter.should_ignore_migration('test_app2', '0001'))
|
|
|
|
def test_ignore_migration_name_contains(self):
|
|
linter = MigrationLinter(
|
|
fixtures.ADD_NOT_NULL_COLUMN_PROJECT,
|
|
ignore_name_contains='foo')
|
|
self.assertTrue(linter.should_ignore_migration('test_app', '0001_foo'))
|
|
self.assertFalse(linter.should_ignore_migration('test_app', '0002_bar'))
|
|
|
|
def test_ignore_migration_full_name(self):
|
|
linter = MigrationLinter(
|
|
fixtures.ADD_NOT_NULL_COLUMN_PROJECT,
|
|
ignore_name=('0001_foo',))
|
|
self.assertTrue(linter.should_ignore_migration('test_app', '0001_foo'))
|
|
self.assertFalse(linter.should_ignore_migration('test_app', '0002_bar'))
|
|
|
|
def test_gather_all_migrations(self):
|
|
linter = MigrationLinter(fixtures.CORRECT_PROJECT)
|
|
migrations = linter._gather_all_migrations()
|
|
self.assertEqual(len(migrations), 3)
|
|
self.assertEqual(migrations[0][0], 'test_app1')
|
|
self.assertEqual(migrations[0][1], '0001_initial')
|
|
self.assertEqual(migrations[1][0], 'test_app1')
|
|
self.assertEqual(migrations[1][1], '0002_a_new_null_field')
|
|
self.assertEqual(migrations[2][0], 'test_app2')
|
|
self.assertEqual(migrations[2][1], '0001_foo')
|