Updating the get customer command (#196)

microsoft · Nov 22, 2019 · 0aaa8b6 · 0aaa8b6
1 parent fe1ce1f
commit 0aaa8b6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/PowerShell/Commands/GetPartnerCustomer.cs b/src/PowerShell/Commands/GetPartnerCustomer.cs
@@ -11,6 +11,7 @@ namespace Microsoft.Store.PartnerCenter.PowerShell.Commands
     using Enumerators;
     using Models.Authentication;
     using Models.Customers;
+    using Network;
     using PartnerCenter.Models;
     using PartnerCenter.Models.Customers;
     using PartnerCenter.Models.Query;
@@ -56,7 +57,9 @@ public override void ExecuteCmdlet()
             }
             if (ParameterSetName.Equals("ByDomain", StringComparison.InvariantCultureIgnoreCase))
             {
-                Graph.IGraphServiceClient client = PartnerSession.Instance.ClientFactory.CreateGraphServiceClient();
+                Graph.GraphServiceClient client = PartnerSession.Instance.ClientFactory.CreateGraphServiceClient() as Graph.GraphServiceClient;
+                client.AuthenticationProvider = new GraphAuthenticationProvider();
+
                 Graph.IGraphServiceContractsCollectionPage data = client.Contracts.Request().Filter($"defaultDomainName eq '{Domain}'").GetAsync().ConfigureAwait(false).GetAwaiter().GetResult();
 
                 if (data.CurrentPage != null && data.CurrentPage.Any())