Allow to set access token explicitly when running UI tests.

sourcegraph · Jan 17, 2025 · cc781f6 · cc781f6
1 parent 4352eae
commit cc781f6
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 0 deletions.
diff --git a/src/Cody.Core/Infrastructure/ConfigurationService.cs b/src/Cody.Core/Infrastructure/ConfigurationService.cs
@@ -75,6 +75,13 @@ public ExtensionConfiguration GetConfiguration()
                 CustomConfiguration = GetCustomConfiguration()
             };
 
+            if (_userSettingsService.ForceAccessTokenForUITests)
+            {
+#pragma warning disable CS0618 // Type or member is obsolete
+                config.AccessToken = _userSettingsService.AccessToken;
+#pragma warning restore CS0618 // Type or member is obsolete
+            }
+
             return config;
         }
 

diff --git a/src/Cody.Core/Settings/IUserSettingsService.cs b/src/Cody.Core/Settings/IUserSettingsService.cs
@@ -11,5 +11,6 @@ public interface IUserSettingsService
         string CustomConfiguration { get; set; }
         bool AcceptNonTrustedCert { get; set; }
         bool AutomaticallyTriggerCompletions { get; set; }
+        bool ForceAccessTokenForUITests { get; set; }
     }
 }
diff --git a/src/Cody.Core/Settings/UserSettingsService.cs b/src/Cody.Core/Settings/UserSettingsService.cs
@@ -72,6 +72,9 @@ public string AccessToken
             }
             set
             {
+                if (!ForceAccessTokenForUITests)
+                    throw new Exception("Setting access token explicitly only allowed when running UI tests!");
+
                 var userToken = _secretStorage.AccessToken;
                 if (!string.Equals(value, userToken, StringComparison.InvariantCulture))
                 {
@@ -105,5 +108,7 @@ public bool AutomaticallyTriggerCompletions
             }
             set => Set(nameof(AutomaticallyTriggerCompletions), value.ToString());
         }
+
+        public bool ForceAccessTokenForUITests { get; set; }
     }
 }
diff --git a/src/Cody.VisualStudio.Tests/PlaywrightTestsBase.cs b/src/Cody.VisualStudio.Tests/PlaywrightTestsBase.cs
@@ -58,6 +58,7 @@ private async Task InitializeAsync()
                 if (accessToken != null)
                 {
                     WriteLog("Using Access Token.");
+                    CodyPackage.UserSettingsService.ForceAccessTokenForUITests = true;
                     CodyPackage.UserSettingsService.AccessToken = accessToken;
                 }
 

diff --git a/src/Cody.VisualStudio/Services/SecretStorageService.cs b/src/Cody.VisualStudio/Services/SecretStorageService.cs
@@ -113,7 +113,11 @@ public string AccessToken
             }
             set
             {
+                var oldAccessToken = AccessToken;
                 Set(AccessTokenKey, value);
+
+                if (oldAccessToken != value)
+                    AuthorizationDetailsChanged?.Invoke(this, EventArgs.Empty);
             }
         }
     }