W-20199916: refresh connection before retrieving metadata

src/agentPublisher.ts

@@ -86,8 +86,6 @@ export class AgentPublisher {
    * @returns Promise<PublishAgent> The publish response
    */
   public async publishAgentJson(): Promise<PublishAgent> {
-    // store the access token so we can restore it afterwards
-    const accessToken = this.connection.accessToken;
     // Ensure we use the correct connection for this API call
     await useNamedUserJwt(this.connection);
 
@@ -105,8 +103,9 @@ export class AgentPublisher {
       const url = botId ? `${this.API_URL}/${botId}/versions` : this.API_URL;
       const response = await this.maybeMock.request<PublishAgentJsonResponse>('POST', url, body, this.API_HEADERS);
 
-      // restore the access token
-      this.connection.accessToken = accessToken;
+      // stop using named user jwt access token
+      delete this.connection.accessToken;
+      await this.connection.refreshAuth();
 
       if (response.botId && response.botVersionId) {
         // we've published the AgentJson, now we need to:

test/agentPublisher.test.ts

@@ -110,6 +110,9 @@ describe('AgentPublisher', () => {
       // Mock useNamedUserJwt to return the connection without making HTTP calls
       $$.SANDBOX.stub(utils, 'useNamedUserJwt').resolves(connection);
 
+      // Mock connection.refreshAuth to avoid making HTTP calls during auth refresh
+      $$.SANDBOX.stub(connection, 'refreshAuth').resolves();
+
       // Mock the private methods
       // eslint-disable-next-line @typescript-eslint/no-explicit-any
       const retrieveAgentMetadataStub = $$.SANDBOX.stub(publisher as any, 'retrieveAgentMetadata').resolves();
@@ -131,6 +134,9 @@ describe('AgentPublisher', () => {
       // Mock useNamedUserJwt to return the connection without making HTTP calls
       $$.SANDBOX.stub(utils, 'useNamedUserJwt').resolves(connection);
 
+      // Mock connection.refreshAuth to avoid making HTTP calls during auth refresh
+      $$.SANDBOX.stub(connection, 'refreshAuth').resolves();
+
       // Mock connection.singleRecordQuery to return undefined (no existing bot)
       $$.SANDBOX.stub(connection, 'singleRecordQuery')
         .withArgs("SELECT Id FROM BotDefinition WHERE DeveloperName='test_agent'")
@@ -157,6 +163,9 @@ describe('AgentPublisher', () => {
       // Mock useNamedUserJwt to return the connection without making HTTP calls
       $$.SANDBOX.stub(utils, 'useNamedUserJwt').resolves(connection);
 
+      // Mock connection.refreshAuth to avoid making HTTP calls during auth refresh
+      $$.SANDBOX.stub(connection, 'refreshAuth').resolves();
+
       try {
         await publisher.publishAgentJson();
         expect.fail('Expected error was not thrown');

test/agents.test.ts

@@ -293,6 +293,8 @@ describe('Agents', () => {
 
       // Mock useNamedUserJwt to return the connection without making HTTP calls
       $$.SANDBOX.stub(utils, 'useNamedUserJwt').resolves(connection);
+      // Mock connection.refreshAuth to avoid making HTTP calls during auth refresh
+      $$.SANDBOX.stub(connection, 'refreshAuth').resolves();
 
       try {
         await Agent.publishAgentJson(connection, sfProject, agentJson);