@@ -313,32 +313,52 @@ describe('Test getStoredInstallationId', () => {
313313 it ( 'returns stored installation ID when configured' , async ( ) => {
314314 const installationIdParam = `/actions-runner/${ ENVIRONMENT } /github_app_installation_id` ;
315315 process . env . PARAMETER_GITHUB_APP_INSTALLATION_ID_NAME = installationIdParam ;
316- mockedGet . mockResolvedValueOnce ( GITHUB_APP_ID ) . mockResolvedValueOnce ( b64 ) . mockResolvedValueOnce ( '12345' ) ;
316+ mockedGetParameters . mockResolvedValueOnce (
317+ new Map ( [
318+ [ PARAMETER_GITHUB_APP_ID_NAME , GITHUB_APP_ID ] ,
319+ [ PARAMETER_GITHUB_APP_KEY_BASE64_NAME , b64 ] ,
320+ [ installationIdParam , '12345' ] ,
321+ ] ) ,
322+ ) ;
317323
318324 const result = await getStoredInstallationId ( 0 ) ;
319325 expect ( result ) . toBe ( 12345 ) ;
320- expect ( getParameter ) . toHaveBeenCalledWith ( installationIdParam ) ;
321326 } ) ;
322327
323328 it ( 'returns undefined when installation ID param is empty' , async ( ) => {
324329 process . env . PARAMETER_GITHUB_APP_INSTALLATION_ID_NAME = '' ;
325- mockedGet . mockResolvedValueOnce ( GITHUB_APP_ID ) . mockResolvedValueOnce ( b64 ) ;
330+ mockedGetParameters . mockResolvedValueOnce (
331+ new Map ( [
332+ [ PARAMETER_GITHUB_APP_ID_NAME , GITHUB_APP_ID ] ,
333+ [ PARAMETER_GITHUB_APP_KEY_BASE64_NAME , b64 ] ,
334+ ] ) ,
335+ ) ;
326336
327337 const result = await getStoredInstallationId ( 0 ) ;
328338 expect ( result ) . toBeUndefined ( ) ;
329339 } ) ;
330340
331341 it ( 'returns undefined when env var is not set' , async ( ) => {
332342 delete process . env . PARAMETER_GITHUB_APP_INSTALLATION_ID_NAME ;
333- mockedGet . mockResolvedValueOnce ( GITHUB_APP_ID ) . mockResolvedValueOnce ( b64 ) ;
343+ mockedGetParameters . mockResolvedValueOnce (
344+ new Map ( [
345+ [ PARAMETER_GITHUB_APP_ID_NAME , GITHUB_APP_ID ] ,
346+ [ PARAMETER_GITHUB_APP_KEY_BASE64_NAME , b64 ] ,
347+ ] ) ,
348+ ) ;
334349
335350 const result = await getStoredInstallationId ( 0 ) ;
336351 expect ( result ) . toBeUndefined ( ) ;
337352 } ) ;
338353
339354 it ( 'returns undefined for out-of-bounds appIndex' , async ( ) => {
340355 process . env . PARAMETER_GITHUB_APP_INSTALLATION_ID_NAME = '' ;
341- mockedGet . mockResolvedValueOnce ( GITHUB_APP_ID ) . mockResolvedValueOnce ( b64 ) ;
356+ mockedGetParameters . mockResolvedValueOnce (
357+ new Map ( [
358+ [ PARAMETER_GITHUB_APP_ID_NAME , GITHUB_APP_ID ] ,
359+ [ PARAMETER_GITHUB_APP_KEY_BASE64_NAME , b64 ] ,
360+ ] ) ,
361+ ) ;
342362
343363 const result = await getStoredInstallationId ( 99 ) ;
344364 expect ( result ) . toBeUndefined ( ) ;
@@ -355,12 +375,15 @@ describe('Test getStoredInstallationId', () => {
355375 process . env . PARAMETER_GITHUB_APP_KEY_BASE64_NAME = `${ app1KeyParam } :${ app2KeyParam } ` ;
356376 process . env . PARAMETER_GITHUB_APP_INSTALLATION_ID_NAME = `:${ app2InstallParam } ` ;
357377
358- mockedGet
359- . mockResolvedValueOnce ( '1' ) // app1 id
360- . mockResolvedValueOnce ( b64 ) // app1 key
361- . mockResolvedValueOnce ( '2' ) // app2 id
362- . mockResolvedValueOnce ( b64 ) // app2 key
363- . mockResolvedValueOnce ( '67890' ) ; // app2 installation id
378+ mockedGetParameters . mockResolvedValueOnce (
379+ new Map ( [
380+ [ app1IdParam , '1' ] ,
381+ [ app1KeyParam , b64 ] ,
382+ [ app2IdParam , '2' ] ,
383+ [ app2KeyParam , b64 ] ,
384+ [ app2InstallParam , '67890' ] ,
385+ ] ) ,
386+ ) ;
364387
365388 // Primary app (index 0) has no stored installation ID
366389 const result0 = await getStoredInstallationId ( 0 ) ;
0 commit comments