Updated test to use OAuth 2.0

code4me · Apr 10, 2023 · 9012310 · 9012310
1 parent e3f94b6
commit 9012310
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 8 deletions.
diff --git a/Scr/Sdk4me.GraphQL.Tests/Client/Client.cs b/Scr/Sdk4me.GraphQL.Tests/Client/Client.cs
@@ -11,23 +11,42 @@ public static Sdk4meClient Get()
             if (client != null)
                 return client;
 
-            //get account and token
+            //get all information
             IConfigurationProvider secretProvider = new ConfigurationBuilder().AddUserSecrets<ConnectionSecret>().Build().Providers.First();
-            if (!secretProvider.TryGet("Account", out string? account) || !secretProvider.TryGet("Token", out string? token))
+            if (!secretProvider.TryGet("Account", out string? account) 
+                || !secretProvider.TryGet("Token", out string? token) 
+                || !secretProvider.TryGet("ClientID", out string? clientID)
+                || !secretProvider.TryGet("ClientSecret", out string? clientSecret))
             {
-                Assert.Fail("No account of token information available via the user secret file");
+                Assert.Fail("Missing information in the user secret file");
                 return null;
             }
 
-            if (account == null || token == null)
+            if (account == null || token == null || clientID == null || clientSecret == null)
             {
-                Assert.Fail("No account of token information available via the user secret file");
+                Assert.Fail("Null value information in the user secret file");
                 return null;
             }
 
-            //connect to client
-            client = new(new AuthenticationToken(token), account, EnvironmentType.Demo, EnvironmentRegion.EU, 5);
-            client.MaximumQueryDepthLevelConnections = 3;
+            if (!string.IsNullOrEmpty(clientID) && !string.IsNullOrWhiteSpace(clientSecret))
+            {
+                client = new(new AuthenticationToken(clientID, clientSecret), account, EnvironmentType.Demo, EnvironmentRegion.EU, 5)
+                {
+                    MaximumQueryDepthLevelConnections = 3
+                };
+            }
+            else if (!string.IsNullOrEmpty(token))
+            {
+                client = new(new AuthenticationToken(token), account, EnvironmentType.Demo, EnvironmentRegion.EU, 5)
+                {
+                    MaximumQueryDepthLevelConnections = 3
+                };
+            }
+            else
+            {
+                Assert.Fail("Invalid information in the user secret file");
+                return null;
+            }
             return client;
         }
     }

diff --git a/Scr/Sdk4me.GraphQL.Tests/Client/ConnectionSecret.cs b/Scr/Sdk4me.GraphQL.Tests/Client/ConnectionSecret.cs
@@ -5,5 +5,9 @@ internal class ConnectionSecret
         public string Account { get; set; } = string.Empty;
 
         public string Token { get; set; } = string.Empty;
+
+        public string ClientID { get; set; } = string.Empty;
+
+        public string ClientSecret { get; set; } = string.Empty;
     }
 }