Update tests for AWS SDK v3

arithmetric · Oct 19, 2023 · dbf70d7 · dbf70d7
1 parent 5c3dc02
commit dbf70d7
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 22 deletions.
diff --git a/test/fetchMessage.js b/test/fetchMessage.js
@@ -3,6 +3,8 @@
 
 var assert = require("assert");
 
+const {GetObjectCommand, CopyObjectCommand} = require('@aws-sdk/client-s3');
+
 var index = require("../index");
 
 describe('index.js', function() {
@@ -25,11 +27,17 @@ describe('index.js', function() {
           },
           log: console.log,
           s3: {
-            copyObject: function(options, callback) {
-              callback(null);
-            },
-            getObject: function(options, callback) {
-              callback(null, {Body: "email data"});
+            send: function(options, callback) {
+              if (options instanceof CopyObjectCommand)
+                callback(null);
+              else if (options instanceof GetObjectCommand)
+                callback(null, {
+                  Body: {
+                    transformToString: function() {
+                      return "email data";
+                    }
+                  }
+                });
             }
           }
         };
@@ -55,10 +63,7 @@ describe('index.js', function() {
           },
           log: console.log,
           s3: {
-            copyObject: function(options, callback) {
-              callback(true);
-            },
-            getObject: function(options, callback) {
+            send: function(options, callback) {
               callback(true);
             }
           }
@@ -83,11 +88,11 @@ describe('index.js', function() {
           },
           log: console.log,
           s3: {
-            copyObject: function(options, callback) {
-              callback(null);
-            },
-            getObject: function(options, callback) {
-              callback(true);
+            send: function(options, callback) {
+              if (options instanceof CopyObjectCommand)
+                callback(null);
+              else if (options instanceof GetObjectCommand)
+                callback(true);
             }
           }
         };

diff --git a/test/handler.js b/test/handler.js
@@ -4,6 +4,8 @@
 var assert = require("assert");
 var fs = require("fs");
 
+const {GetObjectCommand, CopyObjectCommand} = require('@aws-sdk/client-s3');
+
 var index = require("../index");
 
 describe('index.js', function() {
@@ -16,15 +18,21 @@ describe('index.js', function() {
       };
       var overrides = {
         s3: {
-          copyObject: function(options, callback) {
-            callback(null);
-          },
-          getObject: function(options, callback) {
-            callback(null, {Body: "email data"});
+          send: function(options, callback) {
+            if (options instanceof CopyObjectCommand)
+              callback(null);
+            else if (options instanceof GetObjectCommand)
+              callback(null, {
+                Body: {
+                  transformToString: function() {
+                    return "email data";
+                  }
+                }
+              });
           }
         },
         ses: {
-          sendRawEmail: function(options, callback) {
+          send: function(options, callback) {
             callback(null, {status: "ok"});
           }
         },

diff --git a/test/sendMessage.js b/test/sendMessage.js
@@ -20,7 +20,7 @@ describe('index.js', function() {
           context: {},
           log: console.log,
           ses: {
-            sendRawEmail: function(options, callback) {
+            send: function(options, callback) {
               callback(null, {status: "ok"});
             }
           }
@@ -45,7 +45,7 @@ describe('index.js', function() {
           context: {},
           log: console.log,
           ses: {
-            sendRawEmail: function(options, callback) {
+            send: function(options, callback) {
               callback(true);
             }
           }