Fix test

apache · Jun 27, 2022 · 726e13a · 726e13a
1 parent c8450a7
commit 726e13a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/...est/java/org/apache/gobblin/service/modules/scheduler/GobblinServiceJobSchedulerTest.java b/...est/java/org/apache/gobblin/service/modules/scheduler/GobblinServiceJobSchedulerTest.java
@@ -31,7 +31,6 @@
 import java.util.Map;
 import java.util.Properties;
 import org.apache.gobblin.configuration.ConfigurationKeys;
-import org.apache.gobblin.exception.QuotaExceededException;
 import org.apache.gobblin.runtime.JobException;
 import org.apache.gobblin.runtime.api.FlowSpec;
 import org.apache.gobblin.runtime.api.JobSpec;
@@ -338,7 +337,7 @@ public void testJobSchedulerAddFlowQuotaExceeded() throws Exception {
     scheduler.setActive(true);
 
     scheduler.onAddSpec(flowSpec0); //Ignore the response for this request
-    Assert.assertThrows(QuotaExceededException.class, () -> scheduler.onAddSpec(flowSpec1));
+    Assert.assertThrows(RuntimeException.class, () -> scheduler.onAddSpec(flowSpec1));
 
     Assert.assertEquals(scheduler.scheduledFlowSpecs.size(), 1);
     // Second flow should not be added to scheduled flows since it was rejected