test

Author: Will-ShaoHua

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

@@ -40,7 +40,7 @@ describe('crossFileContextUtil', function () {
         })
 
         it('opentabs context should fetch 3 chunks and each chunk should contains 50 lines', async function () {
-            sinon.stub(FeatureConfigProvider.instance, 'isNewProjectContextGroup').alwaysReturned(false)
+            sinon.stub(FeatureConfigProvider.instance, 'getProjectContextGroup').alwaysReturned('control')
             await toTextEditor(aStringWithLineCount(200), 'CrossFile.java', tempFolder, { preview: false })
             const myCurrentEditor = await toTextEditor('', 'TargetFile.java', tempFolder, {
                 preview: false,
@@ -212,7 +212,7 @@ describe('crossFileContextUtil', function () {
 
         fileExtLists.forEach((fileExt) => {
             it('should be non empty', async function () {
-                sinon.stub(FeatureConfigProvider.instance, 'isNewProjectContextGroup').alwaysReturned(false)
+                sinon.stub(FeatureConfigProvider.instance, 'getProjectContextGroup').alwaysReturned('control')
                 const editor = await toTextEditor('content-1', `file-1.${fileExt}`, tempFolder)
                 await toTextEditor('content-2', `file-2.${fileExt}`, tempFolder, { preview: false })
                 await toTextEditor('content-3', `file-3.${fileExt}`, tempFolder, { preview: false })

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, 'isNewProjectContextGroup').alwaysReturned(false)
+        sinon.stub(FeatureConfigProvider.instance, 'getProjectContextGroup').alwaysReturned('control')
     })
 
     afterEach(function () {

packages/core/src/test/shared/featureConfig.test.ts

@@ -94,4 +94,44 @@ describe('FeatureConfigProvider', () => {
     it('should test feature-does-not-exist as disabled', async () => {
         assert.strictEqual(FeatureConfigProvider.isEnabled('feature-does-not-exist' as FeatureName), false)
     })
+
+    describe('getProjectContextGroup', function () {
+        beforeEach(function () {
+            sinon.restore()
+        })
+
+        afterEach(function () {
+            sinon.restore()
+        })
+
+        const cases: { variationName: string; expected: string }[] = [
+            { variationName: 'CONTROL', expected: 'control' },
+            { variationName: 'TREATMENT_1', expected: 't1' },
+            { variationName: 'TREATMENT_2', expected: 't2' },
+        ]
+
+        for (const tuple of cases) {
+            it(`should return ${tuple.expected} when variation name is ${tuple.variationName}`, async function () {
+                const clientSpy = await createSpyClient()
+                sinon.stub(clientSpy, 'listFeatureEvaluations').returns({
+                    promise: () =>
+                        Promise.resolve({
+                            $response: {
+                                requestId: '',
+                            },
+                            featureEvaluations: [
+                                {
+                                    feature: 'ProjectContextV2',
+                                    variation: tuple.variationName,
+                                    value: { stringValue: 'foo' },
+                                },
+                            ],
+                        }),
+                } as Request<ListFeatureEvaluationsResponse, AWSError>)
+                await FeatureConfigProvider.instance.fetchFeatureConfigs()
+
+                assert.strictEqual(FeatureConfigProvider.instance.getProjectContextGroup(), tuple.expected)
+            })
+        }
+    })
 })