[DSM] Add a wait for active stream to the putTestRecords function which was flaking when the stream was inactive (#5202)

Author: ericfirth

packages/datadog-plugin-aws-sdk/test/kinesis_helpers.js

@@ -55,23 +55,25 @@ function putTestRecord (kinesis, streamName, data, cb) {
 }
 
 function putTestRecords (kinesis, streamName, cb) {
-  kinesis.putRecords({
-    Records: [
-      {
-        PartitionKey: id().toString(),
-        Data: dataBufferCustom(1)
-      },
-      {
-        PartitionKey: id().toString(),
-        Data: dataBufferCustom(2)
-      },
-      {
-        PartitionKey: id().toString(),
-        Data: dataBufferCustom(3)
-      }
-    ],
-    StreamName: streamName
-  }, cb)
+  waitForActiveStream(kinesis, streamName, () => {
+    kinesis.putRecords({
+      Records: [
+        {
+          PartitionKey: id().toString(),
+          Data: dataBufferCustom(1)
+        },
+        {
+          PartitionKey: id().toString(),
+          Data: dataBufferCustom(2)
+        },
+        {
+          PartitionKey: id().toString(),
+          Data: dataBufferCustom(3)
+        }
+      ],
+      StreamName: streamName
+    }, cb)
+  })
 }
 
 function waitForActiveStream (kinesis, streamName, cb) {