1
+ import { AuthorizationData } from '@aws-sdk/client-ecr' ;
1
2
import * as aws from '../src/aws' ;
2
- import { AuthorizationData } from 'aws-sdk/clients/ecr' ;
3
3
4
4
describe ( 'isECR' , ( ) => {
5
5
test . each ( [
@@ -55,11 +55,15 @@ describe('getAccountIDs', () => {
55
55
56
56
const mockEcrGetAuthToken = jest . fn ( ) ;
57
57
const mockEcrPublicGetAuthToken = jest . fn ( ) ;
58
- jest . mock ( 'aws-sdk' , ( ) => {
58
+ jest . mock ( '@ aws-sdk/client-ecr ' , ( ) => {
59
59
return {
60
60
ECR : jest . fn ( ( ) => ( {
61
61
getAuthorizationToken : mockEcrGetAuthToken
62
- } ) ) ,
62
+ } ) )
63
+ } ;
64
+ } ) ;
65
+ jest . mock ( '@aws-sdk/client-ecr-public' , ( ) => {
66
+ return {
63
67
ECRPUBLIC : jest . fn ( ( ) => ( {
64
68
getAuthorizationToken : mockEcrPublicGetAuthToken
65
69
} ) )
@@ -126,15 +130,11 @@ describe('getRegistriesData', () => {
126
130
const authData : AuthorizationData [ ] = [ ] ;
127
131
if ( accountIDs . length == 0 ) {
128
132
mockEcrPublicGetAuthToken . mockImplementation ( ( ) => {
129
- return {
130
- promise ( ) {
131
- return Promise . resolve ( {
132
- authorizationData : {
133
- authorizationToken : Buffer . from ( `AWS:world` ) . toString ( 'base64' ) ,
134
- }
135
- } ) ;
133
+ return Promise . resolve ( {
134
+ authorizationData : {
135
+ authorizationToken : Buffer . from ( `AWS:world` ) . toString ( 'base64' ) ,
136
136
}
137
- } ;
137
+ } ) ;
138
138
} ) ;
139
139
} else {
140
140
aws . getAccountIDs ( registry ) . forEach ( accountID => {
@@ -144,13 +144,9 @@ describe('getRegistriesData', () => {
144
144
} ) ;
145
145
} ) ;
146
146
mockEcrGetAuthToken . mockImplementation ( ( ) => {
147
- return {
148
- promise ( ) {
149
- return Promise . resolve ( {
150
- authorizationData : authData
151
- } ) ;
152
- }
153
- } ;
147
+ return Promise . resolve ( {
148
+ authorizationData : authData
149
+ } ) ;
154
150
} ) ;
155
151
}
156
152
const regData = await aws . getRegistriesData ( registry ) ;
0 commit comments