test(whl_library): test a recent fix for pipstar

MODULE.bazel

@@ -242,6 +242,7 @@ use_repo(
     "pkgutil_nspkg2",
     "rules_python_runtime_env_tc_info",
     "somepkg_with_build_files",
+    "whl_library_extras_direct_dep",
     "whl_with_build_files",
 )
 

python/private/internal_dev_deps.bzl

@@ -91,6 +91,36 @@ def _internal_dev_deps_impl(mctx):
         enable_implicit_namespace_pkgs = False,
     )
 
+    _whl_library_from_dir(
+        name = "whl_library_extras_direct_dep",
+        root = "//tests/pypi/whl_library/testdata/pkg:BUILD.bazel",
+        output = "pkg-1.0-any-none-any.whl",
+        requirement = "pkg[optional]",
+        # The following is necessary to enable pipstar and make tests faster
+        config_load = "@rules_python//tests/pypi/whl_library/testdata:packages.bzl",
+        dep_template = "@whl_library_extras_{name}//:{target}",
+    )
+    _whl_library_from_dir(
+        name = "whl_library_extras_optional_dep",
+        root = "//tests/pypi/whl_library/testdata/optional_dep:BUILD.bazel",
+        output = "optional-dep-1.0-any-none-any.whl",
+        requirement = "optional_dep",
+        # The following is necessary to enable pipstar and make tests faster
+        config_load = "@rules_python//tests/pypi/whl_library/testdata:packages.bzl",
+    )
+
+def _whl_library_from_dir(*, name, output, root, **kwargs):
+    whl_from_dir_repo(
+        name = "{}_whl".format(name),
+        root = root,
+        output = output,
+    )
+    whl_library(
+        name = name,
+        whl_file = "@{}_whl//:{}".format(name, output),
+        **kwargs
+    )
+
 internal_dev_deps = module_extension(
     implementation = _internal_dev_deps_impl,
     doc = "This extension creates internal rules_python dev dependencies.",

tests/pypi/whl_library/BUILD.bazel

@@ -0,0 +1,9 @@
+load("//python:py_test.bzl", "py_test")
+
+py_test(
+    name = "whl_library_extras_test",
+    srcs = ["whl_library_extras_test.py"],
+    deps = [
+        "@whl_library_extras_direct_dep//:pkg",
+    ],
+)

tests/pypi/whl_library/testdata/optional_dep/optional-dep-1.0.dist-info/METADATA

@@ -0,0 +1,2 @@
+Name: optional-dep
+Version: 1.0

tests/pypi/whl_library/testdata/optional_dep/optional-dep-1.0.dist-info/WHEEL

@@ -0,0 +1 @@
+Wheel-Version: 1.0

tests/pypi/whl_library/testdata/optional_dep/optional_dep.py

@@ -0,0 +1 @@
+I_AM_OPTIONAL = True

tests/pypi/whl_library/testdata/packages.bzl

@@ -0,0 +1,6 @@
+"""A list of packages that this logical testdata hub repo contains."""
+
+packages = [
+    "optional_dep",
+    "pkg",
+]

tests/pypi/whl_library/testdata/pkg/pkg-1.0.dist-info/METADATA

@@ -0,0 +1,4 @@
+Name: ns-sub1
+Version: 1.0
+Requires-Dist: optional_dep; extra == "optional"
+Provides-Extra: optional

tests/pypi/whl_library/testdata/pkg/pkg-1.0.dist-info/WHEEL

@@ -0,0 +1 @@
+Wheel-Version: 1.0

tests/pypi/whl_library/testdata/pkg/pkg.py

@@ -0,0 +1,6 @@
+try:
+    import optional_dep
+
+    WITH_EXTRAS = True
+except ImportError:
+    WITH_EXTRAS = False

tests/pypi/whl_library/whl_library_extras_test.py

@@ -0,0 +1,13 @@
+import unittest
+
+
+class NamespacePackagesTest(unittest.TestCase):
+
+    def test_extras_propagated(self):
+        import pkg
+
+        self.assertEqual(pkg.WITH_EXTRAS, True)
+
+
+if __name__ == "__main__":
+    unittest.main()