Fix some tests (#8656)

* Fix some tests

* Improve test

Author: alexr00

src/test/issues/stateManager.test.ts

@@ -161,9 +161,17 @@ describe('StateManager branch behavior with useBranchForIssues setting', functio
 				credentialStore: { isAnyAuthenticated: () => true, getCurrentUser: async () => ({ login: 'testuser' }) },
 			} as any, mockContext);
 
+			(stateManager as any)._queries = [{ label: 'Test', query: 'is:open assignee:@me repo:owner/repo ', groupBy: [] }];
+
 			// Manually trigger the setIssueData flow
 			await (stateManager as any).setIssueData(mockFolderManager);
 
+			// Await the collection promise so setIssues completes
+			const collection = stateManager.getIssueCollection(mockUri);
+			const testQueryPromise = collection.get('Test');
+			assert.ok(testQueryPromise, 'Expected issue collection to contain the \'Test\' query label');
+			await testQueryPromise;
+
 			// If we get here without assertion failures in getIssues, the test passed
 		} finally {
 			vscode.workspace.getConfiguration = originalGetConfiguration;
@@ -251,6 +259,8 @@ describe('StateManager branch behavior with useBranchForIssues setting', functio
 				credentialStore: { isAnyAuthenticated: () => true, getCurrentUser: async () => ({ login: 'testuser' }) },
 			} as any, mockContext);
 
+			(sm as any)._queries = [{ label: 'Test', query: 'is:open repo:owner/repo', groupBy: [] }];
+
 			await (sm as any).setIssueData(mockFolderManager);
 
 			// Verify that the promises in issueCollection resolve (not reject)