Fix tests

sourcegraph · Oct 16, 2024 · 48da69c · 48da69c
1 parent a856cfd
commit 48da69c
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/src/integrationTest/kotlin/com/sourcegraph/cody/util/CodyIntegrationTextFixture.kt b/src/integrationTest/kotlin/com/sourcegraph/cody/util/CodyIntegrationTextFixture.kt
@@ -21,7 +21,7 @@ import com.intellij.testFramework.fixtures.BasePlatformTestCase
 import com.intellij.testFramework.runInEdtAndWait
 import com.sourcegraph.cody.agent.CodyAgentService
 import com.sourcegraph.cody.agent.protocol_generated.ProtocolCodeLens
-import com.sourcegraph.cody.auth.CodyPersistentAccountsHost
+import com.sourcegraph.cody.auth.CodyAccount
 import com.sourcegraph.cody.auth.SourcegraphServerPath
 import com.sourcegraph.cody.edit.lenses.LensListener
 import com.sourcegraph.cody.edit.lenses.LensesService
@@ -94,13 +94,9 @@ open class CodyIntegrationTextFixture : BasePlatformTestCase(), LensListener {
   // change anything.
   private fun initCredentialsAndAgent() {
     val credentials = TestingCredentials.dotcom
-    CodyPersistentAccountsHost(project)
-        .addAccount(
-            SourcegraphServerPath.from(credentials.serverEndpoint, ""),
-            login = "test_user",
-            displayName = "Test User",
-            token = credentials.token ?: credentials.redactedToken,
-            id = "random-unique-testing-id-1337")
+    val account = CodyAccount(SourcegraphServerPath.from(credentials.serverEndpoint, ""))
+    account.storeToken(credentials.token ?: credentials.redactedToken,)
+    CodyAccount.setActiveAccount(account)
 
     assertNotNull(
         "Unable to start agent in a timely fashion!",