fix: fixing test cases

Author: laileni-aws

packages/core/src/test/shared/vscode/env.test.ts

@@ -365,29 +365,60 @@ VERSION_ID="22.04"
     })
 
     describe('getComputeEnvType', async function () {
+        let fsExistsStub: sinon.SinonStub
+        let fsReadFileStub: sinon.SinonStub
+        let moduleLoadStub: sinon.SinonStub
+
+        beforeEach(function () {
+            // Mock fs module for isAmazonLinux2() calls
+            const fsMock = {
+                existsSync: sandbox.stub().returns(false),
+                readFileSync: sandbox.stub().returns(''),
+            }
+            fsExistsStub = fsMock.existsSync
+            fsReadFileStub = fsMock.readFileSync
+
+            // Stub Module._load to intercept require calls
+            const Module = require('module')
+            moduleLoadStub = sandbox.stub(Module, '_load').callThrough()
+            moduleLoadStub.withArgs('fs').returns(fsMock)
+        })
+
         it('cloudDesktop', async function () {
             sandbox.stub(process, 'platform').value('linux')
             sandbox.stub(vscode.env, 'remoteName').value('ssh-remote')
+            sandbox.stub(globals, 'isWeb').returns(false)
             stubOsVersion('5.10.220-188.869.amzn2int.x86_64')
             sandbox.stub(ChildProcess.prototype, 'run').resolves({ exitCode: 0 } as any)
 
+            // Mock fs to return false so it falls back to kernel check (which should return true for AL2)
+            fsExistsStub.returns(false)
+
             assert.deepStrictEqual(await getComputeEnvType(), 'cloudDesktop-amzn')
         })
 
         it('ec2-internal', async function () {
             sandbox.stub(process, 'platform').value('linux')
             sandbox.stub(vscode.env, 'remoteName').value('ssh-remote')
+            sandbox.stub(globals, 'isWeb').returns(false)
             stubOsVersion('5.10.220-188.869.amzn2int.x86_64')
             sandbox.stub(ChildProcess.prototype, 'run').resolves({ exitCode: 1 } as any)
 
+            // Mock fs to return false so it falls back to kernel check (which should return true for AL2)
+            fsExistsStub.returns(false)
+
             assert.deepStrictEqual(await getComputeEnvType(), 'ec2-amzn')
         })
 
         it('ec2', async function () {
             sandbox.stub(process, 'platform').value('linux')
             sandbox.stub(vscode.env, 'remoteName').value('ssh-remote')
+            sandbox.stub(globals, 'isWeb').returns(false)
             stubOsVersion('5.10.220-188.869.NOT_INTERNAL.x86_64')
 
+            // Mock fs to return false so it falls back to kernel check (which should return false for non-AL2)
+            fsExistsStub.returns(false)
+
             assert.deepStrictEqual(await getComputeEnvType(), 'ec2')
         })
     })