a

Author: Will-ShaoHua

packages/amazonq/test/unit/codewhisperer/util/supplemetalContextUtil.test.ts

@@ -21,7 +21,7 @@ describe('supplementalContextUtil', function () {
 
     beforeEach(async function () {
         testFolder = await TestFolder.create()
-        sinon.stub(FeatureConfigProvider.instance, 'getProjectContextGroup').alwaysReturned('control')
+        sinon.stub(FeatureConfigProvider.instance, 'getProjectContextGroup').returns('control')
     })
 
     afterEach(function () {
@@ -31,8 +31,6 @@ describe('supplementalContextUtil', function () {
     describe('fetchSupplementalContext', function () {
         describe('openTabsContext', function () {
             it('opentabContext should include chunks if non empty', async function () {
-                sinon.stub(FeatureConfigProvider.instance, 'getProjectContextGroup').alwaysReturned('control')
-
                 await toTextEditor('class Foo', 'Foo.java', testFolder.path, { preview: false })
                 await toTextEditor('class Bar', 'Bar.java', testFolder.path, { preview: false })
                 await toTextEditor('class Baz', 'Baz.java', testFolder.path, { preview: false })
@@ -46,8 +44,6 @@ describe('supplementalContextUtil', function () {
             })
 
             it('opentabsContext should filter out empty chunks', async function () {
-                sinon.stub(FeatureConfigProvider.instance, 'getProjectContextGroup').alwaysReturned('control')
-
                 // open 3 files as supplemental context candidate files but none of them have contents
                 await toTextEditor('', 'Foo.java', testFolder.path, { preview: false })
                 await toTextEditor('', 'Bar.java', testFolder.path, { preview: false })