fix authUtil test

yuxianrz · yuxianrz · commit df3d5fcc23bf · 2025-08-04T16:56:15.000-04:00
diff --git a/packages/core/src/test/amazonqDoc/utils.ts b/packages/core/src/test/amazonqDoc/utils.ts
@@ -108,6 +108,7 @@ export function createMockAuthUtil(sandbox: sinon.SinonSandbox) {
     const mockLspAuth: Partial<LanguageClientAuth> = {
         registerSsoTokenChangedHandler: sinon.stub().resolves(),
         registerStsCredentialChangedHandler: sinon.stub().resolves(),
+        registerGetMfaCodeHandler: sinon.stub().resolves(),
     }
     AuthUtil.create(mockLspAuth as LanguageClientAuth)
     sandbox.stub(AuthUtil.instance.regionProfileManager, 'onDidChangeRegionProfile').resolves()
diff --git a/packages/core/src/test/testAuthUtil.ts b/packages/core/src/test/testAuthUtil.ts
@@ -44,6 +44,7 @@ export async function createTestAuthUtil() {
     const mockLspAuth: Partial<LanguageClientAuth> = {
         registerSsoTokenChangedHandler: sinon.stub().resolves(),
         registerStsCredentialChangedHandler: sinon.stub().resolves(),
+        registerGetMfaCodeHandler: sinon.stub().resolves(),
         updateSsoProfile: sinon.stub().resolves(),
         getSsoToken: sinon.stub().resolves(fakeToken),
         getIamCredential: sinon.stub().resolves(fakeCredential),

Original file line number	Diff line number	Diff line change
`@@ -108,6 +108,7 @@ export function createMockAuthUtil(sandbox: sinon.SinonSandbox) {`
`108`	`108`	`const mockLspAuth: Partial<LanguageClientAuth> = {`
`109`	`109`	`registerSsoTokenChangedHandler: sinon.stub().resolves(),`
`110`	`110`	`registerStsCredentialChangedHandler: sinon.stub().resolves(),`
	`111`	`+ registerGetMfaCodeHandler: sinon.stub().resolves(),`
`111`	`112`	`}`
`112`	`113`	`AuthUtil.create(mockLspAuth as LanguageClientAuth)`
`113`	`114`	`sandbox.stub(AuthUtil.instance.regionProfileManager, 'onDidChangeRegionProfile').resolves()`