change assertion. add stopEngine

apache · Sep 20, 2024 · 092668a · 092668a
1 parent 7aca1fd
commit 092668a
Showing 1 changed file with 17 additions and 7 deletions.
diff --git a/...src/test/java/org/drools/compiler/integrationtests/TimerAndCalendarFireUntilHaltTest.java b/...src/test/java/org/drools/compiler/integrationtests/TimerAndCalendarFireUntilHaltTest.java
@@ -118,11 +118,14 @@ public void testTimerRuleFires() throws Exception {
         startEngine();
 
         activateRule();
+        await().until(ruleHasFired("TimerRule", 1));
         advanceTimerOneSecond();
 
         LOG.info("  -- before await : elapsed " + (System.currentTimeMillis() - start) + "ms");
-        await().until(ruleHasFired("TimerRule", 1));
+        await().until(ruleHasFired("TimerRule", 2));
         LOG.info("  -- end : elapsed " + (System.currentTimeMillis() - start) + "ms");
+
+        stopEngine();
     }
 
     @Test(timeout = 10000)
@@ -140,16 +143,17 @@ public void testTimerRuleHaltStopsFiring() throws Exception {
         setupKSessionFor(drl);
         startEngine();
         activateRule();
+        await().until(ruleHasFired("TimerRule", 1));
         advanceTimerOneSecond();
         LOG.info("  -- before await : elapsed " + (System.currentTimeMillis() - start) + "ms");
-        await().until(ruleHasFired("TimerRule", 1));
+        await().until(ruleHasFired("TimerRule", 2));
 
         stopEngine();
 
         advanceTimerOneSecond();
 
         LOG.info("  -- before 2nd await : elapsed " + (System.currentTimeMillis() - start) + "ms");
-        await().during(Duration.ofSeconds(1)).atMost(Duration.ofSeconds(2)).until(ruleHasFired("TimerRule", 1));
+        await().during(Duration.ofSeconds(1)).atMost(Duration.ofSeconds(2)).until(ruleHasFired("TimerRule", 2));
         LOG.info("  -- end : elapsed " + (System.currentTimeMillis() - start) + "ms");
     }
 
@@ -168,18 +172,21 @@ public void testTimerRuleRestartsAfterStop() throws Exception {
         setupKSessionFor(drl);
         startEngine();
         activateRule();
+        await().until(ruleHasFired("TimerRule", 1));
         advanceTimerOneSecond();
         LOG.info("  -- before await : elapsed " + (System.currentTimeMillis() - start) + "ms");
-        await().until(ruleHasFired("TimerRule", 1));
+        await().until(ruleHasFired("TimerRule", 2));
 
         stopEngine();
         startEngine();
 
         advanceTimerOneSecond();
 
         LOG.info("  -- before 2nd await : elapsed " + (System.currentTimeMillis() - start) + "ms");
-        await().during(Duration.ofSeconds(1)).atMost(Duration.ofSeconds(2)).until(ruleHasFired("TimerRule", 2));
+        await().during(Duration.ofSeconds(1)).atMost(Duration.ofSeconds(2)).until(ruleHasFired("TimerRule", 3));
         LOG.info("  -- end : elapsed " + (System.currentTimeMillis() - start) + "ms");
+
+        stopEngine();
     }
 
     @Test(timeout = 10000)
@@ -197,10 +204,11 @@ public void testTimerRuleDoesRestartsIfNoLongerHolds() throws Exception {
         setupKSessionFor(drl);
         startEngine();
         activateRule();
+        await().until(ruleHasFired("TimerRule", 1));
         advanceTimerOneSecond();
 
         LOG.info("  -- before await : elapsed " + (System.currentTimeMillis() - start) + "ms");
-        await().until(ruleHasFired("TimerRule", 1));
+        await().until(ruleHasFired("TimerRule", 2));
 
         stopEngine();
         disactivateRule();
@@ -209,8 +217,10 @@ public void testTimerRuleDoesRestartsIfNoLongerHolds() throws Exception {
         advanceTimerOneSecond();
 
         LOG.info("  -- before 2nd await : elapsed " + (System.currentTimeMillis() - start) + "ms");
-        await().during(Duration.ofSeconds(1)).atMost(Duration.ofSeconds(2)).until(ruleHasFired("TimerRule", 1));
+        await().during(Duration.ofSeconds(1)).atMost(Duration.ofSeconds(2)).until(ruleHasFired("TimerRule", 2));
         LOG.info("  -- end : elapsed " + (System.currentTimeMillis() - start) + "ms");
+
+        stopEngine();
     }