refactor: use v3 sdk

Signed-off-by: Markus Maga <[email protected]>
docker · Dec 20, 2021 · 5885569 · 5885569
1 parent d9927c4
commit 5885569
Show file tree

Hide file tree

Showing 5 changed files with 26,615 additions and 37,189 deletions.
diff --git a/__tests__/aws.test.ts b/__tests__/aws.test.ts
@@ -1,5 +1,5 @@
+import {AuthorizationData} from '@aws-sdk/client-ecr';
 import * as aws from '../src/aws';
-import {AuthorizationData} from 'aws-sdk/clients/ecr';
 
 describe('isECR', () => {
   test.each([
@@ -55,11 +55,15 @@ describe('getAccountIDs', () => {
 
 const mockEcrGetAuthToken = jest.fn();
 const mockEcrPublicGetAuthToken = jest.fn();
-jest.mock('aws-sdk', () => {
+jest.mock('@aws-sdk/client-ecr', () => {
   return {
     ECR: jest.fn(() => ({
       getAuthorizationToken: mockEcrGetAuthToken
-    })),
+    }))
+  };
+});
+jest.mock('@aws-sdk/client-ecr-public', () => {
+  return {
     ECRPUBLIC: jest.fn(() => ({
       getAuthorizationToken: mockEcrPublicGetAuthToken
     }))
@@ -126,15 +130,11 @@ describe('getRegistriesData', () => {
     const authData: AuthorizationData[] = [];
     if (accountIDs.length == 0) {
       mockEcrPublicGetAuthToken.mockImplementation(() => {
-        return {
-          promise() {
-            return Promise.resolve({
-              authorizationData: {
-                authorizationToken: Buffer.from(`AWS:world`).toString('base64'),
-              }
-            });
+        return Promise.resolve({
+          authorizationData: {
+            authorizationToken: Buffer.from(`AWS:world`).toString('base64'),
           }
-        };
+        });
       });
     } else {
       aws.getAccountIDs(registry).forEach(accountID => {
@@ -144,13 +144,9 @@ describe('getRegistriesData', () => {
         });
       });
       mockEcrGetAuthToken.mockImplementation(() => {
-        return {
-          promise() {
-            return Promise.resolve({
-              authorizationData: authData
-            });
-          }
-        };
+        return Promise.resolve({
+          authorizationData: authData
+        });
       });
     }
     const regData = await aws.getRegistriesData(registry);