fix databricks test after CI refactor (OpenLineage#3075)

Signed-off-by: Pawel Leszczynski <[email protected]>
nubank · Sep 10, 2024 · 8bceec8 · 8bceec8
1 parent 2cce532
commit 8bceec8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/integration/spark/app/src/test/java/io/openlineage/spark/agent/DatabricksUtils.java b/integration/spark/app/src/test/java/io/openlineage/spark/agent/DatabricksUtils.java
@@ -193,6 +193,7 @@ private static String createCluster(WorkspaceClient workspace) {
             .build();
 
     log.info("Creating cluster");
+    workspace.config().setHttpTimeoutSeconds(600); // 10 minutes, otherwise it's rather setup issue
     CreateClusterResponse response =
         workspace
             .apiClient()
@@ -221,7 +222,7 @@ private static void uploadOpenlineageJar(WorkspaceClient workspace) {
     Path jarFile =
         Files.list(Paths.get("../build/libs/"))
             .filter(p -> p.getFileName().toString().startsWith("openlineage-spark_"))
-            .filter(p -> p.getFileName().toString().endsWith("jar"))
+            .filter(p -> p.getFileName().toString().endsWith("-SNAPSHOT.jar"))
             .findAny()
             .orElseThrow(() -> new RuntimeException("openlineage-spark jar not found"));