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

Add methods for iterating over all account identifiers

Browse Source
This commit is contained in:
Jon Chambers
2025-02-18 09:19:14 -05:00
committed by Jon Chambers
parent b203344ed4
commit b8b17ae473
3 changed files with 43 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
service/src/test/java/org/whispersystems/textsecuregcm/storage/AccountsTest.java
View File

@@ -30,10 +30,12 @@ import java.util.ArrayList;
import java.util.Arrays;
import java.util.Base64;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Random;
import java.util.Set;
import java.util.UUID;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionException;
@@ -56,6 +58,7 @@ import org.junit.jupiter.params.provider.ValueSource;
import org.signal.libsignal.zkgroup.backups.BackupCredentialType;
import org.whispersystems.textsecuregcm.auth.UnidentifiedAccessUtil;
import org.whispersystems.textsecuregcm.configuration.dynamic.DynamicConfiguration;
import org.whispersystems.textsecuregcm.identity.IdentityType;
import org.whispersystems.textsecuregcm.storage.DynamoDbExtensionSchema.Tables;
import org.whispersystems.textsecuregcm.tests.util.AccountsHelper;
import org.whispersystems.textsecuregcm.tests.util.DevicesHelper;
@@ -731,6 +734,22 @@ class AccountsTest {
retrievedAccounts.stream().map(Account::getUuid).collect(Collectors.toSet()));
}
@Test
void testGetAllAccountIdentifiers() {
final Set<UUID> expectedAccountIdentifiers = new HashSet<>();
for (int i = 1; i <= 100; i++) {
final Account account = generateAccount("+1" + String.format("%03d", i), UUID.randomUUID(), UUID.randomUUID());
expectedAccountIdentifiers.add(account.getIdentifier(IdentityType.ACI));
createAccount(account);
}
@SuppressWarnings("DataFlowIssue") final Set<UUID> retrievedAccountIdentifiers =
new HashSet<>(accounts.getAllAccountIdentifiers(2, Schedulers.parallel()).collectList().block());
assertEquals(expectedAccountIdentifiers, retrievedAccountIdentifiers);
}
@Test
void testDelete() {
final Device deletedDevice = generateDevice(DEVICE_ID_1);