test(notifications): integ test from hosted endpoints

Author: tomcat323

packages/core/src/notifications/rules.ts

@@ -8,6 +8,7 @@ import * as semver from 'semver'
 import globals from '../shared/extensionGlobals'
 import { ConditionalClause, RuleContext, DisplayIf, CriteriaCondition, ToolkitNotification, AuthState } from './types'
 import { getComputeEnvType, getOperatingSystem } from '../shared/telemetry/util'
+import { isAutomation } from '../shared/vscode/env'
 
 /**
  * Evaluates if a given version fits into the parameters specified by a notification, e.g:
@@ -69,7 +70,8 @@ export class RuleEngine {
 
     private evaluate(condition: DisplayIf): boolean {
         const currentExt = globals.context.extension.id
-        if (condition.extensionId !== currentExt) {
+        // if in test, skip the extension id check since its fake
+        if (condition.extensionId !== currentExt && !isAutomation()) {
             return false
         }
 

packages/core/src/testInteg/notifications/notifications.test.ts

@@ -0,0 +1,115 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0 fort
+ */
+
+import { RemoteFetcher } from '../../notifications/controller'
+import { NotificationsController } from '../../notifications/controller'
+import { NotificationsNode } from '../../notifications/panelNode'
+import { RuleEngine } from '../../notifications/rules'
+import { NotificationsState, RuleContext } from '../../notifications/types'
+import globals from '../../shared/extensionGlobals'
+import assert from 'assert'
+import { VSCODE_EXTENSION_ID } from '../../shared/extensions'
+import sinon from 'sinon'
+
+describe('Notifications Integration Test', function () {
+    let fetcher: RemoteFetcher
+    let panelNode: NotificationsNode
+    let controller: NotificationsController
+
+    function getEngine(activeExtensions: string[], connected: boolean): RuleEngine {
+        const authState = connected ? 'connected' : 'notConnected'
+
+        const ruleContext: RuleContext = {
+            ideVersion: '1.83.0',
+            extensionVersion: '1.20.0',
+            os: 'LINUX',
+            computeEnv: 'local',
+            authTypes: ['builderId'],
+            authRegions: ['us-east-2'],
+            authStates: [authState],
+            authScopes: ['codewhisperer:completions', 'codewhisperer:analysis'],
+            installedExtensions: Object.keys(VSCODE_EXTENSION_ID),
+            activeExtensions: activeExtensions,
+        }
+
+        return new RuleEngine(ruleContext)
+    }
+
+    beforeEach(async function () {
+        panelNode = NotificationsNode.instance
+        // fetch from test host file folder
+        fetcher = new RemoteFetcher(
+            'https://idetoolkits-hostedfiles.amazonaws.com/Notifications/integ/VSCode/startup/1.x.json',
+            'https://idetoolkits-hostedfiles.amazonaws.com/Notifications/integ/VSCode/emergency/1.x.json'
+        )
+        controller = new NotificationsController(panelNode, fetcher)
+    })
+
+    // Clear all global states after each test
+    afterEach(async function () {
+        await globals.globalState.update(controller.storageKey, undefined)
+    })
+
+    it('Receive notifications polling from endpoint', async function () {
+        const engine = getEngine([VSCODE_EXTENSION_ID.amazonq], true)
+
+        await controller.pollForStartUp(engine)
+        await controller.pollForEmergencies(engine)
+
+        // Verify that notifications are stored in the state
+        const state = globals.globalState.get<NotificationsState>(controller.storageKey)
+        assert.ok(state, 'State should be defined')
+        assert.ok(state.startUp.payload?.notifications, 'StartUp received')
+        assert.strictEqual(
+            state?.startUp.payload?.notifications.length,
+            2,
+            'There should be 2 startup notifications stored'
+        )
+        assert.ok(state?.emergency.payload?.notifications, 'Emergency received')
+        assert.strictEqual(
+            state?.emergency.payload?.notifications.length,
+            2,
+            'There should be 2 emergency notifications stored'
+        )
+    })
+
+    it('Display notification according to criterias', async function () {
+        const engine = getEngine([VSCODE_EXTENSION_ID.amazonq], false)
+        await controller.pollForEmergencies(engine)
+
+        // Verify that only the not authed notification is displayed
+        const displayedNotifications = panelNode.emergencyNotifications
+        assert.strictEqual(displayedNotifications.length, 1, 'Only one notification displayed"')
+        assert.strictEqual(
+            displayedNotifications[0].id,
+            'emergency1',
+            'The displayed notification have the ID "emergency1"'
+        )
+    })
+
+    it('Should trigger onReceive only for newly received notifications', async function () {
+        const engine = getEngine([VSCODE_EXTENSION_ID.amazonq], true)
+        const onReceiveSpy = sinon.spy(panelNode, 'onReceiveNotifications')
+
+        // Simulate alreadying receving a notification
+        const existingNotificationID = 'startup1'
+        const state = globals.globalState.get<NotificationsState>(controller.storageKey)
+        if (state) {
+            state.newlyReceived.push(existingNotificationID)
+            await globals.globalState.update(controller.storageKey, state)
+        }
+
+        // Poll for new notification
+        await controller.pollForStartUp(engine)
+
+        /**
+         * Since we simulated startup1 is already received
+         * onReceived should be called exactly once for startup2
+         */
+        assert.ok(onReceiveSpy.calledOnce, 'only one new notification received')
+
+        onReceiveSpy.restore()
+    })
+})