Refresh PR tree when repos are discovered after initial load (#7989)

Author: kycutler

src/test/view/prsTree.test.ts

@@ -141,6 +141,40 @@ describe('GitHub Pull Requests view', function () {
 		);
 	});
 
+	it('refreshes tree when GitHub repositories are discovered in existing folder manager', async function () {
+		const repository = new MockRepository();
+		repository.addRemote('origin', 'git@github.com:aaa/bbb');
+		const folderManager = new FolderRepositoryManager(0, context, repository, telemetry, new GitApiImpl(reposManager), credentialStore, createPrHelper, mockThemeWatcher);
+		sinon.stub(folderManager, 'getPullRequestDefaults').returns(Promise.resolve({ owner: 'aaa', repo: 'bbb', base: 'main' }));
+		reposManager.insertFolderManager(folderManager);
+		provider.initialize([], mockNotificationsManager as NotificationsManager);
+
+		// Initially no children because no GitHub repositories are loaded yet
+		let rootNodes = await provider.getChildren();
+		assert.strictEqual(rootNodes.length, 0);
+
+		// Listen to the prsTreeModel's onDidChangeData event which is what actually drives the tree refresh
+		const onDidChangeDataSpy = sinon.spy();
+		provider.prsTreeModel.onDidChangeData(onDidChangeDataSpy);
+
+		// Simulate GitHub repositories being discovered (as happens when remotes load after activation)
+		sinon.stub(credentialStore, 'isAuthenticated').returns(true);
+		await folderManager.updateRepositories();
+
+		// Verify that the tree model's data change event was triggered
+		assert(onDidChangeDataSpy.calledWith(folderManager),
+			'Tree model should fire data change event with the folder manager when GitHub repositories are discovered');
+
+		// Verify tree now has content
+		rootNodes = await provider.getChildren();
+		const treeItems = await Promise.all(rootNodes.map(node => node.getTreeItem()));
+		assert.deepStrictEqual(
+			treeItems.map(n => n.label),
+			['Copilot on My Behalf', 'Local Pull Request Branches', 'Waiting For My Review', 'Created By Me', 'All Open'],
+			'Tree should display categories after GitHub repositories are discovered',
+		);
+	});
+
 	describe('Local Pull Request Branches', function () {
 		it('creates a node for each local pull request', async function () {
 			const url = 'git@github.com:aaa/bbb';

src/view/prsTreeModel.ts

@@ -118,6 +118,10 @@ export class PrsTreeModel extends Disposable {
 			}
 		}));
 
+		this._register(this._reposManager.onDidChangeAnyGitHubRepository((folderManager) => {
+			this._onDidChangeData.fire(folderManager);
+		}));
+
 		const expandedQueries = this._context.workspaceState.get(EXPANDED_QUERIES_STATE, undefined);
 		if (expandedQueries) {
 			this._expandedQueries = new Set(expandedQueries);