diff --git a/service/src/test/java/org/whispersystems/textsecuregcm/push/PushNotificationManagerTest.java b/service/src/test/java/org/whispersystems/textsecuregcm/push/PushNotificationManagerTest.java index 40f6e2c8b..d456d5cee 100644 --- a/service/src/test/java/org/whispersystems/textsecuregcm/push/PushNotificationManagerTest.java +++ b/service/src/test/java/org/whispersystems/textsecuregcm/push/PushNotificationManagerTest.java @@ -151,6 +151,8 @@ class PushNotificationManagerTest { when(device.getApnId()).thenReturn(deviceToken); when(apnSender.sendNotification(any())) .thenReturn(CompletableFuture.completedFuture(new SendPushNotificationResult(true, null, false))); + when(apnPushNotificationScheduler.scheduleBackgroundNotification(account, device)) + .thenReturn(CompletableFuture.completedFuture(null)); } else { when(device.getGcmId()).thenReturn(deviceToken); when(fcmSender.sendNotification(any()))