remove cloudfetch from test

Author: toddmeng-db

csharp/test/Drivers/Databricks/StatementTests.cs

@@ -436,13 +436,12 @@ protected override void PrepareCreateTableWithPrimaryKeys(out string sqlUpdate,
 
         // NOTE: this is a thirty minute test. As of writing, databricks commands have 20 minutes of idle time (and checked every 5 mintues)
         [SkippableTheory]
-        [InlineData(true, "CloudFetch enabled")]
-        [InlineData(false, "CloudFetch disabled")]
+        [InlineData(true, "CloudFetch enabled")] // TODO: test cloudfetch disabled
         public async Task StatusPollerKeepsQueryAlive(bool useCloudFetch, string configName)
         {
             OutputHelper?.WriteLine($"Testing status poller with long delay between reads ({configName})");
 
-            // Create a connection using the test configuration with a small batch size
+            // Create a connection using the test configuration
             var connectionParams = new Dictionary<string, string>
             {
                 [DatabricksParameters.UseCloudFetch] = useCloudFetch.ToString().ToLower()
@@ -451,24 +450,18 @@ public async Task StatusPollerKeepsQueryAlive(bool useCloudFetch, string configN
             using var statement = connection.CreateStatement();
 
             // Execute a query that should return data - using a larger dataset to ensure multiple batches
-            statement.SqlQuery = "SELECT id, CAST(id AS STRING) as id_string, id * 2 as id_doubled FROM RANGE(30000000)";
+            statement.SqlQuery = "SELECT id, CAST(id AS STRING) as id_string, id * 2 as id_doubled FROM RANGE(3000000)";
             QueryResult result = statement.ExecuteQuery();
 
             Assert.NotNull(result.Stream);
 
-            // Read first batch
-            using var firstBatch = await result.Stream.ReadNextRecordBatchAsync();
-            Assert.NotNull(firstBatch);
-            int firstBatchRows = firstBatch.Length;
-            OutputHelper?.WriteLine($"First batch: Read {firstBatchRows} rows");
-
             // Simulate a long delay (30 minutes)
             OutputHelper?.WriteLine("Simulating 30 minute delay...");
             await Task.Delay(TimeSpan.FromMinutes(30));
 
             // Read remaining batches
-            int totalRows = firstBatchRows;
-            int batchCount = 1;
+            int totalRows = 0;
+            int batchCount = 0;
 
             while (result.Stream != null)
             {
@@ -482,7 +475,7 @@ public async Task StatusPollerKeepsQueryAlive(bool useCloudFetch, string configN
             }
 
             // Verify we got all rows
-            Assert.Equal(30000000, totalRows);
+            Assert.Equal(3000000, totalRows);
             Assert.True(batchCount > 1, "Should have read multiple batches");
             OutputHelper?.WriteLine($"Successfully read {totalRows} rows in {batchCount} batches after 30 minute delay with {configName}");
         }