From 18bde40aa9295948437c9b3433301f251fca7cd0 Mon Sep 17 00:00:00 2001 From: Quinn Klassen Date: Wed, 9 Oct 2024 16:43:12 -0700 Subject: [PATCH] fix stub --- .../io/temporal/internal/worker/WorkflowWorkerTest.java | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/temporal-sdk/src/test/java/io/temporal/internal/worker/WorkflowWorkerTest.java b/temporal-sdk/src/test/java/io/temporal/internal/worker/WorkflowWorkerTest.java index 5f9249566..18acaf3e2 100644 --- a/temporal-sdk/src/test/java/io/temporal/internal/worker/WorkflowWorkerTest.java +++ b/temporal-sdk/src/test/java/io/temporal/internal/worker/WorkflowWorkerTest.java @@ -266,9 +266,15 @@ public void respondWorkflowTaskFailureMetricTest() throws Exception { eagerActivityDispatcher, slotSupplier); + WorkflowServiceGrpc.WorkflowServiceFutureStub futureStub = + mock(WorkflowServiceGrpc.WorkflowServiceFutureStub.class); + when(futureStub.shutdownWorker(any(ShutdownWorkerRequest.class))) + .thenReturn(Futures.immediateFuture(ShutdownWorkerResponse.newBuilder().build())); + WorkflowServiceGrpc.WorkflowServiceBlockingStub blockingStub = mock(WorkflowServiceGrpc.WorkflowServiceBlockingStub.class); when(client.blockingStub()).thenReturn(blockingStub); + when(client.futureStub()).thenReturn(futureStub); when(blockingStub.withOption(any(), any())).thenReturn(blockingStub); PollWorkflowTaskQueueResponse pollResponse =