From 25a26eea7c1f5518e464f97d8a04f6d936e67871 Mon Sep 17 00:00:00 2001
From: Anthony Kim <anthonykim@microsoft.com>
Date: Fri, 21 Jun 2024 13:31:11 -0700
Subject: [PATCH] fix

---
 src/test/providers/repl.unit.test.ts | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/src/test/providers/repl.unit.test.ts b/src/test/providers/repl.unit.test.ts
index f9de026db51e..72adfa95a4a0 100644
--- a/src/test/providers/repl.unit.test.ts
+++ b/src/test/providers/repl.unit.test.ts
@@ -26,7 +26,6 @@ suite('REPL Provider', () => {
     let activeResourceService: TypeMoq.IMock<IActiveResourceService>;
     let replProvider: ReplProvider;
     let interpreterService: TypeMoq.IMock<IInterpreterService>;
-    const executionService = TypeMoq.Mock.ofType<ICodeExecutionService>();
     setup(() => {
         serviceContainer = TypeMoq.Mock.ofType<IServiceContainer>();
         commandManager = TypeMoq.Mock.ofType<ICommandManager>();
@@ -36,12 +35,9 @@ suite('REPL Provider', () => {
         activeResourceService = TypeMoq.Mock.ofType<IActiveResourceService>();
         serviceContainer.setup((c) => c.get(ICommandManager)).returns(() => commandManager.object);
         serviceContainer.setup((c) => c.get(IWorkspaceService)).returns(() => workspace.object);
-        // serviceContainer
-        //     .setup((c) => c.get(ICodeExecutionService, TypeMoq.It.isValue('repl')))
-        //     .returns(() => codeExecutionService.object);
         serviceContainer
             .setup((s) => s.get(TypeMoq.It.isValue(ICodeExecutionService), TypeMoq.It.isValue('standard')))
-            .returns(() => executionService.object);
+            .returns(() => codeExecutionService.object);
         serviceContainer.setup((c) => c.get(IDocumentManager)).returns(() => documentManager.object);
         serviceContainer.setup((c) => c.get(IActiveResourceService)).returns(() => activeResourceService.object);
         interpreterService = TypeMoq.Mock.ofType<IInterpreterService>();
@@ -103,7 +99,7 @@ suite('REPL Provider', () => {
         await commandHandler!.call(replProvider);
 
         serviceContainer.verify(
-            (c) => c.get(TypeMoq.It.isValue(ICodeExecutionService), TypeMoq.It.isAny()),
+            (c) => c.get(TypeMoq.It.isValue(ICodeExecutionService), TypeMoq.It.isValue('standard')),
             TypeMoq.Times.once(),
         );
         codeExecutionService.verify((c) => c.initializeRepl(TypeMoq.It.isValue(resource)), TypeMoq.Times.once());