Fix timeouts from aws-sdk kinesis tests

DataDog · Dec 28, 2023 · 307d4cd · 307d4cd
1 parent 2127b02
commit 307d4cd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/packages/datadog-plugin-aws-sdk/test/kinesis.spec.js b/packages/datadog-plugin-aws-sdk/test/kinesis.spec.js
@@ -41,7 +41,7 @@ describe('Kinesis', () => {
       }, (err, res) => {
         if (err) return done(err)
 
-        helpers.waitForActiveStream(this, kinesis, done)
+        helpers.waitForActiveStream(kinesis, done)
       })
     })
 

diff --git a/packages/datadog-plugin-aws-sdk/test/kinesis_helpers.js b/packages/datadog-plugin-aws-sdk/test/kinesis_helpers.js
@@ -45,17 +45,15 @@ function putTestRecord (kinesis, data, cb) {
   }, cb)
 }
 
-function waitForActiveStream (mocha, kinesis, cb) {
+function waitForActiveStream (kinesis, cb) {
   kinesis.describeStream({
     StreamName: 'MyStream'
   }, (err, data) => {
     if (err) {
-      mocha.timeout(2000)
-      return waitForActiveStream(mocha, kinesis, cb)
+      return waitForActiveStream(kinesis, cb)
     }
     if (data.StreamDescription.StreamStatus !== 'ACTIVE') {
-      mocha.timeout(2000)
-      return waitForActiveStream(mocha, kinesis, cb)
+      return waitForActiveStream(kinesis, cb)
     }
 
     cb()