#23 Add tests for caching

tests/functional/test_cache.py

@@ -0,0 +1,85 @@
+# 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')
+
+
+