diff --git a/server/src/test/java/com/metamx/druid/master/DruidMasterRuleRunnerTest.java b/server/src/test/java/com/metamx/druid/master/DruidMasterRuleRunnerTest.java index 5076ffbefe1..c31dc670b82 100644 --- a/server/src/test/java/com/metamx/druid/master/DruidMasterRuleRunnerTest.java +++ b/server/src/test/java/com/metamx/druid/master/DruidMasterRuleRunnerTest.java @@ -357,7 +357,7 @@ public class DruidMasterRuleRunnerTest public void testRunTwoTiersTierDoesNotExist() throws Exception { emitter.emit(EasyMock.anyObject()); - EasyMock.expectLastCall().atLeastOnce(); + EasyMock.expectLastCall().times(12); EasyMock.replay(emitter); EasyMock.expect(databaseRuleManager.getRulesWithDefault(EasyMock.anyObject())).andReturn( @@ -408,7 +408,7 @@ public class DruidMasterRuleRunnerTest public void testRunRuleDoesNotExist() throws Exception { emitter.emit(EasyMock.anyObject()); - EasyMock.expectLastCall().atLeastOnce(); + EasyMock.expectLastCall().times(availableSegments.size()); EasyMock.replay(emitter); EasyMock.expect(databaseRuleManager.getRulesWithDefault(EasyMock.anyObject())).andReturn(