Fake paths with endswith

Author: stsewd

readthedocs/rtd_tests/tests/test_builds.py

@@ -15,7 +15,7 @@
 from readthedocs.doc_builder.python_environments import Virtualenv
 from readthedocs.projects.models import Project
 from readthedocs.projects.tasks import UpdateDocsTask
-from readthedocs.rtd_tests.mocks.paths import fake_paths
+from readthedocs.rtd_tests.mocks.paths import fake_paths_by_endswith
 from readthedocs.rtd_tests.tests.test_config_wrapper import create_load
 
 from ..mocks.environment import EnvironmentMockGroup
@@ -76,14 +76,12 @@ def test_build_generate_index_file_force(self):
         python_env = Virtualenv(version=version, build_env=build_env)
         base_builder = MkdocsHTML(build_env, python_env)
 
-        def look_index_path(path):
-            if path.endswith('README.md'):
-                return True
-            elif path.endswith('index.md'):
-                return False
-            return False
+        paths = {
+            'README.md': True,
+            'index.md': False,
+        }
 
-        with mock.patch('readthedocs.doc_builder.base.open', mock.mock_open()) as mock_open, fake_paths(look_index_path):
+        with mock.patch('readthedocs.doc_builder.base.open', mock.mock_open()) as mock_open, fake_paths_by_endswith(paths):
             result = base_builder.create_index(extension='md', force_index=True)
             mock_open.assert_called_once_with(
                 os.path.join(base_builder.docs_dir(), 'index.md'), 'w+'
@@ -109,14 +107,12 @@ def test_build_get_index_file_force(self):
         python_env = Virtualenv(version=version, build_env=build_env)
         base_builder = MkdocsHTML(build_env, python_env)
 
-        def look_index_path(path):
-            if path.endswith('README.md'):
-                return True
-            elif path.endswith('index.md'):
-                return True
-            return False
+        paths = {
+            'README.md': True,
+            'index.md': True,
+        }
 
-        with mock.patch('readthedocs.doc_builder.base.open', mock.mock_open()) as mock_open, fake_paths(look_index_path):
+        with mock.patch('readthedocs.doc_builder.base.open', mock.mock_open()) as mock_open, fake_paths_by_endswith(paths):
             result = base_builder.create_index(extension='md', force_index=True)
             self.assertEqual(len(mock_open.mock_calls), 0)
             self.assertEqual(result, 'index')
@@ -140,14 +136,12 @@ def test_build_generate_index_file(self):
         python_env = Virtualenv(version=version, build_env=build_env)
         base_builder = SphinxHTML(build_env, python_env)
 
-        def look_index_path(path):
-            if path.endswith('README.md'):
-                return False
-            elif path.endswith('index.md'):
-                return False
-            return False
+        paths = {
+            'README.md': False,
+            'index,md': False,
+        }
 
-        with mock.patch('readthedocs.doc_builder.base.open', mock.mock_open()) as mock_open, fake_paths(look_index_path):
+        with mock.patch('readthedocs.doc_builder.base.open', mock.mock_open()) as mock_open, fake_paths_by_endswith(paths):
             result = base_builder.create_index(extension='md')
             mock_open.assert_called_once_with(
                 os.path.join(base_builder.docs_dir(), 'index.md'), 'w+'
@@ -173,14 +167,12 @@ def test_get_readme_file(self):
         python_env = Virtualenv(version=version, build_env=build_env)
         base_builder = SphinxHTML(build_env, python_env)
 
-        def look_index_path(path):
-            if path.endswith('README.md'):
-                return True
-            elif path.endswith('index.md'):
-                return False
-            return False
+        paths = {
+            'README.md': True,
+            'index.md': False,
+        }
 
-        with mock.patch('readthedocs.doc_builder.base.open', mock.mock_open()) as mock_open, fake_paths(look_index_path):
+        with mock.patch('readthedocs.doc_builder.base.open', mock.mock_open()) as mock_open, fake_paths_by_endswith(paths):
             result = base_builder.create_index(extension='md')
             self.assertEqual(len(mock_open.mock_calls), 0)
             self.assertEqual(result, 'README')
@@ -204,14 +196,12 @@ def test_get_index_file(self):
         python_env = Virtualenv(version=version, build_env=build_env)
         base_builder = SphinxHTML(build_env, python_env)
 
-        def look_index_path(path):
-            if path.endswith('README.md'):
-                return False
-            elif path.endswith('index.md'):
-                return True
-            return False
+        paths = {
+            'README.md': False,
+            'index.md': True,
+        }
 
-        with mock.patch('readthedocs.doc_builder.base.open', mock.mock_open()) as mock_open, fake_paths(look_index_path):
+        with mock.patch('readthedocs.doc_builder.base.open', mock.mock_open()) as mock_open, fake_paths_by_endswith(paths):
             result = base_builder.create_index(extension='md')
             self.assertEqual(len(mock_open.mock_calls), 0)
             self.assertEqual(result, 'index')