SNOW-648958 Add support for new OKTA OIE to JDBC Driver (#1122)

* Add support for new OKTA OIE to JDBC Driver
snowflakedb · Aug 23, 2022 · 1658330 · 1658330
1 parent db5bb61
commit 1658330
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main/java/net/snowflake/client/core/SessionUtil.java b/src/main/java/net/snowflake/client/core/SessionUtil.java
@@ -1150,7 +1150,10 @@ private static String federatedFlowStep3(SFLoginInput loginInput, String tokenUr
 
       // session token is in the data field of the returned json response
       final JsonNode jsonNode = mapper.readTree(idpResponse);
-      oneTimeToken = jsonNode.get("cookieToken").asText();
+      oneTimeToken =
+          jsonNode.get("sessionToken") != null
+              ? jsonNode.get("sessionToken").asText()
+              : jsonNode.get("cookieToken").asText();
     } catch (IOException | URISyntaxException ex) {
       handleFederatedFlowError(loginInput, ex);
     }