Fix unit tests
diff --git a/extensions-contrib/kubernetes-overlord-extensions/src/test/java/org/apache/druid/k8s/overlord/taskadapter/PodTemplateTaskAdapterTest.java b/extensions-contrib/kubernetes-overlord-extensions/src/test/java/org/apache/druid/k8s/overlord/taskadapter/PodTemplateTaskAdapterTest.java
index 5d0ce10..1796bb2 100644
--- a/extensions-contrib/kubernetes-overlord-extensions/src/test/java/org/apache/druid/k8s/overlord/taskadapter/PodTemplateTaskAdapterTest.java
+++ b/extensions-contrib/kubernetes-overlord-extensions/src/test/java/org/apache/druid/k8s/overlord/taskadapter/PodTemplateTaskAdapterTest.java
@@ -94,8 +94,8 @@
@Test
public void test_fromTask_withoutBasePodTemplateInRuntimeProperites_raisesIAE()
{
- Assert.assertThrows(
- "Pod template task adapter requires a base pod template to be specified",
+ Exception exception = Assert.assertThrows(
+ "No base prop should throw an IAE",
IAE.class,
() -> new PodTemplateTaskAdapter(
taskRunnerConfig,
@@ -105,6 +105,7 @@
new Properties(),
taskLogs
));
+ Assert.assertEquals(exception.getMessage(), "Pod template task adapter requires a base pod template to be specified under druid.indexer.runner.k8s.podTemplate.base");
}
@Test
@@ -115,8 +116,8 @@
Properties props = new Properties();
props.setProperty("druid.indexer.runner.k8s.podTemplate.base", templatePath.toString());
- Assert.assertThrows(
- "Pod template task adapter requires a base pod template to be specified",
+ Exception exception = Assert.assertThrows(
+ "Empty base pod template should throw a exception",
IAE.class,
() -> new PodTemplateTaskAdapter(
taskRunnerConfig,
@@ -126,6 +127,9 @@
props,
taskLogs
));
+
+ Assert.assertTrue(exception.getMessage().contains("Failed to load pod template file for"));
+
}
@Test
