wyl/hertzbeat
1
0
Fork 0
mirror of https://github.com/dromara/hertzbeat.git synced 2026-09-17 09:40:58 +00:00
Code Issues Projects Releases Wiki Activity

Stabilize migration recovery timing test

Browse Source
This commit is contained in:
Logic
2026-08-27 19:00:04 +08:00
parent b4ca58e57b
commit 40e7485be1
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hertzbeat-manager/src/test/java/org/apache/hertzbeat/manager/setup/workflow/DeploymentMigrationCommandRunnerRecoveryTest.java
+3
View File
@@ -65,12 +65,14 @@ class DeploymentMigrationCommandRunnerRecoveryTest {
FileMigrationOperationStore store = new FileMigrationOperationStore(root.resolve("timeout"));
RetainedCutoverCoordinator coordinator = mock(RetainedCutoverCoordinator.class);
CountDownLatch allowPreparation = new CountDownLatch(1);
CountDownLatch preparationPublished = new CountDownLatch(1);
AtomicInteger executions = new AtomicInteger();
when(coordinator.execute(any(), any(), any(), any(), any(), any(), any()))
.thenAnswer(invocation -> {
executions.incrementAndGet();
assertThat(allowPreparation.await(5, SECONDS)).isTrue();
prepare(invocation);
preparationPublished.countDown();
return retained();
});
DeploymentMigrationCommandRunner runner = runner(
@@ -83,6 +85,7 @@ class DeploymentMigrationCommandRunnerRecoveryTest {
assertThat(runner.activeOperationId()).contains(OPERATION);
allowPreparation.countDown();
assertThat(preparationPublished.await(5, SECONDS)).isTrue();
MigrationViewAssert.running(runner.start(request()));
assertThat(executions).hasValue(1);
} finally {