feat: add tests for new strip mode

Author: ajaxbits

tests/conftest.py

@@ -1,3 +1,4 @@
+import json
 from dataclasses import dataclass
 from pathlib import Path
 from typing import Self
@@ -24,6 +25,7 @@ def new(cls, path: str, contents: str = "") -> Self:
 
 @dataclass
 class Data:
+    files_excluded_from_bundle: list[File]  # relative to packages_dir
     project_files: list[File]  # relative to packages_dir
     pyproject: PythonProject
     python_version: str
@@ -34,10 +36,12 @@ def new(
         cls,
         project_name: str,
         project_files: list[File],
+        files_excluded_from_bundle: list[File],
         python_version: str = "3.13",
     ) -> Self:
         pyproject = _new_python_project(name=project_name)
         return cls(
+            files_excluded_from_bundle=files_excluded_from_bundle,
             project_files=project_files,
             pyproject=pyproject,
             python_version=python_version,
@@ -86,25 +90,50 @@ def verify_file_reproducibility(file_info: list[ZipInfo], expected_file_date_tim
         assert info.date_time == expected_file_date_time
 
 @pytest.fixture
-def test_data(tmp_path: Path):
+def test_files(tmp_path: Path):
+    files_excluded_from_bundle = [
+        File.new("__pycache__/_virtualenv.cpython-313.pyc"),
+        File.new("project_1.dist-info/RECORD"),
+        File.new("project_1.dist-info/direct_url.json", contents=json.dumps({"url": str(tmp_path)})),
+    ]
     files = [
         File.new("project_1/__init__.py"),
         File.new("project_1/project1.py"),
+        File.new("project_1.dist-info/METADATA"),
         File.new("small_dependency/__init__.py"),
         File.new("small_dependency/small_dependency.py", "# This is a small dependency"),
+        *files_excluded_from_bundle,
     ]
-    data = Data.new(project_name="project-1", project_files=files).commit(loc=tmp_path)
-    yield data
+    yield files, files_excluded_from_bundle, tmp_path
 
 @pytest.fixture
-def test_data_nested(tmp_path: Path):
+def test_files_nested(test_files):
+    files, files_excluded_from_bundle, tmp_path = test_files
     files = [
-        File.new("project_1/__init__.py"),
-        File.new("project_1/project1.py"),
-        File.new("small_dependency/__init__.py"),
-        File.new("small_dependency/small_dependency.py", "# This is a small dependency"),
-        File.new("gigantic_dependency/__init__.py"),
-        File.new("gigantic_dependency/gigantic.py", "a" * Packager.AWS_LAMBDA_MAX_UNZIP_SIZE),
+        *files,
+        *[
+            File.new("gigantic_dependency/__init__.py"),
+            File.new("gigantic_dependency/gigantic.py", "a" * Packager.AWS_LAMBDA_MAX_UNZIP_SIZE),
+        ],
     ]
-    data = Data.new(project_name="project-1", project_files=files).commit(loc=tmp_path)
+    yield files, files_excluded_from_bundle, tmp_path
+
+@pytest.fixture
+def test_data(test_files):
+    files, files_excluded_from_bundle, loc = test_files
+    data = Data.new(
+        project_name="project-1",
+        project_files=files,
+        files_excluded_from_bundle=files_excluded_from_bundle,
+    ).commit(loc=loc)
+    yield data
+
+@pytest.fixture
+def test_data_nested(test_files_nested):
+    files, files_excluded_from_bundle, loc = test_files_nested
+    data = Data.new(
+        project_name="project-1-nested",
+        project_files=files,
+        files_excluded_from_bundle=files_excluded_from_bundle,
+    ).commit(loc=loc)
     yield data

tests/test_package_python_function.py

@@ -67,7 +67,10 @@ def test_package_python_function(
             verify_file_reproducibility(zip.infolist(), expected_file_date_time=expected_date_time)
 
         for file in test_data.project_files:
-            assert (verify_dir / file.path).exists()
+            if file in test_data.files_excluded_from_bundle:
+                assert not (verify_dir / file.path).exists()
+            else:
+                assert (verify_dir / file.path).exists()
 
 @pytest.mark.parametrize(
     "src_epoch, expected_exception, expected_date_time",
@@ -131,5 +134,9 @@ def test_package_python_function_nested(
             with zipfile.ZipFile(inner_zip, "r") as izip:
                 izip.extractall(verify_dir)
                 verify_file_reproducibility(izip.infolist(), expected_file_date_time=expected_date_time)
+
                 for file in test_data_nested.project_files:
-                    assert (verify_dir / file.path).exists()
+                    if file in test_data_nested.files_excluded_from_bundle:
+                        assert not (verify_dir / file.path).exists()
+                    else:
+                        assert (verify_dir / file.path).exists()