Disable compression in message-too-large tests

Author: Sushisource

temporal-sdk/src/test/java/io/temporal/workflow/GrpcMessageTooLargeTest.java

@@ -13,6 +13,8 @@
 import io.temporal.internal.replay.ReplayWorkflowTaskHandler;
 import io.temporal.internal.retryer.GrpcMessageTooLargeException;
 import io.temporal.internal.worker.PollerOptions;
+import io.temporal.serviceclient.GrpcCompression;
+import io.temporal.serviceclient.WorkflowServiceStubsOptions;
 import io.temporal.testUtils.LoggerUtils;
 import io.temporal.testing.internal.SDKTestWorkflowRule;
 import io.temporal.workflow.shared.TestActivities;
@@ -22,8 +24,10 @@
 import org.junit.Test;
 
 public class GrpcMessageTooLargeTest {
-  private static final String QUERY_ERROR_MESSAGE =
-      "Failed to send query response: RESOURCE_EXHAUSTED: grpc: received message larger than max";
+  // This string is kept intentionally short to match multiple possible too-large error messages
+  private static final String TOO_BIG_ERR_MESSAGE = "larger than max";
+  private static final WorkflowServiceStubsOptions GRPC_COMPRESSION_DISABLED =
+      WorkflowServiceStubsOptions.newBuilder().setGrpcCompression(GrpcCompression.NONE).build();
   private static final String VERY_LARGE_DATA;
 
   static {
@@ -39,21 +43,31 @@ public class GrpcMessageTooLargeTest {
   @Rule
   public SDKTestWorkflowRule activityStartWorkflowRule =
       SDKTestWorkflowRule.newBuilder()
+          .setWorkflowServiceStubsOptions(GRPC_COMPRESSION_DISABLED)
           .setWorkflowTypes(ActivityStartWorkflowImpl.class)
           .setActivityImplementations(new TestActivityImpl())
           .build();
 
   @Rule
   public SDKTestWorkflowRule failureWorkflowRule =
-      SDKTestWorkflowRule.newBuilder().setWorkflowTypes(FailureWorkflowImpl.class).build();
+      SDKTestWorkflowRule.newBuilder()
+          .setWorkflowServiceStubsOptions(GRPC_COMPRESSION_DISABLED)
+          .setWorkflowTypes(FailureWorkflowImpl.class)
+          .build();
 
   @Rule
   public SDKTestWorkflowRule querySuccessWorkflowRule =
-      SDKTestWorkflowRule.newBuilder().setWorkflowTypes(QuerySuccessWorkflowImpl.class).build();
+      SDKTestWorkflowRule.newBuilder()
+          .setWorkflowServiceStubsOptions(GRPC_COMPRESSION_DISABLED)
+          .setWorkflowTypes(QuerySuccessWorkflowImpl.class)
+          .build();
 
   @Rule
   public SDKTestWorkflowRule queryFailureWorkflowRule =
-      SDKTestWorkflowRule.newBuilder().setWorkflowTypes(QueryFailureWorkflowImpl.class).build();
+      SDKTestWorkflowRule.newBuilder()
+          .setWorkflowServiceStubsOptions(GRPC_COMPRESSION_DISABLED)
+          .setWorkflowTypes(QueryFailureWorkflowImpl.class)
+          .build();
 
   @Test
   public void workflowStartTooLarge() {
@@ -120,7 +134,7 @@ public void queryResultTooLarge() {
     assertNotNull(e.getCause());
     // The exception will not contain the original failure object, so instead of type check we're
     // checking the message to ensure the correct error is being sent.
-    assertTrue(e.getCause().getMessage().contains(QUERY_ERROR_MESSAGE));
+    assertTrue(e.getCause().getMessage().contains(TOO_BIG_ERR_MESSAGE));
   }
 
   @Test
@@ -132,7 +146,7 @@ public void queryErrorTooLarge() {
     WorkflowQueryException e = assertThrows(WorkflowQueryException.class, workflow::query);
 
     assertNotNull(e.getCause());
-    assertTrue(e.getCause().getMessage().contains(QUERY_ERROR_MESSAGE));
+    assertTrue(e.getCause().getMessage().contains(TOO_BIG_ERR_MESSAGE));
   }
 
   private static <T> T createWorkflowStub(Class<T> clazz, SDKTestWorkflowRule workflowRule) {