commit 27419e33dc8c48fb5ca6d97f5d6788f9aca2032f
author Fabian Meumertzheim <fabian@meumertzhe.im> Fri Jan 26 11:36:52 2024 -0800
committer Copybara-Service <copybara-worker@google.com> Fri Jan 26 11:38:45 2024 -0800
tree 931e9cd9884035ec50c09086a9e0ca220903cfb0
parent 57c18017de2c6f3ec655d931e3a3ed8060cf84c8

Fix inconsistent dep graph stubs in Bzlmod tests

These cause failures when relying on `depGraph` containing `ModuleKey`s for all modules, which is the case in production.

Work towards #20997

Closes #21037.

PiperOrigin-RevId: 601822357
Change-Id: Ifcad9d7b73835491c1f1fca975e05834057a6825

src/test/java/com/google/devtools/build/lib/bazel/bzlmod/BazelDepGraphFunctionTest.java

diff --git a/src/test/java/com/google/devtools/build/lib/bazel/bzlmod/BazelDepGraphFunctionTest.java b/src/test/java/com/google/devtools/build/lib/bazel/bzlmod/BazelDepGraphFunctionTest.java
index 38b0583..31aacbc 100644
--- a/src/test/java/com/google/devtools/build/lib/bazel/bzlmod/BazelDepGraphFunctionTest.java
+++ b/src/test/java/com/google/devtools/build/lib/bazel/bzlmod/BazelDepGraphFunctionTest.java
@@ -226,11 +226,13 @@
 
   @Test
   public void createValue_moduleExtensions() throws Exception {
+    ModuleKey rjeKey = createModuleKey("rules_jvm_external", "1.0");
+    ModuleKey rpyKey = createModuleKey("rules_python", "2.0");
     Module root =
         buildModule("root", "1.0")
             .setKey(ModuleKey.ROOT)
-            .addDep("rje", createModuleKey("rules_jvm_external", "1.0"))
-            .addDep("rpy", createModuleKey("rules_python", "2.0"))
+            .addDep("rje", rjeKey)
+            .addDep("rpy", rpyKey)
             .addExtensionUsage(
                 createModuleExtensionUsage("@rje//:defs.bzl", "maven", "av", "autovalue"))
             .addExtensionUsage(
@@ -240,7 +242,7 @@
     Module dep =
         buildModule("dep", "2.0")
             .setKey(depKey)
-            .addDep("rules_python", createModuleKey("rules_python", "2.0"))
+            .addDep("rules_python", rpyKey)
             .addExtensionUsage(
                 createModuleExtensionUsage("@rules_python//:defs.bzl", "pip", "np", "numpy"))
             .addExtensionUsage(
@@ -248,7 +250,16 @@
             .addExtensionUsage(
                 createModuleExtensionUsage("//incredible:conflict.bzl", "myext", "twoext", "myext"))
             .build();
-    ImmutableMap<ModuleKey, Module> depGraph = ImmutableMap.of(ModuleKey.ROOT, root, depKey, dep);
+    ImmutableMap<ModuleKey, Module> depGraph =
+        ImmutableMap.of(
+            ModuleKey.ROOT,
+            root,
+            depKey,
+            dep,
+            rjeKey,
+            buildModule("rules_jvm_external", "1.0").setKey(rjeKey).build(),
+            rpyKey,
+            buildModule("rules_python", "2.0").setKey(rpyKey).build());
 
     ModuleExtensionId maven =
         ModuleExtensionId.create(
@@ -326,6 +337,7 @@
   public void useExtensionBadLabelFails() throws Exception {
     Module root =
         buildModule("module", "1.0")
+            .setKey(ModuleKey.ROOT)
             .addExtensionUsage(createModuleExtensionUsage("@foo//:defs.bzl", "bar"))
             .build();
     ImmutableMap<ModuleKey, Module> depGraph = ImmutableMap.of(ModuleKey.ROOT, root);