fix table1 async query get status

delta-io · Oct 9, 2024 · a9ffd0a · a9ffd0a
1 parent cd40567
commit a9ffd0a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/server/src/main/scala/io/delta/sharing/server/DeltaSharingService.scala b/server/src/main/scala/io/delta/sharing/server/DeltaSharingService.scala
@@ -344,17 +344,18 @@ class DeltaSharingService(serverConfig: ServerConfig) {
     }
 
     // simulate async query with 50% chance of return
-    // asynchronously client should be able to hand both cases
-    // this is for test purpose only and shouldn't cause flakiness
-    if(rand.nextInt(100) > 50) {
+    // asynchronously for a specific table
+    // client should be able to handle both cases and for server
+    // test please use other table names.
+    if(rand.nextInt(100) > 50 && table == "table2" && !request.pageToken.isDefined) {
         streamingOutput(
           Some(0),
           "parquet",
           Seq(
             SingleAction(queryStatus = QueryStatus(queryId))
           )
         )
-      } else {
+    } else {
 
       // we are reusing the table here to simulate a view query result
       val tableConfig = sharedTableManager.getTable(share, schema, table)

diff --git a/server/src/test/scala/io/delta/sharing/server/DeltaSharingServiceSuite.scala b/server/src/test/scala/io/delta/sharing/server/DeltaSharingServiceSuite.scala
@@ -694,7 +694,7 @@ class DeltaSharingServiceSuite extends FunSuite with BeforeAndAfterAll {
       )
 
       val lines = response.split("\n")
-      print(s"response: $response")
+      print(s"--response--: $response")
       assert(lines.length == 4)
     }
   }