SNOW-950923 fixed typo in refactor

snowflakedb · Feb 21, 2024 · 9297cff · 9297cff
1 parent 3cbf4a3
commit 9297cff
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Snowflake.Data.Tests/IntegrationTests/SFConnectionIT.cs b/Snowflake.Data.Tests/IntegrationTests/SFConnectionIT.cs
@@ -505,7 +505,7 @@ public void TestDefaultLoginTimeout()
                 {
                     if (e.InnerException is SnowflakeDbException)
                     {
-                        SnowflakeDbExceptionAssert.HasErrorCode(e.InnerException, SFError.INTERNAL_ERROR);
+                        SnowflakeDbExceptionAssert.HasErrorCode(e.InnerException, SFError.REQUEST_TIMEOUT);
 
                         stopwatch.Stop();
                         int delta = 10; // in case server time slower.
@@ -803,7 +803,7 @@ public void TestUnknownAuthenticator()
                 }
                 catch (SnowflakeDbException e)
                 {
-                    SnowflakeDbExceptionAssert.HasErrorCode(e.InnerException, SFError.UNKNOWN_AUTHENTICATOR);
+                    SnowflakeDbExceptionAssert.HasErrorCode(e, SFError.UNKNOWN_AUTHENTICATOR);
                 }
 
             }
@@ -858,7 +858,7 @@ public void TestOktaConnectionUntilMaxTimeout()
                 catch (Exception e)
                 {
                     Assert.IsInstanceOf<SnowflakeDbException>(e);
-                    SnowflakeDbExceptionAssert.HasErrorCode(e.InnerException, SFError.INTERNAL_ERROR);
+                    SnowflakeDbExceptionAssert.HasErrorCode(e, SFError.INTERNAL_ERROR);
                     Assert.IsTrue(e.Message.Contains(
                         $"The retry count has reached its limit of {expectedMaxRetryCount} and " +
                         $"the timeout elapsed has reached its limit of {expectedMaxConnectionTimeout} " +