diff --git a/mayan/apps/dependencies/tests/mocks.py b/mayan/apps/dependencies/tests/mocks.py new file mode 100644 index 0000000000..1bc8bda2ff --- /dev/null +++ b/mayan/apps/dependencies/tests/mocks.py @@ -0,0 +1,11 @@ +from __future__ import unicode_literals + +from ..classes import Dependency, Provider + + +class TestProvider(Provider): + """Test provider""" + + +class TestDependency(Dependency): + provider_class = TestProvider diff --git a/mayan/apps/dependencies/tests/test_classes.py b/mayan/apps/dependencies/tests/test_classes.py index 9ba3a68d9c..63312b51e6 100644 --- a/mayan/apps/dependencies/tests/test_classes.py +++ b/mayan/apps/dependencies/tests/test_classes.py @@ -7,15 +7,7 @@ from mayan.apps.common.tests import BaseTestCase from mayan.apps.common.tests.utils import mute_stdout from mayan.apps.storage.utils import mkdtemp -from ..classes import Dependency, Provider - - -class TestProvider(Provider): - """Test provider""" - - -class TestDependency(Dependency): - provider_class = TestProvider +from .mocks import TestDependency class DependencyClassTestCase(BaseTestCase): @@ -40,7 +32,7 @@ class DependencyClassTestCase(BaseTestCase): super(DependencyClassTestCase, self).tearDown() shutil.rmtree(self.temporary_directory, ignore_errors=True) - def test_file_patching(self): + def _patch_test_file(self): replace_list = [ { 'filename_pattern': '*', @@ -59,8 +51,11 @@ class DependencyClassTestCase(BaseTestCase): ) with self.path_test_file.open(mode='r') as file_object: - final_text = file_object.read() + self.final_text = file_object.read() + + def test_file_patching(self): + self._patch_test_file() self.assertEqual( - final_text, '@import url({});'.format(self.test_replace_text) + self.final_text, '@import url({});'.format(self.test_replace_text) )