add tests to ensure credentials are refreshed on expiration

Author: Hweinstock

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

@@ -19,12 +19,54 @@ import { Client } from '@aws-sdk/smithy-client'
 import { extensionVersion } from '../../shared'
 import { assertTelemetry } from '../testUtil'
 import { telemetry } from '../../shared/telemetry'
+import { CredentialsShim } from '../../auth/deprecated/loginManager'
+import { Credentials } from '@aws-sdk/types'
+
+class MockCredentialsShim implements CredentialsShim {
+    public constructor(
+        public credentials: Credentials,
+        public readonly refreshedCredentials: Credentials
+    ) {}
+
+    public expire(): void {
+        this.credentials = {
+            ...this.credentials,
+            expiration: new Date(Date.now() - 1000 * 60 * 60 * 24),
+        }
+    }
+    public async get(): Promise<Credentials> {
+        return this.credentials
+    }
+
+    public async refresh(): Promise<Credentials> {
+        return this.refreshedCredentials
+    }
+}
 
 describe('DefaultAwsClientBuilderV3', function () {
     let builder: AWSClientBuilderV3
+    let fakeContext: FakeAwsContext
+    let mockCredsShim: MockCredentialsShim
+    let oldCreds: Credentials
+    let newCreds: Credentials
 
     beforeEach(function () {
-        builder = new DefaultAWSClientBuilderV3(new FakeAwsContext())
+        fakeContext = new FakeAwsContext()
+        oldCreds = {
+            accessKeyId: 'old',
+            secretAccessKey: 'old',
+            sessionToken: 'old',
+            expiration: new Date(Date.now() + 1000 * 60 * 60 * 24),
+        }
+        newCreds = {
+            accessKeyId: 'new',
+            secretAccessKey: 'new',
+            sessionToken: 'new',
+            expiration: new Date(Date.now() + 1000 * 60 * 60 * 24 * 2),
+        }
+        mockCredsShim = new MockCredentialsShim(oldCreds, newCreds)
+        fakeContext.credentialsShim = mockCredsShim
+        builder = new DefaultAWSClientBuilderV3(fakeContext)
     })
 
     describe('createAndConfigureSdkClient', function () {
@@ -61,6 +103,13 @@ describe('DefaultAwsClientBuilderV3', function () {
 
             assert.strictEqual(service.config.customUserAgent[0][0], 'CUSTOM USER AGENT')
         })
+
+        it('refreshes credentials when they expire', async function () {
+            const service = await builder.createAwsService(Client as any)
+            assert.strictEqual(await service.config.credentials(), oldCreds)
+            mockCredsShim.expire()
+            assert.strictEqual(await service.config.credentials(), newCreds)
+        })
     })
 })