Use getAccountIdentifier in device transfer context

This commit is contained in:
Chris Eager
2026-08-11 09:12:36 -04:00
committed by Jon Chambers
parent d69a5f1f33
commit b63f176ad7
2 changed files with 7 additions and 7 deletions
@@ -1524,8 +1524,8 @@ public class AccountsManager extends RedisPubSubAdapter<String, String> implemen
}
public CompletableFuture<Optional<TransferArchiveResult>> waitForTransferArchive(final Account account, final Device device, final Duration timeout) {
final DeviceIdentifier deviceIdentifier = new DeviceIdentifier(account.getIdentifier(IdentityType.ACI), device.getId(), device.getRegistrationId(IdentityType.ACI));
final String registrationIdTransferArchiveKey = getRegistrationIdTransferArchiveKey(account.getIdentifier(IdentityType.ACI), device.getId(), device.getRegistrationId(IdentityType.ACI));
final DeviceIdentifier deviceIdentifier = new DeviceIdentifier(account.getAccountIdentifier(), device.getId(), device.getRegistrationId(IdentityType.ACI));
final String registrationIdTransferArchiveKey = getRegistrationIdTransferArchiveKey(account.getAccountIdentifier(), device.getId(), device.getRegistrationId(IdentityType.ACI));
return waitForPubSubKey(waitForTransferArchiveFuturesByDeviceIdentifier,
deviceIdentifier,
@@ -1541,7 +1541,7 @@ public class AccountsManager extends RedisPubSubAdapter<String, String> implemen
try {
final String transferArchiveJson = SystemMapper.jsonMapper().writeValueAsString(transferArchiveResult);
final String key = getRegistrationIdTransferArchiveKey(account.getIdentifier(IdentityType.ACI), destinationDeviceId, registrationId);
final String key = getRegistrationIdTransferArchiveKey(account.getAccountIdentifier(), destinationDeviceId, registrationId);
return ResilienceUtil.getGeneralRedisRetry(RETRY_NAME)
.executeCompletionStage(retryExecutor, () -> pubSubRedisClient.withConnection(connection -> connection.async()
@@ -97,7 +97,7 @@ public class AccountsManagerDeviceTransferIntegrationTest {
when(device.getRegistrationId(IdentityType.ACI)).thenReturn(registrationId);
final Account account = mock(Account.class);
when(account.getIdentifier(IdentityType.ACI)).thenReturn(accountIdentifier);
when(account.getAccountIdentifier()).thenReturn(accountIdentifier);
final CompletableFuture<Optional<TransferArchiveResult>> displacedFuture =
accountsManager.waitForTransferArchive(account, device, Duration.ofSeconds(5));
@@ -126,7 +126,7 @@ public class AccountsManagerDeviceTransferIntegrationTest {
when(device.getRegistrationId(IdentityType.ACI)).thenReturn(registrationId);
final Account account = mock(Account.class);
when(account.getIdentifier(IdentityType.ACI)).thenReturn(accountIdentifier);
when(account.getAccountIdentifier()).thenReturn(accountIdentifier);
accountsManager.recordTransferArchiveUpload(account, deviceId, registrationId, transferArchive).join();
@@ -148,7 +148,7 @@ public class AccountsManagerDeviceTransferIntegrationTest {
when(device.getRegistrationId(IdentityType.ACI)).thenReturn(registrationId);
final Account account = mock(Account.class);
when(account.getIdentifier(IdentityType.ACI)).thenReturn(accountIdentifier);
when(account.getAccountIdentifier()).thenReturn(accountIdentifier);
accountsManager.recordTransferArchiveUpload(account, deviceId,
registrationId, transferArchiveError).join();
@@ -165,7 +165,7 @@ public class AccountsManagerDeviceTransferIntegrationTest {
when(device.getRegistrationId(IdentityType.ACI)).thenReturn(123);
final Account account = mock(Account.class);
when(account.getIdentifier(IdentityType.ACI)).thenReturn(accountIdentifier);
when(account.getAccountIdentifier()).thenReturn(accountIdentifier);
assertEquals(Optional.empty(),
accountsManager.waitForTransferArchive(account, device, Duration.ofMillis(1)).join());