Signal/Signal-Server
1
0
Fork 0
mirror of https://github.com/signalapp/Signal-Server synced 2026-04-22 07:58:07 +01:00
Code Issues Packages Projects Releases Wiki Activity

Remove test references to legacy ratelimit HTTP status code

Browse Source
This commit is contained in:
Katherine
2024-10-28 15:20:08 -04:00
committed by GitHub
parent 17dfd914d5
commit 3287085ef9
6 changed files with 19 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
service/src/test/java/org/whispersystems/textsecuregcm/util/MockUtils.java
View File

@@ -98,8 +98,7 @@ public final class MockUtils {
public static Duration updateRateLimiterResponseToFail(
final RateLimiter mockRateLimiter,
final String input,
final Duration retryAfter,
final boolean legacyStatusCode) {
final Duration retryAfter) {
try {
final RateLimitExceededException exception = new RateLimitExceededException(retryAfter);
doThrow(exception).when(mockRateLimiter).validate(eq(input));
@@ -114,8 +113,7 @@ public final class MockUtils {
public static Duration updateRateLimiterResponseToFail(
final RateLimiter mockRateLimiter,
final UUID input,
final Duration retryAfter,
final boolean legacyStatusCode) {
final Duration retryAfter) {
try {
final RateLimitExceededException exception = new RateLimitExceededException(retryAfter);
doThrow(exception).when(mockRateLimiter).validate(eq(input));
@@ -131,22 +129,20 @@ public final class MockUtils {
final RateLimiters rateLimitersMock,
final RateLimiters.For handle,
final String input,
final Duration retryAfter,
final boolean legacyStatusCode) {
final Duration retryAfter) {
final RateLimiter mockRateLimiter = Mockito.mock(RateLimiter.class);
doReturn(mockRateLimiter).when(rateLimitersMock).forDescriptor(eq(handle));
return updateRateLimiterResponseToFail(mockRateLimiter, input, retryAfter, legacyStatusCode);
return updateRateLimiterResponseToFail(mockRateLimiter, input, retryAfter);
}
public static Duration updateRateLimiterResponseToFail(
final RateLimiters rateLimitersMock,
final RateLimiters.For handle,
final UUID input,
final Duration retryAfter,
final boolean legacyStatusCode) {
final Duration retryAfter) {
final RateLimiter mockRateLimiter = Mockito.mock(RateLimiter.class);
doReturn(mockRateLimiter).when(rateLimitersMock).forDescriptor(eq(handle));
return updateRateLimiterResponseToFail(mockRateLimiter, input, retryAfter, legacyStatusCode);
return updateRateLimiterResponseToFail(mockRateLimiter, input, retryAfter);
}
public static SecretBytes randomSecretBytes(final int size) {