From c483cea88c7d58052258d9761a356d026682a531 Mon Sep 17 00:00:00 2001
From: Justin M <63619224+JustinMDotNet@users.noreply.github.com>
Date: Mon, 29 Mar 2021 15:08:09 -0700
Subject: [PATCH] Renamed RefreshAzureToken to RefreshAuthToken. Changed
 provider value to authenticationType. (#1183)

---
 .../Connection/ConnectionService.cs                           | 4 ++--
 src/Microsoft.Kusto.ServiceLayer/DataSource/KustoClient.cs    | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/Microsoft.Kusto.ServiceLayer/Connection/ConnectionService.cs b/src/Microsoft.Kusto.ServiceLayer/Connection/ConnectionService.cs
index 1ce46ba5..4d4c8987 100644
--- a/src/Microsoft.Kusto.ServiceLayer/Connection/ConnectionService.cs
+++ b/src/Microsoft.Kusto.ServiceLayer/Connection/ConnectionService.cs
@@ -266,7 +266,7 @@ namespace Microsoft.Kusto.ServiceLayer.Connection
             return completeParams;
         }
 
-        internal string RefreshAzureToken(string ownerUri)
+        internal string RefreshAuthToken(string ownerUri)
         {
             TryFindConnection(ownerUri, out ConnectionInfo connection);
 
@@ -274,7 +274,7 @@ namespace Microsoft.Kusto.ServiceLayer.Connection
             {
                 AccountId = connection.ConnectionDetails.GetOptionValue("azureAccount", string.Empty),
                 Authority = connection.ConnectionDetails.GetOptionValue("azureTenantId", string.Empty),
-                Provider = "Azure",
+                Provider = connection.ConnectionDetails.AuthenticationType,
                 Resource = "SQL"
             };
 
diff --git a/src/Microsoft.Kusto.ServiceLayer/DataSource/KustoClient.cs b/src/Microsoft.Kusto.ServiceLayer/DataSource/KustoClient.cs
index 931edf77..da5d4fcb 100644
--- a/src/Microsoft.Kusto.ServiceLayer/DataSource/KustoClient.cs
+++ b/src/Microsoft.Kusto.ServiceLayer/DataSource/KustoClient.cs
@@ -57,7 +57,7 @@ namespace Microsoft.Kusto.ServiceLayer.DataSource
 
         private void RefreshAuthToken()
         {
-            string accountToken = ConnectionService.Instance.RefreshAzureToken(_ownerUri);
+            string accountToken = ConnectionService.Instance.RefreshAuthToken(_ownerUri);
             _kustoQueryProvider.Dispose();
             _kustoAdminProvider.Dispose();