32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
from unittest.mock import patch
|
|
|
|
from django.core.management import call_command
|
|
from django.test import TransactionTestCase
|
|
|
|
|
|
class LintMigrationsCommandTestCase(TransactionTestCase):
|
|
databases = {"default", "sqlite"}
|
|
|
|
def test_plain(self):
|
|
with self.assertRaises(SystemExit):
|
|
call_command("lintmigrations")
|
|
|
|
def test_config_file_app_label(self):
|
|
with patch(
|
|
"django_migration_linter.management.commands.lintmigrations.Command.read_config_file"
|
|
) as config_fn:
|
|
config_fn.return_value = {"app_label": "app_correct"}
|
|
call_command("lintmigrations")
|
|
|
|
def test_command_line_app_label(self):
|
|
call_command("lintmigrations", app_label="app_correct")
|
|
|
|
def test_command_line_and_config_file_app_label(self):
|
|
with patch(
|
|
"django_migration_linter.management.commands.lintmigrations.Command.read_config_file"
|
|
) as config_fn:
|
|
config_fn.return_value = {"app_label": "app_correct"}
|
|
|
|
with self.assertRaises(SystemExit):
|
|
call_command("lintmigrations", app_label="app_drop_table")
|