SNOW-2038030: Add check for expired session when getting result with ID

Author: sfc-gh-ext-simba-lf

Snowflake.Data/Core/SFStatement.cs

@@ -572,6 +572,12 @@ internal async Task<SFBaseResultSet> GetResultWithIdAsync(string resultId, Cance
             var req = BuildResultRequestWithId(resultId);
             QueryExecResponse response = null;
             response = await _restRequester.GetAsync<QueryExecResponse>(req, cancellationToken).ConfigureAwait(false);
+            while (SessionExpired(response))
+            {
+                await SfSession.renewSessionAsync(cancellationToken).ConfigureAwait(false);
+                req.authorizationToken = string.Format(SF_AUTHORIZATION_SNOWFLAKE_FMT, SfSession.sessionToken);
+                response = await _restRequester.GetAsync<QueryExecResponse>(req, cancellationToken).ConfigureAwait(false);
+            }
             return BuildResultSet(response, cancellationToken);
         }
 
@@ -580,6 +586,12 @@ internal SFBaseResultSet GetResultWithId(string resultId)
             var req = BuildResultRequestWithId(resultId);
             QueryExecResponse response = null;
             response = _restRequester.Get<QueryExecResponse>(req);
+            while (SessionExpired(response))
+            {
+                SfSession.renewSession();
+                req.authorizationToken = string.Format(SF_AUTHORIZATION_SNOWFLAKE_FMT, SfSession.sessionToken);
+                response = _restRequester.Get<QueryExecResponse>(req);
+            }
             return BuildResultSet(response, CancellationToken.None);
         }