test(amazonq): small improvement to developer profiles test

Author: patrickvbeurden

server/aws-lsp-codewhisperer/src/shared/amazonQServiceManager/qDeveloperProfiles.test.ts

@@ -1,5 +1,5 @@
 import * as assert from 'assert'
-import { StubbedInstance, stubInterface } from 'ts-sinon'
+import sinon, { StubbedInstance, stubInterface } from 'ts-sinon'
 import { CodeWhispererServiceToken } from '../codeWhispererService'
 import { SsoConnectionType } from '../utils'
 import { AWSInitializationOptions, Logging } from '@aws/language-server-runtimes/server-interface'
@@ -46,11 +46,6 @@ describe('ListAllAvailableProfiles Handler', () => {
         $response: {} as any,
     }
 
-    const listAvailableProfilesResponseWithNextToken = {
-        ...listAvailableProfilesResponse,
-        nextToken: 'some-random-next-token',
-    }
-
     beforeEach(() => {
         logging = stubInterface<Logging>()
         codeWhispererService = stubInterface<CodeWhispererServiceToken>()
@@ -86,6 +81,12 @@ describe('ListAllAvailableProfiles Handler', () => {
 
     describe('Pagination', () => {
         const MAX_EXPECTED_PAGES = 10
+        const SOME_NEXT_TOKEN = 'some-random-next-token'
+
+        const listAvailableProfilesResponseWithNextToken = {
+            ...listAvailableProfilesResponse,
+            nextToken: SOME_NEXT_TOKEN,
+        }
 
         it('should paginate if nextToken is defined', async () => {
             const EXPECTED_CALLS = 3
@@ -104,7 +105,11 @@ describe('ListAllAvailableProfiles Handler', () => {
                 endpoints: SOME_AWS_Q_ENDPOINT,
             })
 
-            assert.strictEqual(codeWhispererService.listAvailableProfiles.callCount, EXPECTED_CALLS)
+            sinon.assert.calledThrice(codeWhispererService.listAvailableProfiles)
+            assert.strictEqual(codeWhispererService.listAvailableProfiles.firstCall.args[0].nextToken, undefined)
+            assert.strictEqual(codeWhispererService.listAvailableProfiles.secondCall.args[0].nextToken, SOME_NEXT_TOKEN)
+            assert.strictEqual(codeWhispererService.listAvailableProfiles.thirdCall.args[0].nextToken, SOME_NEXT_TOKEN)
+
             assert.deepStrictEqual(profiles, Array(EXPECTED_CALLS).fill(EXPECTED_DEVELOPER_PROFILES_LIST[0]))
         })
 
@@ -118,6 +123,13 @@ describe('ListAllAvailableProfiles Handler', () => {
             })
 
             assert.strictEqual(codeWhispererService.listAvailableProfiles.callCount, MAX_EXPECTED_PAGES)
+            codeWhispererService.listAvailableProfiles.getCalls().forEach((call, index) => {
+                if (index === 0) {
+                    assert.strictEqual(call.args[0].nextToken, undefined)
+                } else {
+                    assert.strictEqual(call.args[0].nextToken, SOME_NEXT_TOKEN)
+                }
+            })
             assert.deepStrictEqual(profiles, Array(MAX_EXPECTED_PAGES).fill(EXPECTED_DEVELOPER_PROFILES_LIST[0]))
         })
     })