Create variable for max status retry

snowflakedb · Mar 22, 2024 · 1d7dafb · 1d7dafb
1 parent 6ec657b
commit 1d7dafb
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/Snowflake.Data.Tests/IntegrationTests/SFDbCommandIT.cs b/Snowflake.Data.Tests/IntegrationTests/SFDbCommandIT.cs
@@ -317,13 +317,14 @@ public async Task TestFailedAsyncExecQueryThrowsErrorAsync()
                 using (SnowflakeDbCommand cmd = (SnowflakeDbCommand)conn.CreateCommand())
                 {
                     // Arrange
+                    var statusMaxRetryCount = 5;
                     var statusRetryCount = 0;
                     cmd.CommandText = $"SELECT * FROM FAKE_TABLE;";
 
                     // Act
                     var queryId = await cmd.ExecuteAsyncInAsyncMode(CancellationToken.None).ConfigureAwait(false);
                     var queryStatus = await cmd.GetQueryStatusAsync(queryId, CancellationToken.None).ConfigureAwait(false);
-                    while (statusRetryCount < 5 && QueryStatuses.IsStillRunning(queryStatus))
+                    while (statusRetryCount < statusMaxRetryCount && QueryStatuses.IsStillRunning(queryStatus))
                     {
                         Thread.Sleep(1000);
                         queryStatus = await cmd.GetQueryStatusAsync(queryId, CancellationToken.None).ConfigureAwait(false);
@@ -1410,12 +1411,13 @@ public void TestFailedAsyncExecQueryThrowsError()
                 using (SnowflakeDbCommand cmd = (SnowflakeDbCommand)conn.CreateCommand())
                 {
                     // Arrange
+                    var statusMaxRetryCount = 5;
                     var statusRetryCount = 0;
                     cmd.CommandText = $"SELECT * FROM FAKE_TABLE;";
 
                     // Act
                     queryId = cmd.ExecuteInAsyncMode();
-                    while (statusRetryCount < 5 && QueryStatuses.IsStillRunning(cmd.GetQueryStatus(queryId)))
+                    while (statusRetryCount < statusMaxRetryCount && QueryStatuses.IsStillRunning(cmd.GetQueryStatus(queryId)))
                     {
                         Thread.Sleep(1000);
                         statusRetryCount++;