Organize test helpers

Author: hynek

src/towncrier/test/helpers.py

@@ -0,0 +1,37 @@
+from functools import wraps
+from pathlib import Path
+
+from click.testing import CliRunner
+
+
+def read_all(filename):
+    return Path(filename).read_text()
+
+
+def with_isolated_runner(fn):
+    """
+    Run *fn* within an isolated filesystem and add the kwarg *runner* to its
+    arguments.
+    """
+
+    @wraps(fn)
+    def test(*args, **kw):
+        runner = CliRunner()
+        with runner.isolated_filesystem():
+            return fn(*args, runner=runner, **kw)
+
+    return test
+
+
+def setup_simple_project(
+    *, config=None, pyproject_path="pyproject.toml", mkdir_newsfragments=True
+):
+    if config is None:
+        config = "[tool.towncrier]\n" 'package = "foo"\n'
+
+    Path(pyproject_path).write_text(config)
+    Path("foo").mkdir()
+    Path("foo/__init__.py").write_text('__version__ = "1.2.3"\n')
+
+    if mkdir_newsfragments:
+        Path("foo/newsfragments").mkdir()

src/towncrier/test/test_build.py

@@ -5,7 +5,6 @@
 import tempfile
 
 from datetime import date
-from functools import wraps
 from pathlib import Path
 from subprocess import call
 from textwrap import dedent
@@ -16,35 +15,7 @@
 
 from .._shell import cli
 from ..build import _main
-
-
-def setup_simple_project():
-    with open("pyproject.toml", "w") as f:
-        f.write("[tool.towncrier]\n" 'package = "foo"\n')
-    os.mkdir("foo")
-    with open("foo/__init__.py", "w") as f:
-        f.write('__version__ = "1.2.3"\n')
-    os.mkdir("foo/newsfragments")
-
-
-def read_all(filename):
-    with open(filename) as f:
-        return f.read()
-
-
-def with_isolated_runner(fn):
-    """
-    Run *fn* within an isolated filesystem and add the kwarg *runner* to its
-    arguments.
-    """
-
-    @wraps(fn)
-    def test(*args, **kw):
-        runner = CliRunner()
-        with runner.isolated_filesystem():
-            return fn(*args, runner=runner, **kw)
-
-    return test
+from .helpers import read_all, setup_simple_project, with_isolated_runner
 
 
 class TestCli(TestCase):

src/towncrier/test/test_check.py

@@ -13,18 +13,15 @@
 from towncrier import check
 from towncrier.check import _main as towncrier_check
 
+from .helpers import setup_simple_project
+
 
 def create_project(pyproject_path="pyproject.toml", main_branch="main"):
     """
     Create the project files in the main branch that already has a
     news-fragment and then switch to a new in-work branch.
     """
-    with open(pyproject_path, "w") as f:
-        f.write("[tool.towncrier]\n" 'package = "foo"\n')
-    os.mkdir("foo")
-    with open("foo/__init__.py", "w") as f:
-        f.write('__version__ = "1.2.3"\n')
-    os.mkdir("foo/newsfragments")
+    setup_simple_project(pyproject_path=pyproject_path)
     fragment_path = "foo/newsfragments/123.feature"
     with open(fragment_path, "w") as f:
         f.write("Adds levitation")

src/towncrier/test/test_create.py

@@ -10,26 +10,7 @@
 from twisted.trial.unittest import TestCase
 
 from ..create import _main
-
-
-def setup_simple_project(config=None, mkdir=True):
-    if not config:
-        config = dedent(
-            """\
-            [tool.towncrier]
-            package = "foo"
-            """
-        )
-
-    with open("pyproject.toml", "w") as f:
-        f.write(config)
-
-    os.mkdir("foo")
-    with open("foo/__init__.py", "w") as f:
-        f.write('__version__ = "1.2.3"\n')
-
-    if mkdir:
-        os.mkdir("foo/newsfragments")
+from .helpers import setup_simple_project
 
 
 class TestCli(TestCase):
@@ -41,7 +22,7 @@ def _test_success(
         runner = CliRunner()
 
         with runner.isolated_filesystem():
-            setup_simple_project(config, mkdir)
+            setup_simple_project(config=config, mkdir_newsfragments=mkdir)
 
             args = ["123.feature.rst"]
             if content is None:
@@ -95,7 +76,7 @@ def test_edit_abort(self):
             runner = CliRunner()
 
             with runner.isolated_filesystem():
-                setup_simple_project(config=None, mkdir=True)
+                setup_simple_project(config=None, mkdir_newsfragments=True)
                 result = runner.invoke(_main, ["123.feature.rst", "--edit"])
                 self.assertEqual([], os.listdir("foo/newsfragments"))
                 self.assertEqual(1, result.exit_code)
@@ -141,7 +122,7 @@ def test_different_directory(self):
         )
 
         with runner.isolated_filesystem():
-            setup_simple_project(config, mkdir=False)
+            setup_simple_project(config=config, mkdir_newsfragments=False)
             os.mkdir("releasenotes")
 
             result = runner.invoke(_main, ["123.feature.rst"])