From 86f435531637b1f3059fc9138f44f19e88359f6b Mon Sep 17 00:00:00 2001
From: Doris Ge <dorisge@google.com>
Date: Thu, 2 Mar 2023 23:12:26 -0800
Subject: [PATCH] Add integration tests for `sendEach` and `sendMulticast`

---
 test/integration/messaging.spec.ts | 50 ++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)

diff --git a/test/integration/messaging.spec.ts b/test/integration/messaging.spec.ts
index bc3533893e..0a17a2d750 100644
--- a/test/integration/messaging.spec.ts
+++ b/test/integration/messaging.spec.ts
@@ -108,6 +108,37 @@ describe('admin.messaging', () => {
       });
   });
 
+  it('sendEach()', () => {
+    const messages: Message[] = [message, message, message];
+    return getMessaging().sendEach(messages, true)
+      .then((response) => {
+        expect(response.responses.length).to.equal(messages.length);
+        expect(response.successCount).to.equal(messages.length);
+        expect(response.failureCount).to.equal(0);
+        response.responses.forEach((resp) => {
+          expect(resp.success).to.be.true;
+          expect(resp.messageId).matches(/^projects\/.*\/messages\/.*$/);
+        });
+      });
+  });
+
+  it('sendEach(500)', () => {
+    const messages: Message[] = [];
+    for (let i = 0; i < 500; i++) {
+      messages.push({ topic: `foo-bar-${i % 10}` });
+    }
+    return getMessaging().sendEach(messages, true)
+      .then((response) => {
+        expect(response.responses.length).to.equal(messages.length);
+        expect(response.successCount).to.equal(messages.length);
+        expect(response.failureCount).to.equal(0);
+        response.responses.forEach((resp) => {
+          expect(resp.success).to.be.true;
+          expect(resp.messageId).matches(/^projects\/.*\/messages\/.*$/);
+        });
+      });
+  });
+
   it('sendAll()', () => {
     const messages: Message[] = [message, message, message];
     return getMessaging().sendAll(messages, true)
@@ -139,6 +170,25 @@ describe('admin.messaging', () => {
       });
   });
 
+  it('sendEachForMulticast()', () => {
+    const multicastMessage: MulticastMessage = {
+      data: message.data,
+      android: message.android,
+      tokens: ['not-a-token', 'also-not-a-token'],
+    };
+    return getMessaging().sendEachForMulticast(multicastMessage, true)
+      .then((response) => {
+        expect(response.responses.length).to.equal(2);
+        expect(response.successCount).to.equal(0);
+        expect(response.failureCount).to.equal(2);
+        response.responses.forEach((resp) => {
+          expect(resp.success).to.be.false;
+          expect(resp.messageId).to.be.undefined;
+          expect(resp.error).to.have.property('code', 'messaging/invalid-argument');
+        });
+      });
+  });
+
   it('sendMulticast()', () => {
     const multicastMessage: MulticastMessage = {
       data: message.data,