diff --git a/service/src/test/java/org/whispersystems/textsecuregcm/tests/controllers/MessageControllerTest.java b/service/src/test/java/org/whispersystems/textsecuregcm/tests/controllers/MessageControllerTest.java index 45f4e9e1f..00307f2c6 100644 --- a/service/src/test/java/org/whispersystems/textsecuregcm/tests/controllers/MessageControllerTest.java +++ b/service/src/test/java/org/whispersystems/textsecuregcm/tests/controllers/MessageControllerTest.java @@ -70,7 +70,6 @@ import org.whispersystems.textsecuregcm.storage.MessagesManager; import org.whispersystems.textsecuregcm.tests.util.AuthHelper; import org.whispersystems.textsecuregcm.util.Base64; -@Ignore public class MessageControllerTest { private static final String SINGLE_DEVICE_RECIPIENT = "+14151111111"; @@ -85,6 +84,7 @@ public class MessageControllerTest { private final MessagesManager messagesManager = mock(MessagesManager.class); private final RateLimiters rateLimiters = mock(RateLimiters.class); private final RateLimiter rateLimiter = mock(RateLimiter.class); + private final RateLimiter unsealedSenderLimiter = mock(RateLimiter.class); private final ApnFallbackManager apnFallbackManager = mock(ApnFallbackManager.class); private final FeatureFlagsManager featureFlagsManager = mock(FeatureFlagsManager.class); @@ -125,6 +125,7 @@ public class MessageControllerTest { when(accountsManager.get(argThat((ArgumentMatcher) identifier -> identifier != null && identifier.hasNumber() && identifier.getNumber().equals(MULTI_DEVICE_RECIPIENT)))).thenReturn(Optional.of(multiDeviceAccount)); when(rateLimiters.getMessagesLimiter()).thenReturn(rateLimiter); + when(rateLimiters.getUnsealedSenderLimiter()).thenReturn(unsealedSenderLimiter); } @Test