node-lambda 1.3.0 → 2.0.0-alpha.0

package/test/s3_events.js

@@ -1,9 +1,14 @@
 'use strict'
 
-const assert = require('chai').assert
-const path = require('path')
-const aws = require('aws-sdk-mock')
-aws.setSDK(path.resolve('node_modules/aws-sdk'))
+let assert
+import('chai').then(chai => {
+  assert = chai.assert
+})
+const { S3Client, PutBucketNotificationConfigurationCommand } = require('@aws-sdk/client-s3')
+const { LambdaClient, AddPermissionCommand } = require('@aws-sdk/client-lambda')
+const { mockClient } = require('aws-sdk-client-mock')
+const mockS3Client = mockClient(S3Client)
+const mockLambdaClient = mockClient(LambdaClient)
 const S3Events = require('../lib/s3_events')
 
 const params = {
@@ -30,22 +35,15 @@ const mockResponse = {
 
 let s3Events = null
 
-/* global before, after, describe, it */
+/* global before, describe, it */
 describe('lib/s3_events', () => {
   before(() => {
-    aws.mock('Lambda', 'addPermission', (params, callback) => {
-      callback(null, mockResponse.addPermission)
-    })
-    aws.mock('S3', 'putBucketNotificationConfiguration', (params, callback) => {
-      callback(null, mockResponse.putBucketNotificationConfiguration)
-    })
-
-    s3Events = new S3Events(require('aws-sdk'))
-  })
+    mockS3Client.reset()
+    mockS3Client.on(PutBucketNotificationConfigurationCommand).resolves(mockResponse.putBucketNotificationConfiguration)
+    mockLambdaClient.reset()
+    mockLambdaClient.on(AddPermissionCommand).resolves(mockResponse.addPermission)
 
-  after(() => {
-    aws.restore('Lambda')
-    aws.restore('S3')
+    s3Events = new S3Events({ region: 'us-west-1' })
   })
 
   describe('_functionName', () => {

package/test/schedule_events.js

@@ -1,9 +1,15 @@
 'use strict'
 
-const assert = require('chai').assert
+let assert
+import('chai').then(chai => {
+  assert = chai.assert
+})
 const path = require('path')
-const aws = require('aws-sdk-mock')
-aws.setSDK(path.resolve('node_modules/aws-sdk'))
+const { CloudWatchEventsClient, PutRuleCommand, PutTargetsCommand } = require('@aws-sdk/client-cloudwatch-events')
+const { LambdaClient, AddPermissionCommand } = require('@aws-sdk/client-lambda')
+const { mockClient } = require('aws-sdk-client-mock')
+const mockCloudWatchEventsClient = mockClient(CloudWatchEventsClient)
+const mockLambdaClient = mockClient(LambdaClient)
 const ScheduleEvents = require(path.join('..', 'lib', 'schedule_events'))
 
 const params = {
@@ -41,22 +47,14 @@ let schedule = null
 /* global before, after, describe, it */
 describe('lib/schedule_events', () => {
   before(() => {
-    aws.mock('CloudWatchEvents', 'putRule', (params, callback) => {
-      callback(null, mockResponse.putRule)
-    })
-    aws.mock('CloudWatchEvents', 'putTargets', (params, callback) => {
-      callback(null, mockResponse.putTargets)
-    })
-    aws.mock('Lambda', 'addPermission', (params, callback) => {
-      callback(null, mockResponse.addPermission)
-    })
+    mockCloudWatchEventsClient.reset()
+    mockCloudWatchEventsClient.on(PutRuleCommand).resolves(mockResponse.putRule)
+    mockCloudWatchEventsClient.on(PutTargetsCommand).resolves(mockResponse.putTargets)
 
-    schedule = new ScheduleEvents(require('aws-sdk'))
-  })
+    mockLambdaClient.reset()
+    mockLambdaClient.on(AddPermissionCommand).resolves(mockResponse.addPermission)
 
-  after(() => {
-    aws.restore('CloudWatchEvents')
-    aws.restore('Lambda')
+    schedule = new ScheduleEvents({ region: 'us-west-1' })
   })
 
   describe('_ruleDescription (default)', () => {