test: verify default and after transition

Verify health state per default and after transition to a role

(cherry picked from commit f22e89b)

broker/src/test/java/io/camunda/zeebe/broker/system/partitions/ZeebePartitionTest.java

@@ -357,8 +357,6 @@ public void shouldReportUnhealthyPerDefault() {
     // given
     final var captor = ArgumentCaptor.forClass(ZeebePartitionHealth.class);
     schedulerRule.submitActor(partition);
-    partition.onNewRole(Role.LEADER, 1);
-    schedulerRule.workUntilDone();
 
     // when
     schedulerRule.workUntilDone();
@@ -372,6 +370,24 @@ public void shouldReportUnhealthyPerDefault() {
     assertThat(healthReport.getIssue().message()).contains("Initial state");
   }
 
+  @Test
+  public void shouldReportHealthyAfterTransition() {
+    // given
+    final var captor = ArgumentCaptor.forClass(ZeebePartitionHealth.class);
+    schedulerRule.submitActor(partition);
+
+    // when
+    partition.onNewRole(Role.LEADER, 1);
+    schedulerRule.workUntilDone();
+
+    // then
+    verify(healthMonitor).registerComponent(any(), captor.capture());
+
+    final var zeebePartitionHealth = captor.getValue();
+    final HealthReport healthReport = zeebePartitionHealth.getHealthReport();
+    assertThat(healthReport.getStatus()).isEqualTo(HealthStatus.HEALTHY);
+  }
+
   @Test
   public void shouldReportUnhealthyWhenNoDiskAvailable() {
     // given