test: introduce an e2e test asserting that bzlmod works

MODULE.bazel

@@ -0,0 +1,18 @@
+"aspect-build/bazel-lib"
+
+module(
+    name = "aspect_bazel_lib",
+    version = "0.0.0",
+    compatibility_level = 1,
+    toolchains_to_register = [
+        "@jq_toolchains//:all",
+        "@yq_toolchains//:all",
+    ],
+)
+
+bazel_dep(name = "bazel_skylib", version = "1.1.1")
+bazel_dep(name = "platforms", version = "0.0.4")
+
+ext = use_extension("@aspect_bazel_lib//lib:extensions.bzl", "lib")
+use_repo(ext, "jq_toolchains")
+use_repo(ext, "yq_toolchains")

WORKSPACE

@@ -25,6 +25,12 @@ local_repository(
     path = "./lib/tests/external_test_repo",
 )
 
+# Avoid descending into the e2e workspaces
+local_repository(
+    name = "e2e_bzlmod",
+    path = "./e2e/bzlmod",
+)
+
 ############################################
 # Gazelle, for generating bzl_library targets
 load("@io_bazel_rules_go//go:deps.bzl", "go_register_toolchains", "go_rules_dependencies")

e2e/bzlmod/.bazelrc

@@ -0,0 +1,2 @@
+common --experimental_enable_bzlmod
+common --registry=https://raw.githubusercontent.com/aspect-build/bazel-central-registry/main/

e2e/bzlmod/BUILD.bazel

@@ -0,0 +1,8 @@
+load("@aspect_bazel_lib//lib:jq.bzl", "jq")
+
+# No sources produces null (equivalent to --null-input)
+jq(
+    name = "case_no_sources",
+    srcs = [],
+    filter = ".",
+)

e2e/bzlmod/MODULE.bazel

@@ -0,0 +1,19 @@
+module(
+    name = "e2e_bzlmod",
+    version = "0.0.0",
+    compatibility_level = 1,
+    toolchains_to_register = [
+        "@jq_toolchains//:all",
+    ],
+)
+
+bazel_dep(name = "aspect_bazel_lib", version = "99.99.99")
+
+local_path_override(
+    module_name = "aspect_bazel_lib",
+    path = "../..",
+)
+
+ext = use_extension("@aspect_bazel_lib//lib:extensions.bzl", "lib")
+use_repo(ext, "jq_toolchains")
+use_repo(ext, "yq_toolchains")

lib/extensions.bzl

@@ -0,0 +1,17 @@
+"Module extensions for use with bzlmod"
+
+load(
+    "@aspect_bazel_lib//lib:repositories.bzl",
+    "DEFAULT_JQ_VERSION",
+    "DEFAULT_YQ_VERSION",
+    "register_jq_toolchains",
+    "register_yq_toolchains",
+)
+
+def _toolchain_extension(_):
+    register_yq_toolchains(version = DEFAULT_YQ_VERSION, register = False)
+    register_jq_toolchains(version = DEFAULT_JQ_VERSION, register = False)
+
+lib = module_extension(
+    implementation = _toolchain_extension,
+)

lib/repositories.bzl

@@ -21,20 +21,23 @@ def aspect_bazel_lib_dependencies():
 DEFAULT_JQ_VERSION = _DEFAULT_JQ_VERSION
 DEFAULT_YQ_VERSION = _DEFAULT_YQ_VERSION
 
-def register_jq_toolchains(name = "jq", version = DEFAULT_JQ_VERSION):
+def register_jq_toolchains(name = "jq", version = DEFAULT_JQ_VERSION, register = True):
     """Registers jq toolchain and repositories
 
     Args:
         name: override the prefix for the generated toolchain repositories
         version: the version of jq to execute (see https://github.com/stedolan/jq/releases)
+        register: whether to call through to native.register_toolchains.
+            Should be True for WORKSPACE users, but false when used under bzlmod extension
     """
     for [platform, meta] in JQ_PLATFORMS.items():
         jq_platform_repo(
             name = "%s_%s" % (name, platform),
             platform = platform,
             version = version,
         )
-        native.register_toolchains("@%s_toolchains//:%s_toolchain" % (name, platform))
+        if register:
+            native.register_toolchains("@%s_toolchains//:%s_toolchain" % (name, platform))
 
     jq_host_alias_repo(
         name = "%s_host" % name,
@@ -46,20 +49,23 @@ def register_jq_toolchains(name = "jq", version = DEFAULT_JQ_VERSION):
         user_repository_name = name,
     )
 
-def register_yq_toolchains(name = "yq", version = DEFAULT_YQ_VERSION):
+def register_yq_toolchains(name = "yq", version = DEFAULT_YQ_VERSION, register = True):
     """Registers yq toolchain and repositories
 
     Args:
         name: override the prefix for the generated toolchain repositories
         version: the version of yq to execute (see https://github.com/mikefarah/yq/releases)
+        register: whether to call through to native.register_toolchains.
+            Should be True for WORKSPACE users, but false when used under bzlmod extension
     """
     for [platform, meta] in YQ_PLATFORMS.items():
         yq_platform_repo(
             name = "%s_%s" % (name, platform),
             platform = platform,
             version = version,
         )
-        native.register_toolchains("@%s_toolchains//:%s_toolchain" % (name, platform))
+        if register:
+            native.register_toolchains("@%s_toolchains//:%s_toolchain" % (name, platform))
 
     yq_host_alias_repo(
         name = "%s_host" % name,