Add tests

Author: sfc-gh-ext-simba-lf

Snowflake.Data.Tests/Mock/MockRestSessionExpired.cs

@@ -16,6 +16,8 @@ class MockRestSessionExpired : IMockRestRequester
     {
         static private readonly String EXPIRED_SESSION_TOKEN="session_expired_token";
 
+        static private readonly String NEW_SESSION_TOKEN="new_session_token";
+
         static private readonly String TOKEN_FMT = "Snowflake Token=\"{0}\"";
 
         static private readonly int SESSION_EXPIRED_CODE = 390112;
@@ -58,7 +60,7 @@ public Task<T> PostAsync<T>(IRestRequest request, CancellationToken cancellation
                     };
                     return Task.FromResult<T>((T)(object)queryExecResponse);
                 }
-                else if (sfRequest.authorizationToken.Equals(String.Format(TOKEN_FMT, "new_session_token")))
+                else if (sfRequest.authorizationToken.Equals(String.Format(TOKEN_FMT, NEW_SESSION_TOKEN)))
                 {
                     SecondTimeRequestID = ExtractRequestID(sfRequest.Url.Query);
                     QueryExecResponse queryExecResponse = new QueryExecResponse
@@ -97,7 +99,7 @@ public Task<T> PostAsync<T>(IRestRequest request, CancellationToken cancellation
                     success = true,
                     data = new RenewSessionResponseData()
                     {
-                        sessionToken = "new_session_token",
+                        sessionToken = NEW_SESSION_TOKEN,
                         masterToken = "new_master_token"
                     }
                 });
@@ -120,6 +122,37 @@ public T Get<T>(IRestRequest request)
 
         public Task<T> GetAsync<T>(IRestRequest request, CancellationToken cancellationToken)
         {
+            SFRestRequest sfRequest = (SFRestRequest)request;
+            if (sfRequest.authorizationToken.Equals(String.Format(TOKEN_FMT, EXPIRED_SESSION_TOKEN)))
+            {
+                    QueryExecResponse queryExecResponse = new QueryExecResponse
+                    {
+                        success = false,
+                        code = SESSION_EXPIRED_CODE
+                    };
+                    return Task.FromResult<T>((T)(object)queryExecResponse);
+            }
+            if (sfRequest.authorizationToken.Equals(String.Format(TOKEN_FMT, NEW_SESSION_TOKEN)))
+            {
+                    QueryExecResponse queryExecResponse = new QueryExecResponse
+                    {
+                        success = true,
+                        data = new QueryExecResponseData
+                        {
+                            rowSet = new string[,] { { "abc" } },
+                            rowType = new List<ExecResponseRowType>()
+                            {
+                                new ExecResponseRowType
+                                {
+                                    name = "colOne",
+                                    type = SFDataType.TEXT.ToString()
+                                }
+                            },
+                            parameters = new List<NameValueParameter>()
+                        }
+                    };
+                    return Task.FromResult<T>((T)(object)queryExecResponse);
+            }
             return Task.FromResult<T>((T)(object)null);
         }
 

Snowflake.Data.Tests/UnitTests/SFStatementTest.cs

@@ -7,6 +7,7 @@
 using Snowflake.Data.Core;
 using NUnit.Framework;
 using System;
+using System.Threading.Tasks;
 
 namespace Snowflake.Data.Tests.UnitTests
 {
@@ -32,6 +33,34 @@ public void TestSessionRenew()
             Assert.AreEqual(restRequester.FirstTimeRequestID, restRequester.SecondTimeRequestID);
         }
 
+        [Test]
+        public void TestSessionRenewGetResultWithId()
+        {
+            Mock.MockRestSessionExpired restRequester = new Mock.MockRestSessionExpired();
+            SFSession sfSession = new SFSession("account=test;user=test;password=test", null, restRequester);
+            sfSession.Open();
+            SFStatement statement = new SFStatement(sfSession);
+            SFBaseResultSet resultSet = statement.GetResultWithId("mockId");
+            Assert.AreEqual(true, resultSet.Next());
+            Assert.AreEqual("abc", resultSet.GetString(0));
+            Assert.AreEqual("new_session_token", sfSession.sessionToken);
+            Assert.AreEqual("new_master_token", sfSession.masterToken);
+        }
+
+        [Test]
+        public async Task TestSessionRenewGetResultWithIdAsync()
+        {
+            Mock.MockRestSessionExpired restRequester = new Mock.MockRestSessionExpired();
+            SFSession sfSession = new SFSession("account=test;user=test;password=test", null, restRequester);
+            sfSession.Open();
+            SFStatement statement = new SFStatement(sfSession);
+            SFBaseResultSet resultSet = await statement.GetResultWithIdAsync("mockId", CancellationToken.None);
+            Assert.AreEqual(true, resultSet.Next());
+            Assert.AreEqual("abc", resultSet.GetString(0));
+            Assert.AreEqual("new_session_token", sfSession.sessionToken);
+            Assert.AreEqual("new_master_token", sfSession.masterToken);
+        }
+
         // Mock test for session renew during query execution
         [Test]
         public void TestSessionRenewDuringQueryExec()