[service-bus] Throw an error if the user tries to just send a string (sender.send(string))

sdk/servicebus/service-bus/src/sender.ts

@@ -247,13 +247,12 @@ export class SenderImpl implements Sender {
       return this._sender.sendBatch(batch, options);
     } else if (isServiceBusMessageBatch(messageOrMessagesOrBatch)) {
       return this._sender.sendBatch(messageOrMessagesOrBatch, options);
+    } else if (messageOrMessagesOrBatch != null && typeof messageOrMessagesOrBatch === "object") {
+      return this._sender.send(messageOrMessagesOrBatch, options);
     } else {
-      throwTypeErrorIfParameterMissing(
-        this._context.namespace.connectionId,
-        "message, messages or messageBatch",
-        messageOrMessagesOrBatch
+      throw new TypeError(
+        "Invalid type for message. Must be a ServiceBusMessage, an array of ServiceBusMessage or a ServiceBusMessageBatch"
       );
-      return this._sender.send(messageOrMessagesOrBatch, options);
     }
   }
 

sdk/servicebus/service-bus/test/internal/sender.spec.ts

@@ -5,7 +5,8 @@ import chai from "chai";
 import { ServiceBusMessageBatchImpl } from "../../src/serviceBusMessageBatch";
 import { ClientEntityContext } from "../../src/clientEntityContext";
 import { ServiceBusMessage } from "../../src";
-import { isServiceBusMessageBatch } from "../../src/sender";
+import { isServiceBusMessageBatch, SenderImpl } from "../../src/sender";
+import { DefaultDataTransformer } from "@azure/core-amqp";
 const assert = chai.assert;
 
 describe("sender unit tests", () => {
@@ -18,4 +19,46 @@ describe("sender unit tests", () => {
     assert.isFalse(isServiceBusMessageBatch((4 as any) as ServiceBusMessage));
     assert.isFalse(isServiceBusMessageBatch(({} as any) as ServiceBusMessage));
   });
+
+  it("don't allow Sender.send(string)", async () => {
+    const sender = new SenderImpl(createClientEntityContextForTests());
+
+    try {
+      await sender.send(
+        // @ts-expect-error
+        "hello"
+      );
+    } catch (err) {
+      assert.equal(err.name, "TypeError");
+      assert.equal(
+        err.message,
+        "Invalid type for message. Must be a ServiceBusMessage, an array of ServiceBusMessage or a ServiceBusMessageBatch"
+      );
+    }
+  });
 });
+
+function createClientEntityContextForTests(): ClientEntityContext & { initWasCalled: boolean } {
+  let initWasCalled = false;
+
+  const fakeClientEntityContext = {
+    entityPath: "queue",
+    sender: {
+      credit: 999
+    },
+    namespace: {
+      config: { endpoint: "my.service.bus" },
+      connectionId: "connection-id",
+      dataTransformer: new DefaultDataTransformer(),
+      cbsSession: {
+        cbsLock: "cbs-lock",
+        async init() {
+          initWasCalled = true;
+        }
+      }
+    },
+    initWasCalled
+  };
+
+  return (fakeClientEntityContext as any) as ReturnType<typeof createClientEntityContextForTests>;
+}

sdk/servicebus/service-bus/test/smoketest.spec.ts

@@ -47,7 +47,7 @@ describe("Sample scenarios for track 2", () => {
         serviceBusClient.createReceiver(queueName, "peekLock")
       );
 
-      await sendSampleMessage(sender, "Queue, peek/lock");
+      await sendSampleMessage(sender, "Queue, peek/lock", undefined, "single");
 
       const errors: string[] = [];
       const receivedBodies: string[] = [];
@@ -70,7 +70,7 @@ describe("Sample scenarios for track 2", () => {
         serviceBusClient.createReceiver(queueName, "receiveAndDelete")
       );
 
-      await sendSampleMessage(sender, "Queue, peek/lock, receiveBatch");
+      await sendSampleMessage(sender, "Queue, peek/lock, receiveBatch", undefined, "array");
 
       const receivedBodies: string[] = [];
 
@@ -87,7 +87,7 @@ describe("Sample scenarios for track 2", () => {
         serviceBusClient.createReceiver(queueName, "peekLock")
       );
 
-      await sendSampleMessage(sender, "Queue, peek/lock, iterate messages");
+      await sendSampleMessage(sender, "Queue, peek/lock, iterate messages", undefined, "batch");
 
       // etc...
       // receiver.getRules();
@@ -160,12 +160,8 @@ describe("Sample scenarios for track 2", () => {
           continue;
         }
 
-        try {
-          receivedBodies.push(message.body);
-          break;
-        } catch (err) {
-          throw err;
-        }
+        receivedBodies.push(message.body);
+        break;
       }
 
       await waitAndValidate(
@@ -306,12 +302,8 @@ describe("Sample scenarios for track 2", () => {
           continue;
         }
 
-        try {
-          receivedBodies.push(message.body);
-          break;
-        } catch (err) {
-          throw err;
-        }
+        receivedBodies.push(message.body);
+        break;
       }
 
       await waitAndValidate(
@@ -430,7 +422,12 @@ describe("Sample scenarios for track 2", () => {
     });
   });
 
-  async function sendSampleMessage(sender: Sender, body: string, sessionId?: string) {
+  async function sendSampleMessage(
+    sender: Sender,
+    body: string,
+    sessionId?: string,
+    method: "single" | "array" | "batch" = "single"
+  ): Promise<void> {
     const message: ServiceBusMessage = {
       body
     };
@@ -439,7 +436,22 @@ describe("Sample scenarios for track 2", () => {
       message.sessionId = sessionId;
     }
 
-    await sender.send(message);
+    switch (method) {
+      case "single": {
+        await sender.send(message);
+        break;
+      }
+      case "array": {
+        await sender.send([message]);
+        break;
+      }
+      case "batch": {
+        const batch = await sender.createBatch();
+        assert.isTrue(batch.tryAdd(message));
+        await sender.send(batch);
+        break;
+      }
+    }
   }
 });
 
@@ -461,7 +473,7 @@ async function waitAndValidate(
   receivedBodies: string[],
   errors: string[],
   receiver: Receiver<ReceivedMessage>
-) {
+): Promise<void> {
   const maxChecks = 20;
   let numChecks = 0;