Serialize MiddlemanAction and add strategy=Strategy.Polymorphic for Action and

AbstractAction.

PiperOrigin-RevId: 186316435

2 files changed, 18 insertions, 0 deletions

diff --git a/src/test/java/com/google/devtools/build/lib/BUILD b/src/test/java/com/google/devtools/build/lib/BUILD
index 6e0f993878..fad138c576 100644
--- a/src/test/java/com/google/devtools/build/lib/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/BUILD
@@ -1290,6 +1290,7 @@ java_test(
         "//src/main/java/com/google/devtools/build/lib:util",
         "//src/main/java/com/google/devtools/build/lib/actions",
         "//src/main/java/com/google/devtools/build/lib/cmdline",
+        "//src/main/java/com/google/devtools/build/lib/skyframe/serialization/testutils",
         "//src/main/java/com/google/devtools/build/lib/vfs",
         "//src/main/java/com/google/devtools/build/lib/vfs/inmemoryfs",
         "//src/main/java/com/google/devtools/common/options",
diff --git a/src/test/java/com/google/devtools/build/lib/exec/MiddlemanActionTest.java b/src/test/java/com/google/devtools/build/lib/exec/MiddlemanActionTest.java
index dade507c4c..42a9a1ffaa 100644
--- a/src/test/java/com/google/devtools/build/lib/exec/MiddlemanActionTest.java
+++ b/src/test/java/com/google/devtools/build/lib/exec/MiddlemanActionTest.java
@@ -26,8 +26,10 @@ import com.google.devtools.build.lib.actions.MiddlemanFactory;
 import com.google.devtools.build.lib.analysis.util.AnalysisTestUtil;
 import com.google.devtools.build.lib.analysis.util.BuildViewTestCase;
 import com.google.devtools.build.lib.cmdline.RepositoryName;
+import com.google.devtools.build.lib.skyframe.serialization.testutils.ObjectCodecTester;
 import com.google.devtools.build.lib.testutil.Suite;
 import com.google.devtools.build.lib.testutil.TestSpec;
+import com.google.devtools.build.lib.vfs.FileSystem;
 import java.util.ArrayList;
 import java.util.Arrays;
 import org.junit.Before;
@@ -135,4 +137,19 @@ public class MiddlemanActionTest extends BuildViewTestCase {
     assertThat(Sets.newHashSet(middlemanActionForD.getOutputs()))
         .isNotEqualTo(Sets.newHashSet(middlemanActionForC.getOutputs()));
   }
+
+  @Test
+  public void testCodec() throws Exception {
+    ObjectCodecTester.newBuilder(MiddlemanAction.CODEC)
+        .addSubjects((MiddlemanAction) getGeneratingAction(middle))
+        .addDependency(FileSystem.class, scratch.getFileSystem())
+        .verificationFunction(
+            (first, second) -> {
+              assertThat(first.getActionType()).isEqualTo(second.getActionType());
+              assertThat(first.getInputs()).isEqualTo(second.getInputs());
+              assertThat(first.getOutputs()).isEqualTo(second.getOutputs());
+              assertThat(first.getOwner()).isEqualTo(second.getOwner());
+            })
+        .buildAndRunTests();
+  }
 }