[Fix] Check if queries are null in tests

databricks · Jul 23, 2024 · b82b6b9 · b82b6b9
1 parent c04f58a
commit b82b6b9
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/databricks-sdk-java/src/test/java/com/databricks/sdk/integration/SqlIT.java b/databricks-sdk-java/src/test/java/com/databricks/sdk/integration/SqlIT.java
@@ -22,8 +22,10 @@ void listQueryHistoryTimeRange(WorkspaceClient w) {
         new ListQueryHistoryRequest()
             .setFilterBy(new QueryFilter().setQueryStartTimeRange(timeRange));
     Iterable<QueryInfo> queries = w.queryHistory().list(request).getRes();
-    for (QueryInfo query : queries) {
-      System.out.println(query);
+    if (queries != null) {
+      for (QueryInfo query : queries) {
+        System.out.println(query);
+      }
     }
   }
 
@@ -33,8 +35,10 @@ void listQueryHistoryUserIds(WorkspaceClient w) {
         new ListQueryHistoryRequest()
             .setFilterBy(new QueryFilter().setUserIds(Arrays.asList(123L, 456L)));
     Iterable<QueryInfo> queries = w.queryHistory().list(request).getRes();
-    for (QueryInfo query : queries) {
-      System.out.println(query);
+    if (queries != null) {
+      for (QueryInfo query : queries) {
+        System.out.println(query);
+      }
     }
   }
 }