86 lines
3.0 KiB
Python
86 lines
3.0 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 os
|
|
import shutil
|
|
import tempfile
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
from django_migration_linter.utils import get_default_cache_file
|
|
from tests import fixtures
|
|
|
|
from django_migration_linter import MigrationLinter
|
|
from tests.fixtures import ADD_NOT_NULL_COLUMN_PROJECT
|
|
|
|
|
|
class CacheTest(unittest.TestCase):
|
|
MIGRATION_FILE = os.path.join(ADD_NOT_NULL_COLUMN_PROJECT, 'test_app', 'migrations', '0001_create_table.py')
|
|
fd, temp_path = tempfile.mkstemp()
|
|
|
|
def setUp(self):
|
|
shutil.copy2(self.MIGRATION_FILE, self.temp_path)
|
|
|
|
def tearDown(self):
|
|
shutil.copy2(self.temp_path, self.MIGRATION_FILE)
|
|
os.remove(self.temp_path)
|
|
|
|
def test_cache_normal(self):
|
|
if os.path.exists(get_default_cache_file('test_project_add_not_null_column')):
|
|
os.remove(get_default_cache_file('test_project_add_not_null_column'))
|
|
linter = MigrationLinter(fixtures.ADD_NOT_NULL_COLUMN_PROJECT)
|
|
linter.lint_all_migrations()
|
|
cache = linter.get_cache()
|
|
self.assertEqual(cache['hash']['result'], 'OK')
|
|
self.assertEqual(cache['hashERR']['result'], 'ERR')
|
|
self.assertDictEqual(cache['hashERR']['errors'], ...)
|
|
|
|
with mock.patch(
|
|
'subprocess.Popen'
|
|
) as popen_mock:
|
|
linter.lint_all_migrations()
|
|
self.assertEqual(popen_mock.call_count, 2)
|
|
|
|
with mock.patch(
|
|
'subprocess.Popen'
|
|
) as popen_mock:
|
|
linter.lint_all_migrations()
|
|
|
|
self.assertEqual(popen_mock.call_count, 0)
|
|
self.assertFalse(linter.has_errors())
|
|
|
|
def test_cache_ignored(self):
|
|
pass
|
|
|
|
def test_cache_modified(self):
|
|
if os.path.exists(get_default_cache_file('test_project_add_not_null_column')):
|
|
os.remove(get_default_cache_file('test_project_add_not_null_column'))
|
|
linter = MigrationLinter(fixtures.ADD_NOT_NULL_COLUMN_PROJECT)
|
|
linter.lint_all_migrations()
|
|
cache = linter.get_cache()
|
|
self.assertEqual(cache['hash']['result'], 'OK')
|
|
self.assertEqual(cache['hashERR']['result'], 'ERR')
|
|
self.assertDictEqual(cache['hashERR']['errors'], ...)
|
|
|
|
with open( self.MIGRATION_FILE, "a") as f:
|
|
f.write("# modification at the end of the file\n")
|
|
|
|
linter.lint_all_migrations()
|
|
cache = linter.get_cache()
|
|
|
|
self.assertNotIn('hash', cache)
|
|
self.assertEqual(cache['newhash']['result'], 'OK')
|
|
|
|
|
|
|