Introduce an FcmSender

service/src/test/java/org/whispersystems/textsecuregcm/push/FcmSenderTest.java

@@ -0,0 +1,105 @@
+/*
+ * Copyright 2013-2022 Signal Messenger, LLC
+ * SPDX-License-Identifier: AGPL-3.0-only
+ */
+
+package org.whispersystems.textsecuregcm.push;
+
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.eq;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+import com.google.api.core.SettableApiFuture;
+import com.google.firebase.messaging.FirebaseMessaging;
+import com.google.firebase.messaging.FirebaseMessagingException;
+import com.google.firebase.messaging.Message;
+import com.google.firebase.messaging.MessagingErrorCode;
+import java.util.Optional;
+import java.util.UUID;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.TimeUnit;
+import org.junit.jupiter.api.AfterEach;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.whispersystems.textsecuregcm.storage.Account;
+import org.whispersystems.textsecuregcm.storage.AccountsManager;
+import org.whispersystems.textsecuregcm.storage.Device;
+import org.whispersystems.textsecuregcm.tests.util.AccountsHelper;
+import org.whispersystems.textsecuregcm.tests.util.SynchronousExecutorService;
+import org.whispersystems.textsecuregcm.util.Util;
+
+class FcmSenderTest {
+
+  private ExecutorService executorService;
+  private AccountsManager accountsManager;
+  private FirebaseMessaging firebaseMessaging;
+
+  private FcmSender fcmSender;
+
+  @BeforeEach
+  void setUp() {
+    executorService = new SynchronousExecutorService();
+    accountsManager = mock(AccountsManager.class);
+    firebaseMessaging = mock(FirebaseMessaging.class);
+
+    fcmSender = new FcmSender(executorService, accountsManager, firebaseMessaging);
+  }
+
+  @AfterEach
+  void tearDown() throws InterruptedException {
+    executorService.shutdown();
+
+    //noinspection ResultOfMethodCallIgnored
+    executorService.awaitTermination(1, TimeUnit.SECONDS);
+  }
+
+  @Test
+  void testSendMessage() {
+    AccountsHelper.setupMockUpdate(accountsManager);
+
+    final GcmMessage message = new GcmMessage("foo", UUID.randomUUID(), 1, GcmMessage.Type.NOTIFICATION, Optional.empty());
+
+    final SettableApiFuture<String> sendFuture = SettableApiFuture.create();
+    sendFuture.set("message-id");
+
+    when(firebaseMessaging.sendAsync(any())).thenReturn(sendFuture);
+
+    fcmSender.sendMessage(message);
+
+    verify(firebaseMessaging).sendAsync(any(Message.class));
+  }
+
+  @Test
+  void testSendUninstalled() {
+    final UUID destinationUuid = UUID.randomUUID();
+    final String gcmId = "foo";
+
+    final Account destinationAccount = mock(Account.class);
+    final Device  destinationDevice  = mock(Device.class );
+
+    AccountsHelper.setupMockUpdate(accountsManager);
+
+    when(destinationAccount.getDevice(1)).thenReturn(Optional.of(destinationDevice));
+    when(accountsManager.getByAccountIdentifier(destinationUuid)).thenReturn(Optional.of(destinationAccount));
+    when(destinationDevice.getGcmId()).thenReturn(gcmId);
+
+    final GcmMessage message = new GcmMessage(gcmId, destinationUuid, 1, GcmMessage.Type.NOTIFICATION, Optional.empty());
+
+    final FirebaseMessagingException unregisteredException = mock(FirebaseMessagingException.class);
+    when(unregisteredException.getMessagingErrorCode()).thenReturn(MessagingErrorCode.UNREGISTERED);
+
+    final SettableApiFuture<String> sendFuture = SettableApiFuture.create();
+    sendFuture.setException(unregisteredException);
+
+    when(firebaseMessaging.sendAsync(any())).thenReturn(sendFuture);
+
+    fcmSender.sendMessage(message);
+
+    verify(firebaseMessaging).sendAsync(any(Message.class));
+    verify(accountsManager).getByAccountIdentifier(destinationUuid);
+    verify(accountsManager).updateDevice(eq(destinationAccount), eq(1L), any());
+    verify(destinationDevice).setUninstalledFeedbackTimestamp(Util.todayInMillis());
+  }
+}