Fix tests

Author: alexr00

src/test/github/copilotRemoteAgent.test.ts

@@ -19,7 +19,6 @@ import { GitHubRemote } from '../../common/remote';
 import { Protocol } from '../../common/protocol';
 import { GitHubServerType } from '../../common/authentication';
 import { ReposManagerState } from '../../github/folderRepositoryManager';
-import { CopilotPRStatus } from '../../common/copilot';
 import { GitApiImpl } from '../../api/api1';
 import { MockPrsTreeModel } from '../mocks/mockPRsTreeModel';
 import { PrsTreeModel } from '../../view/prsTreeModel';

src/test/mocks/mockPRsTreeModel.ts

@@ -3,28 +3,39 @@
  *  Licensed under the MIT License. See License.txt in the project root for license information.
  *--------------------------------------------------------------------------------------------*/
 
-import { Event, Disposable } from "vscode";
+import { Event, Disposable, EventEmitter } from "vscode";
 import { CopilotPRStatus } from "../../common/copilot";
-import { CopilotStateModel } from "../../github/copilotPrWatcher";
+import { CodingAgentPRAndStatus, CopilotStateModel } from "../../github/copilotPrWatcher";
 import { FolderRepositoryManager, ItemsResponseResult } from "../../github/folderRepositoryManager";
 import { PullRequestChangeEvent } from "../../github/githubRepository";
 import { PullRequestModel } from "../../github/pullRequestModel";
 import { PRStatusChange, PrsTreeModel } from "../../view/prsTreeModel";
 import { TreeNode } from "../../view/treeNodes/treeNode";
 
 export class MockPrsTreeModel implements Partial<PrsTreeModel> {
-	public onDidChangePrStatus: Event<string[]>;
-	public onDidChangeData: Event<void | FolderRepositoryManager | PullRequestChangeEvent[]>;
+	onDidChangeCopilotStates: Event<void> = new EventEmitter<void>().event;
+	onDidChangeCopilotNotifications: Event<PullRequestModel[]> = new EventEmitter<PullRequestModel[]>().event;
+	clearCopilotCaches(): false | undefined {
+		throw new Error("Method not implemented.");
+	}
+	async refreshCopilotStateChanges(clearCache?: boolean): Promise<boolean> {
+		return false;
+	}
+	getCopilotPullRequests(clearCache?: boolean): Promise<CodingAgentPRAndStatus[]> {
+		throw new Error("Method not implemented.");
+	}
+	public onDidChangePrStatus: Event<string[]> = new EventEmitter<string[]>().event;
+	public onDidChangeData: Event<void | FolderRepositoryManager | PullRequestChangeEvent[]> = new EventEmitter<void>().event;
 	public onLoaded: Event<void>;
 	public copilotStateModel: CopilotStateModel;
 	public updateExpandedQueries(element: TreeNode, isExpanded: boolean): void {
 		throw new Error("Method not implemented.");
 	}
 	get expandedQueries(): Set<string> | undefined {
-		throw new Error("Method not implemented.");
+		return new Set<string>(['All Open']);
 	}
 	get hasLoaded(): boolean {
-		throw new Error("Method not implemented.");
+		return true;
 	}
 	set hasLoaded(value: boolean) {
 		throw new Error("Method not implemented.");
@@ -41,8 +52,12 @@ export class MockPrsTreeModel implements Partial<PrsTreeModel> {
 	public clearCache(silent?: boolean): void {
 		throw new Error("Method not implemented.");
 	}
-	getLocalPullRequests(folderRepoManager: FolderRepositoryManager, update?: boolean): Promise<ItemsResponseResult<PullRequestModel>> {
-		throw new Error("Method not implemented.");
+	async getLocalPullRequests(folderRepoManager: FolderRepositoryManager, update?: boolean): Promise<ItemsResponseResult<PullRequestModel>> {
+		return {
+			hasMorePages: false,
+			items: this._localPullRequests,
+			hasUnsearchedRepositories: false
+		};
 	}
 	getPullRequestsForQuery(folderRepoManager: FolderRepositoryManager, fetchNextPage: boolean, query: string): Promise<ItemsResponseResult<PullRequestModel>> {
 		throw new Error("Method not implemented.");
@@ -51,10 +66,10 @@ export class MockPrsTreeModel implements Partial<PrsTreeModel> {
 		throw new Error("Method not implemented.");
 	}
 	getCopilotNotificationsCount(owner: string, repo: string): number {
-		throw new Error("Method not implemented.");
+		return 0;
 	}
 	get copilotNotificationsCount(): number {
-		throw new Error("Method not implemented.");
+		return 0;
 	}
 	clearAllCopilotNotifications(owner?: string, repo?: string): void {
 		throw new Error("Method not implemented.");
@@ -86,4 +101,8 @@ export class MockPrsTreeModel implements Partial<PrsTreeModel> {
 		throw new Error("Method not implemented.");
 	}
 
+	private _localPullRequests: PullRequestModel[] = [];
+	addLocalPullRequest(pr: PullRequestModel): void {
+		this._localPullRequests.push(pr);
+	}
 }

src/test/view/prsTree.test.ts

@@ -50,13 +50,12 @@ describe('GitHub Pull Requests view', function () {
 	let mockThemeWatcher: MockThemeWatcher;
 	let gitAPI: GitApiImpl;
 	let mockNotificationsManager: MockNotificationManager;
-	let mockPrsTreeModel: PrsTreeModel;
+	let prsTreeModel: PrsTreeModel;
 
 	beforeEach(function () {
 		sinon = createSandbox();
 		MockCommandRegistry.install(sinon);
 		mockThemeWatcher = new MockThemeWatcher();
-		mockPrsTreeModel = new MockPrsTreeModel() as unknown as PrsTreeModel;
 
 		context = new MockExtensionContext();
 
@@ -65,10 +64,11 @@ describe('GitHub Pull Requests view', function () {
 			credentialStore,
 			telemetry,
 		);
+		prsTreeModel = new PrsTreeModel(telemetry, reposManager, context);
 		credentialStore = new CredentialStore(telemetry, context);
 		gitAPI = new GitApiImpl(reposManager);
-		copilotManager = new CopilotRemoteAgentManager(credentialStore, reposManager, telemetry, context, gitAPI, mockPrsTreeModel);
-		provider = new PullRequestsTreeDataProvider(mockPrsTreeModel, telemetry, context, reposManager, copilotManager);
+		copilotManager = new CopilotRemoteAgentManager(credentialStore, reposManager, telemetry, context, gitAPI, prsTreeModel);
+		provider = new PullRequestsTreeDataProvider(prsTreeModel, telemetry, context, reposManager, copilotManager);
 		mockNotificationsManager = new MockNotificationManager();
 		createPrHelper = new CreatePullRequestHelper();