Fix proto and thrift mappers

Author: c-warren

src/main/java/com/uber/cadence/internal/compatibility/proto/RequestMapper.java

@@ -29,6 +29,7 @@
 import static com.uber.cadence.internal.compatibility.proto.Helpers.newFieldMask;
 import static com.uber.cadence.internal.compatibility.proto.Helpers.nullToEmpty;
 import static com.uber.cadence.internal.compatibility.proto.Helpers.secondsToDuration;
+import static com.uber.cadence.internal.compatibility.proto.Helpers.unixNanoToTime;
 import static com.uber.cadence.internal.compatibility.proto.TypeMapper.badBinaries;
 import static com.uber.cadence.internal.compatibility.proto.TypeMapper.clusterReplicationConfigurationArray;
 import static com.uber.cadence.internal.compatibility.proto.TypeMapper.failure;
@@ -403,6 +404,7 @@ public static DescribeWorkflowExecutionRequest describeWorkflowExecutionRequest(
     return DescribeWorkflowExecutionRequest.newBuilder()
         .setDomain(t.getDomain())
         .setWorkflowExecution(workflowExecution(t.getExecution()))
+        .setQueryConsistencyLevel(queryConsistencyLevel(t.getQueryConsistencyLevel()))
         .build();
   }
 
@@ -418,7 +420,8 @@ public static GetWorkflowExecutionHistoryRequest getWorkflowExecutionHistoryRequ
             .setPageSize(t.getMaximumPageSize())
             .setWaitForNewEvent(t.isWaitForNewEvent())
             .setHistoryEventFilterType(eventFilterType(t.HistoryEventFilterType))
-            .setSkipArchival(t.isSkipArchival());
+            .setSkipArchival(t.isSkipArchival())
+            .setQueryConsistencyLevel(queryConsistencyLevel(t.getQueryConsistencyLevel()));
     if (t.getNextPageToken() != null) {
       builder.setNextPageToken(arrayToByteString(t.getNextPageToken()));
     }
@@ -445,6 +448,9 @@ public static SignalWithStartWorkflowExecutionRequest signalWithStartWorkflowExe
             .setSearchAttributes(searchAttributes(t.getSearchAttributes()))
             .setHeader(header(t.getHeader()))
             .setJitterStart(secondsToDuration(t.getJitterStartSeconds()));
+    if (t.isSetFirstRunAtTimestamp()) {
+      builder.setFirstRunAt(unixNanoToTime(t.getFirstRunAtTimestamp()));
+    }
     if (t.getRetryPolicy() != null) {
       builder.setRetryPolicy(retryPolicy(t.getRetryPolicy()));
     }
@@ -530,6 +536,9 @@ public static StartWorkflowExecutionRequest startWorkflowExecutionRequest(
             .setHeader(header(t.getHeader()))
             .setDelayStart(secondsToDuration(t.getDelayStartSeconds()))
             .setJitterStart(secondsToDuration(t.getJitterStartSeconds()));
+    if (t.isSetFirstRunAtTimestamp()) {
+      request.setFirstRunAt(unixNanoToTime(t.getFirstRunAtTimestamp()));
+    }
     if (t.getRetryPolicy() != null) {
       request.setRetryPolicy(retryPolicy(t.getRetryPolicy()));
     }

src/test/java/com/uber/cadence/internal/compatibility/ProtoObjects.java

@@ -984,6 +984,7 @@ public final class ProtoObjects {
       DescribeWorkflowExecutionRequest.newBuilder()
           .setDomain("domain")
           .setWorkflowExecution(WORKFLOW_EXECUTION)
+          .setQueryConsistencyLevel(QueryConsistencyLevel.QUERY_CONSISTENCY_LEVEL_STRONG)
           .build();
 
   public static final GetWorkflowExecutionHistoryRequest GET_WORKFLOW_EXECUTION_HISTORY_REQUEST =
@@ -995,6 +996,7 @@ public final class ProtoObjects {
           .setHistoryEventFilterType(EventFilterType.EVENT_FILTER_TYPE_CLOSE_EVENT)
           .setSkipArchival(true)
           .setNextPageToken(utf8("nextPageToken"))
+          .setQueryConsistencyLevel(QueryConsistencyLevel.QUERY_CONSISTENCY_LEVEL_STRONG)
           .build();
 
   public static final StartWorkflowExecutionRequest START_WORKFLOW_EXECUTION =
@@ -1016,6 +1018,7 @@ public final class ProtoObjects {
           .setHeader(HEADER)
           .setDelayStart(seconds(3))
           .setJitterStart(seconds(0))
+          .setFirstRunAt(timestampNanos(123456789))
           .build();
 
   public static final SignalWithStartWorkflowExecutionRequest SIGNAL_WITH_START_WORKFLOW_EXECUTION =

src/test/java/com/uber/cadence/internal/compatibility/ThriftObjects.java

@@ -843,7 +843,10 @@ public final class ThriftObjects {
           .setQuery("query");
 
   public static final DescribeWorkflowExecutionRequest DESCRIBE_WORKFLOW_EXECUTION_REQUEST =
-      new DescribeWorkflowExecutionRequest().setDomain("domain").setExecution(WORKFLOW_EXECUTION);
+      new DescribeWorkflowExecutionRequest()
+          .setDomain("domain")
+          .setExecution(WORKFLOW_EXECUTION)
+          .setQueryConsistencyLevel(QueryConsistencyLevel.STRONG);
 
   public static final GetWorkflowExecutionHistoryRequest GET_WORKFLOW_EXECUTION_HISTORY_REQUEST =
       new GetWorkflowExecutionHistoryRequest()
@@ -853,7 +856,8 @@ public final class ThriftObjects {
           .setWaitForNewEvent(true)
           .setHistoryEventFilterType(HistoryEventFilterType.CLOSE_EVENT)
           .setSkipArchival(true)
-          .setNextPageToken(utf8("nextPageToken"));
+          .setNextPageToken(utf8("nextPageToken"))
+          .setQueryConsistencyLevel(QueryConsistencyLevel.STRONG);
 
   public static final com.uber.cadence.StartWorkflowExecutionRequest START_WORKFLOW_EXECUTION =
       new com.uber.cadence.StartWorkflowExecutionRequest()
@@ -873,7 +877,8 @@ public final class ThriftObjects {
           .setSearchAttributes(SEARCH_ATTRIBUTES)
           .setHeader(HEADER)
           .setJitterStartSeconds(0)
-          .setDelayStartSeconds(3);
+          .setDelayStartSeconds(3)
+          .setFirstRunAtTimestamp(123456789L);
   public static final com.uber.cadence.SignalWithStartWorkflowExecutionRequest
       SIGNAL_WITH_START_WORKFLOW_EXECUTION =
           new SignalWithStartWorkflowExecutionRequest()
@@ -896,7 +901,8 @@ public final class ThriftObjects {
               .setSearchAttributes(SEARCH_ATTRIBUTES)
               .setHeader(HEADER)
               .setDelayStartSeconds(3)
-              .setJitterStartSeconds(0);
+              .setJitterStartSeconds(0)
+              .setFirstRunAtTimestamp(123456789L);
 
   public static final StartWorkflowExecutionAsyncRequest START_WORKFLOW_EXECUTION_ASYNC_REQUEST =
       new StartWorkflowExecutionAsyncRequest().setRequest(START_WORKFLOW_EXECUTION);