Tests for TestWorkflowEnvironmentInternal

cadence-workflow · Nov 27, 2024 · b4cb524 · b4cb524
1 parent f6343ca
commit b4cb524
Show file tree

Hide file tree

Showing 2 changed files with 698 additions and 2 deletions.
diff --git a/src/main/java/com/uber/cadence/internal/sync/TestWorkflowEnvironmentInternal.java b/src/main/java/com/uber/cadence/internal/sync/TestWorkflowEnvironmentInternal.java
@@ -498,7 +498,7 @@ public void DeprecateDomain(
     public void RestartWorkflowExecution(
         RestartWorkflowExecutionRequest restartRequest, AsyncMethodCallback resultHandler)
         throws TException {
-      impl.RestartWorkflowExecution(restartRequest);
+      impl.RestartWorkflowExecution(restartRequest, resultHandler);
     }
 
     @Override
@@ -743,7 +743,7 @@ public void CountWorkflowExecutions(
 
     @Override
     public void GetSearchAttributes(AsyncMethodCallback resultHandler) throws TException {
-      impl.GetSearchAttributes();
+      impl.GetSearchAttributes(resultHandler);
     }
 
     @Override