11import { describe , it , expect , vi , beforeEach } from 'vitest' ;
2- import { EC2Client , DescribeImagesCommand , DescribeLaunchTemplatesCommand , DescribeLaunchTemplateVersionsCommand , CreateLaunchTemplateVersionCommand , ModifyLaunchTemplateCommand } from '@aws-sdk/client-ec2' ;
3- import { AMIManager } from '../ami' ;
2+ import {
3+ EC2Client ,
4+ DescribeImagesCommand ,
5+ DescribeLaunchTemplatesCommand ,
6+ DescribeLaunchTemplateVersionsCommand ,
7+ CreateLaunchTemplateVersionCommand ,
8+ ModifyLaunchTemplateCommand ,
9+ } from '@aws-sdk/client-ec2' ;
10+ import { AMIManager } from './ami' ;
411
512vi . mock ( '@aws-sdk/client-ec2' ) ;
613vi . mock ( '../../shared/aws-powertools-util' , ( ) => ( {
@@ -34,7 +41,7 @@ describe('AMIManager', () => {
3441
3542 const config = {
3643 owners : [ 'self' ] ,
37- filters : [ { name : 'tag:Environment' , values : [ 'prod' ] } ] ,
44+ filters : [ { Name : 'tag:Environment' , Values : [ 'prod' ] } ] ,
3845 } ;
3946
4047 const result = await amiManager . getLatestAmi ( config ) ;
@@ -47,7 +54,7 @@ describe('AMIManager', () => {
4754
4855 const config = {
4956 owners : [ 'self' ] ,
50- filters : [ { name : 'tag:Environment' , values : [ 'prod' ] } ] ,
57+ filters : [ { Name : 'tag:Environment' , Values : [ 'prod' ] } ] ,
5158 } ;
5259
5360 await expect ( amiManager . getLatestAmi ( config ) ) . rejects . toThrow ( 'No matching AMIs found' ) ;
@@ -57,13 +64,16 @@ describe('AMIManager', () => {
5764 describe ( 'updateLaunchTemplate' , ( ) => {
5865 it ( 'should update launch template with new AMI ID' , async ( ) => {
5966 vi . mocked ( ec2Client . send )
60- . mockResolvedValueOnce ( { // getCurrentAmiId - DescribeLaunchTemplatesCommand
67+ . mockResolvedValueOnce ( {
68+ // getCurrentAmiId - DescribeLaunchTemplatesCommand
6169 LaunchTemplates : [ { LatestVersionNumber : 1 } ] ,
6270 } )
63- . mockResolvedValueOnce ( { // getCurrentAmiId - DescribeLaunchTemplateVersionsCommand
71+ . mockResolvedValueOnce ( {
72+ // getCurrentAmiId - DescribeLaunchTemplateVersionsCommand
6473 LaunchTemplateVersions : [ { LaunchTemplateData : { ImageId : 'ami-old' } } ] ,
6574 } )
66- . mockResolvedValueOnce ( { // updateLaunchTemplate - DescribeLaunchTemplatesCommand
75+ . mockResolvedValueOnce ( {
76+ // updateLaunchTemplate - DescribeLaunchTemplatesCommand
6777 LaunchTemplates : [ { LatestVersionNumber : 1 } ] ,
6878 } ) ;
6979
@@ -77,10 +87,12 @@ describe('AMIManager', () => {
7787
7888 it ( 'should not update if AMI ID is the same' , async ( ) => {
7989 vi . mocked ( ec2Client . send )
80- . mockResolvedValueOnce ( { // getCurrentAmiId - DescribeLaunchTemplatesCommand
90+ . mockResolvedValueOnce ( {
91+ // getCurrentAmiId - DescribeLaunchTemplatesCommand
8192 LaunchTemplates : [ { LatestVersionNumber : 1 } ] ,
8293 } )
83- . mockResolvedValueOnce ( { // getCurrentAmiId - DescribeLaunchTemplateVersionsCommand
94+ . mockResolvedValueOnce ( {
95+ // getCurrentAmiId - DescribeLaunchTemplateVersionsCommand
8496 LaunchTemplateVersions : [ { LaunchTemplateData : { ImageId : 'ami-1' } } ] ,
8597 } ) ;
8698
@@ -93,10 +105,12 @@ describe('AMIManager', () => {
93105
94106 it ( 'should handle dry run mode' , async ( ) => {
95107 vi . mocked ( ec2Client . send )
96- . mockResolvedValueOnce ( { // getCurrentAmiId - DescribeLaunchTemplatesCommand
108+ . mockResolvedValueOnce ( {
109+ // getCurrentAmiId - DescribeLaunchTemplatesCommand
97110 LaunchTemplates : [ { LatestVersionNumber : 1 } ] ,
98111 } )
99- . mockResolvedValueOnce ( { // getCurrentAmiId - DescribeLaunchTemplateVersionsCommand
112+ . mockResolvedValueOnce ( {
113+ // getCurrentAmiId - DescribeLaunchTemplateVersionsCommand
100114 LaunchTemplateVersions : [ { LaunchTemplateData : { ImageId : 'ami-old' } } ] ,
101115 } ) ;
102116
@@ -107,4 +121,4 @@ describe('AMIManager', () => {
107121 expect ( ec2Client . send ) . not . toHaveBeenCalledWith ( expect . any ( CreateLaunchTemplateVersionCommand ) ) ;
108122 } ) ;
109123 } ) ;
110- } ) ;
124+ } ) ;
0 commit comments