test(opentelemetry-plugin-aws-sdk): add few tests to suite

Author: Amir Blum

packages/plugin-aws-sdk/test/aws-sdk.spec.ts

@@ -217,6 +217,23 @@ describe("plugin-aws-sdk", () => {
         );
         done();
       });
+
+      it("should create span if no callback is supplied", (done) => {
+        const s3 = new AWS.S3();
+        const bucketName = "aws-test-bucket";
+
+        s3.putObject({
+          Bucket: bucketName,
+          Key: "key name from tests",
+          Body: "Hello World!",
+        }).send();
+
+        setImmediate(() => {
+          const awsSpans = getAwsSpans();
+          expect(awsSpans.length).toBe(1);
+          done();
+        });
+      });
     });
 
     describe("send return error", () => {

packages/plugin-aws-sdk/test/sqs.spec.ts

@@ -73,6 +73,21 @@ describe("sqs", () => {
     };
 
     it("should set parent context in sqs receive callback", async (done) => {
+      const sqs = new AWS.SQS();
+      sqs.receiveMessage(
+        {
+          QueueUrl: "queue/url/for/unittests",
+        },
+        (err: AWSError, data: AWS.SQS.Types.ReceiveMessageResult) => {
+          expect(err).toBeFalsy();
+          createReceiveChildSpan();
+          expectReceiverWithChildSpan(memoryExporter.getFinishedSpans());
+          done();
+        }
+      );
+    });
+
+    it("should set parent context in sqs receive 'send' callback", async (done) => {
       const sqs = new AWS.SQS();
       sqs
         .receiveMessage({

packages/plugin-aws-sdk/test/testing-utils.ts

@@ -1,10 +1,12 @@
 import AWS from "aws-sdk";
 
 export const mockAwsSend = (sendResult: any, data: any = undefined) => {
-  AWS.Request.prototype.send = function (cb: (error, response) => void) {
-    (this as AWS.Request<any, any>).on("complete", (response) => {
-      cb(response.error, response);
-    });
+  AWS.Request.prototype.send = function (cb?: (error, response) => void) {
+    if (cb) {
+      (this as AWS.Request<any, any>).on("complete", (response) => {
+        cb(response.error, response);
+      });
+    }
     const response = {
       ...sendResult,
       data,