From 5d9bbbf53389c6cd461f4da0c81ca4f073644e9a Mon Sep 17 00:00:00 2001
From: Kartik Raj <karraj@microsoft.com>
Date: Tue, 21 Nov 2023 14:18:03 -0800
Subject: [PATCH] Fix tests

---
 src/client/interpreter/autoSelection/index.ts |  2 +-
 .../autoSelection/index.unit.test.ts          | 19 ++++++++++++++++++-
 2 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/src/client/interpreter/autoSelection/index.ts b/src/client/interpreter/autoSelection/index.ts
index 9259a76415d9..4310374fc00f 100644
--- a/src/client/interpreter/autoSelection/index.ts
+++ b/src/client/interpreter/autoSelection/index.ts
@@ -186,7 +186,7 @@ export class InterpreterAutoSelectionService implements IInterpreterAutoSelectio
 
     private getAutoSelectionQueriedOnceState(): IPersistentState<boolean | undefined> {
         const key = `autoSelectionInterpretersQueriedOnce`;
-        return this.stateFactory.createWorkspacePersistentState(key, undefined);
+        return this.stateFactory.createGlobalPersistentState(key, undefined);
     }
 
     /**
diff --git a/src/test/interpreters/autoSelection/index.unit.test.ts b/src/test/interpreters/autoSelection/index.unit.test.ts
index 914c67bb4a65..6c5473546614 100644
--- a/src/test/interpreters/autoSelection/index.unit.test.ts
+++ b/src/test/interpreters/autoSelection/index.unit.test.ts
@@ -65,7 +65,7 @@ suite('Interpreters - Auto Selection', () => {
         helper = mock(InterpreterHelper);
         proxy = mock(InterpreterAutoSelectionProxyService);
         interpreterService = mock(InterpreterService);
-        experimentService = mock(IExperimentService);
+        experimentService = mock<IExperimentService>();
         when(experimentService.inExperimentSync(anything())).thenReturn(false);
 
         const interpreterComparer = new EnvironmentTypeComparer(instance(helper));
@@ -144,6 +144,12 @@ suite('Interpreters - Auto Selection', () => {
                     undefined,
                 ),
             ).thenReturn(instance(state));
+            when(
+                stateFactory.createGlobalPersistentState<PythonEnvironment | undefined>(
+                    'autoSelectionInterpretersQueriedOnce',
+                    undefined,
+                ),
+            ).thenReturn(instance(state));
             when(workspaceService.getWorkspaceFolderIdentifier(anything(), '')).thenReturn('workspaceIdentifier');
 
             autoSelectionService.onDidChangeAutoSelectedInterpreter(() => {
@@ -212,6 +218,13 @@ suite('Interpreters - Auto Selection', () => {
 
         test('getInterpreters is called with ignoreCache at true if there is no value set in the workspace persistent state', async () => {
             const interpreterComparer = new EnvironmentTypeComparer(instance(helper));
+
+            const globalQueriedState = mock(PersistentState) as PersistentState<boolean | undefined>;
+            when(globalQueriedState.value).thenReturn(true);
+            when(stateFactory.createGlobalPersistentState<boolean | undefined>(anyString(), undefined)).thenReturn(
+                instance(globalQueriedState),
+            );
+
             const queryState = mock(PersistentState) as PersistentState<boolean | undefined>;
 
             when(queryState.value).thenReturn(undefined);
@@ -393,6 +406,10 @@ suite('Interpreters - Auto Selection', () => {
         when(stateFactory.createWorkspacePersistentState<boolean | undefined>(anyString(), undefined)).thenReturn(
             instance(queryState),
         );
+        when(queryState.value).thenReturn(undefined);
+        when(stateFactory.createGlobalPersistentState<boolean | undefined>(anyString(), undefined)).thenReturn(
+            instance(queryState),
+        );
 
         let initialize = false;
         let eventFired = false;