fix tests

DataDog · Jun 24, 2024 · 2162457 · 2162457
1 parent 9d2d848
commit 2162457
Show file tree

Hide file tree

Showing 3 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
@@ -95,13 +95,13 @@ describe('Kinesis', function () {
         helpers.putTestRecords(kinesis, streamName, (err, data) => {
           if (err) return done(err)
 
-          helpers.getTestRecord(kinesis, streamName, data, (err, data) => {
+          helpers.getTestRecord(kinesis, streamName, data.Records[0], (err, data) => {
             if (err) return done(err)
 
             for (const record in data.Records) {
-              const data = Buffer.from(record.Data).toString()
-              expect(data).to.have.property('_datadog')
-              expect(data._datadog).to.have.property('x-datadog-trace-id')
+              const recordData = JSON.parse(Buffer.from(data.Records[record].Data).toString())
+              expect(recordData).to.have.property('_datadog')
+              expect(recordData._datadog).to.have.property('x-datadog-trace-id')
             }
 
             done()

diff --git a/packages/datadog-plugin-aws-sdk/test/sns.spec.js b/packages/datadog-plugin-aws-sdk/test/sns.spec.js
@@ -8,7 +8,6 @@ const { setup } = require('./spec_helpers')
 const { rawExpectedSchema } = require('./sns-naming')
 
 describe('Sns', function () {
-  this.timeout(0)
   setup()
 
   withVersions('aws-sdk', ['aws-sdk', '@aws-sdk/smithy-client'], (version, moduleName) => {

diff --git a/packages/datadog-plugin-aws-sdk/test/sqs.spec.js b/packages/datadog-plugin-aws-sdk/test/sqs.spec.js
@@ -23,7 +23,6 @@ const queueOptionsDsm = getQueueParams(queueNameDSM)
 
 describe('Plugin', () => {
   describe('aws-sdk (sqs)', function () {
-    this.timeout(10000)
     setup()
 
     withVersions('aws-sdk', ['aws-sdk', '@aws-sdk/smithy-client'], (version, moduleName) => {