fix test

Author: anthonykim1

src/test/terminals/codeExecution/helper.test.ts

@@ -34,6 +34,7 @@ import { EnvironmentType, PythonEnvironment } from '../../../client/pythonEnviro
 import { CodeExecutionHelper } from '../../../client/terminals/codeExecution/helper';
 import { ICodeExecutionHelper } from '../../../client/terminals/types';
 import { PYTHON_PATH } from '../../common';
+import { ReplType } from '../../../client/repl/types';
 
 const TEST_FILES_PATH = path.join(EXTENSION_ROOT_DIR, 'src', 'test', 'python_files', 'terminalExec');
 
@@ -160,7 +161,7 @@ suite('Terminal - Code Execution Helper', () => {
         };
         jsonParseStub.returns(mockResult);
 
-        const result = await helper.normalizeLines('print("Looks like you are on 3.13")');
+        const result = await helper.normalizeLines('print("Looks like you are on 3.13")', ReplType.terminal);
 
         expect(result).to.equal(`\u001b[200~print("Looks like you are on 3.13")\u001b[201~`);
         jsonParseStub.restore();
@@ -190,7 +191,7 @@ suite('Terminal - Code Execution Helper', () => {
                 actualProcessService.execObservable.apply(actualProcessService, [file, args, options]),
             );
 
-        const result = await helper.normalizeLines('print("Looks like you are not on 3.13")');
+        const result = await helper.normalizeLines('print("Looks like you are not on 3.13")', ReplType.terminal);
 
         expect(result).to.equal('print("Looks like you are not on 3.13")');
         jsonParseStub.restore();
@@ -207,7 +208,7 @@ suite('Terminal - Code Execution Helper', () => {
                 return ({} as unknown) as ObservableExecutionResult<string>;
             });
 
-        await helper.normalizeLines('print("hello")');
+        await helper.normalizeLines('print("hello")', ReplType.terminal);
 
         expect(execArgs).to.contain('normalizeSelection.py');
     });
@@ -228,7 +229,7 @@ suite('Terminal - Code Execution Helper', () => {
             .returns((file, args, options) =>
                 actualProcessService.execObservable.apply(actualProcessService, [file, args, options]),
             );
-        const normalizedCode = await helper.normalizeLines(source);
+        const normalizedCode = await helper.normalizeLines(source, ReplType.terminal);
         const normalizedExpected = expectedSource.replace(/\r\n/g, '\n');
         expect(normalizedCode).to.be.equal(normalizedExpected);
     }

src/test/terminals/codeExecution/smartSend.test.ts

@@ -25,6 +25,7 @@ import { PYTHON_PATH } from '../../common';
 import { Architecture } from '../../../client/common/utils/platform';
 import { ProcessService } from '../../../client/common/process/proc';
 import { l10n } from '../../mocks/vsc';
+import { ReplType } from '../../../client/repl/types';
 
 const TEST_FILES_PATH = path.join(EXTENSION_ROOT_DIR, 'src', 'test', 'python_files', 'terminalExec');
 
@@ -145,7 +146,7 @@ suite('REPL - Smart Send', () => {
             .setup((p) => p.execObservable(TypeMoq.It.isAny(), TypeMoq.It.isAny(), TypeMoq.It.isAny()))
             .returns((file, args, options) => execObservable.apply(actualProcessService, [file, args, options]));
 
-        await codeExecutionHelper.normalizeLines('my_dict = {', wholeFileContent);
+        await codeExecutionHelper.normalizeLines('my_dict = {', ReplType.terminal, wholeFileContent);
 
         commandManager
             .setup((c) => c.executeCommand('cursorMove', TypeMoq.It.isAny()))
@@ -197,7 +198,11 @@ suite('REPL - Smart Send', () => {
             .setup((p) => p.execObservable(TypeMoq.It.isAny(), TypeMoq.It.isAny(), TypeMoq.It.isAny()))
             .returns((file, args, options) => execObservable.apply(actualProcessService, [file, args, options]));
 
-        const actualSmartOutput = await codeExecutionHelper.normalizeLines('my_dict = {', wholeFileContent);
+        const actualSmartOutput = await codeExecutionHelper.normalizeLines(
+            'my_dict = {',
+            ReplType.terminal,
+            wholeFileContent,
+        );
 
         // my_dict = {  <----- smart shift+enter here
         //     "key1": "value1",
@@ -247,7 +252,11 @@ suite('REPL - Smart Send', () => {
             .setup((p) => p.execObservable(TypeMoq.It.isAny(), TypeMoq.It.isAny(), TypeMoq.It.isAny()))
             .returns((file, args, options) => execObservable.apply(actualProcessService, [file, args, options]));
 
-        const actualNonSmartResult = await codeExecutionHelper.normalizeLines('my_dict = {', wholeFileContent);
+        const actualNonSmartResult = await codeExecutionHelper.normalizeLines(
+            'my_dict = {',
+            ReplType.terminal,
+            wholeFileContent,
+        );
         const expectedNonSmartResult = 'my_dict = {\n\n'; // Standard for previous normalization logic
         expect(actualNonSmartResult).to.be.equal(expectedNonSmartResult);
     });
@@ -285,7 +294,7 @@ suite('REPL - Smart Send', () => {
             .setup((p) => p.execObservable(TypeMoq.It.isAny(), TypeMoq.It.isAny(), TypeMoq.It.isAny()))
             .returns((file, args, options) => execObservable.apply(actualProcessService, [file, args, options]));
 
-        await codeExecutionHelper.normalizeLines('my_dict = {', wholeFileContent);
+        await codeExecutionHelper.normalizeLines('my_dict = {', ReplType.terminal, wholeFileContent);
 
         applicationShell
             .setup((a) =>