|
31 | 31 | import io.temporal.client.WorkflowOptions;
|
32 | 32 | import io.temporal.common.reporter.TestStatsReporter;
|
33 | 33 | import io.temporal.serviceclient.MetricsTag;
|
| 34 | +import io.temporal.testUtils.Eventually; |
34 | 35 | import io.temporal.testing.internal.SDKTestWorkflowRule;
|
35 | 36 | import io.temporal.worker.tuning.*;
|
36 | 37 | import io.temporal.workflow.*;
|
@@ -115,10 +116,16 @@ public void canShutdownInTheMiddle() throws InterruptedException {
|
115 | 116 | workflow.activitiesStarted();
|
116 | 117 | testWorkflowRule.getTestEnvironment().getWorkerFactory().shutdownNow();
|
117 | 118 | testWorkflowRule.getTestEnvironment().getWorkerFactory().awaitTermination(3, TimeUnit.SECONDS);
|
118 |
| - reporter.assertGauge(MetricsType.WORKER_TASK_SLOTS_USED, getWorkerTags("WorkflowWorker"), 0); |
119 |
| - reporter.assertGauge(MetricsType.WORKER_TASK_SLOTS_USED, getWorkerTags("ActivityWorker"), 0); |
120 |
| - reporter.assertGauge( |
121 |
| - MetricsType.WORKER_TASK_SLOTS_USED, getWorkerTags("LocalActivityWorker"), 0); |
| 119 | + Eventually.assertEventually( |
| 120 | + Duration.ofMillis(2000), |
| 121 | + () -> { |
| 122 | + reporter.assertGauge( |
| 123 | + MetricsType.WORKER_TASK_SLOTS_USED, getWorkerTags("WorkflowWorker"), 0); |
| 124 | + reporter.assertGauge( |
| 125 | + MetricsType.WORKER_TASK_SLOTS_USED, getWorkerTags("ActivityWorker"), 0); |
| 126 | + reporter.assertGauge( |
| 127 | + MetricsType.WORKER_TASK_SLOTS_USED, getWorkerTags("LocalActivityWorker"), 0); |
| 128 | + }); |
122 | 129 | }
|
123 | 130 |
|
124 | 131 | @WorkflowInterface
|
|
0 commit comments