From 641c671ea342f147aa5a272bc7cabf2d6b6088a1 Mon Sep 17 00:00:00 2001
From: Julio Merino <julio.merino+oss@snowflake.com>
Date: Fri, 11 Aug 2023 16:46:34 -0700
Subject: [PATCH] Fix tests

---
 .../devtools/build/lib/rules/cpp/CppLinkActionTest.java      | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/test/java/com/google/devtools/build/lib/rules/cpp/CppLinkActionTest.java b/src/test/java/com/google/devtools/build/lib/rules/cpp/CppLinkActionTest.java
index 7d1ce17c041c62..968141da3efa36 100644
--- a/src/test/java/com/google/devtools/build/lib/rules/cpp/CppLinkActionTest.java
+++ b/src/test/java/com/google/devtools/build/lib/rules/cpp/CppLinkActionTest.java
@@ -17,6 +17,7 @@
 import static com.google.common.truth.Truth.assertThat;
 import static org.junit.Assert.assertThrows;
 import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
 
 import com.google.common.base.Joiner;
 import com.google.common.base.Preconditions;
@@ -33,6 +34,7 @@
 import com.google.devtools.build.lib.actions.Artifact.TreeFileArtifact;
 import com.google.devtools.build.lib.actions.ArtifactRoot;
 import com.google.devtools.build.lib.actions.ArtifactRoot.RootType;
+import com.google.devtools.build.lib.actions.InputMetadataProvider;
 import com.google.devtools.build.lib.actions.ResourceSet;
 import com.google.devtools.build.lib.actions.util.ActionsTestUtil;
 import com.google.devtools.build.lib.analysis.ConfiguredTarget;
@@ -787,7 +789,10 @@ private static NestedSet<Artifact> createInputs(RuleContext ruleContext, int cou
 
   private ResourceSet estimateResourceConsumptionLocal(
       RuleContext ruleContext, OS os, int inputsCount) throws Exception {
+    InputMetadataProvider inputMetadataProvider = mock(InputMetadataProvider.class);
+
     ActionExecutionContext actionExecutionContext = mock(ActionExecutionContext.class);
+    when(actionExecutionContext.getInputMetadataProvider()).thenReturn(inputMetadataProvider);
 
     NestedSet<Artifact> inputs = createInputs(ruleContext, inputsCount);
     try {