Add test for scripts generated in build.py

Author: robtaylor

tests/masonry/builders/fixtures/script_generated/build.py

@@ -0,0 +1,21 @@
+from pathlib import Path
+from setuptools.command.build_py import build_py
+
+
+class BuildPyCommand(build_py):
+    def run(self):
+        with open('script_generated/generated/file.py', 'w') as out:
+            print("#!/bin/env python3\nprint('Success!')\n", file=out)
+        return super().run()
+
+
+def build(setup_kwargs):
+    setup_kwargs.update(
+        {
+            "cmdclass": {
+                "build_py": BuildPyCommand,
+            },
+            "scripts": ["script_generated/generated/file.py"]
+        }
+
+    )

tests/masonry/builders/fixtures/script_generated/pyproject.toml

@@ -0,0 +1,16 @@
+[tool.poetry]
+name = "script_generated"
+version = "0.1"
+description = ""
+authors = ["Evgenii Gorchakov <[email protected]>", "Rob Taylor <[email protected]>"]
+
+[tool.poetry.dependencies]
+python = "^3.8"
+
+[build-system]
+requires = ["poetry-core", "setuptools"]
+build-backend = "poetry.core.masonry.api"
+
+[tool.poetry.build]
+generate-setup-file = true
+script = "build.py"

tests/masonry/builders/fixtures/script_generated/script_generated/__init__.py

@@ -93,6 +93,28 @@ def test_build_wheel_extended() -> None:
         assert whl.exists()
         validate_wheel_contents(name="extended", version="0.1", path=whl.as_posix())
 
+
+@pytest.mark.skipif(
+    sys.platform == "win32"
+    and sys.version_info <= (3, 6)
+    or platform.python_implementation().lower() == "pypy",
+    reason="Disable test on Windows for Python <=3.6 and for PyPy",
+)
+def test_build_wheel_script_generated() -> None:
+    with temporary_directory() as tmp_dir, cwd(
+        os.path.join(fixtures, "script_generated")
+    ):
+        filename = api.build_wheel(tmp_dir)
+        whl = Path(tmp_dir) / filename
+        assert whl.exists()
+        validate_wheel_contents(
+            name="script_generated",
+            version="0.1",
+            path=whl.as_posix(),
+            data_files=["scripts/file.py"],
+        )
+
+
 def test_build_sdist() -> None:
     with temporary_directory() as tmp_dir, cwd(os.path.join(fixtures, "complete")):
         filename = api.build_sdist(tmp_dir)

tests/masonry/builders/fixtures/script_generated/script_generated/generated/__init__.py

@@ -60,16 +60,24 @@ def subprocess_run(*args: str, **kwargs: Any) -> subprocess.CompletedProcess[str
 
 
 def validate_wheel_contents(
-    name: str, version: str, path: str, files: list[str] | None = None
+    name: str,
+    version: str,
+    path: str,
+    files: list[str] | None = None,
+    data_files: list[str] | None = None,
 ) -> None:
     dist_info = f"{name}-{version}.dist-info"
+    dist_data = f"{name}-{version}.data"
     files = files or []
+    data_files = data_files or []
 
     with zipfile.ZipFile(path) as z:
         namelist = z.namelist()
         # we use concatenation here for PY2 compat
         for filename in ["WHEEL", "METADATA", "RECORD"] + files:
             assert f"{dist_info}/{filename}" in namelist
+        for filename in data_files:
+            assert f"{dist_data}/{filename}" in namelist
 
 
 def validate_sdist_contents(