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

Update backup media metrics on all TTL refreshes

Browse Source
This commit is contained in:
Ravi Khadiwala
2025-09-18 12:30:14 -05:00
committed by ravi-signal
parent 1bba30a81e
commit a647c1bfdf
3 changed files with 43 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
service/src/test/java/org/whispersystems/textsecuregcm/backup/BackupsDbTest.java
View File

@@ -54,22 +54,30 @@ public class BackupsDbTest {
@Test
public void trackMediaStats() {
final AuthenticatedBackupUser backupUser = backupUser(TestRandomUtil.nextBytes(16), BackupCredentialType.MEDIA, BackupLevel.PAID);
// add at least one message backup so we can describe it
backupsDb.addMessageBackup(backupUser).join();
int total = 0;
for (int i = 0; i < 5; i++) {
this.backupsDb.trackMedia(backupUser, 1, i).join();
total += i;
final BackupsDb.BackupDescription description = this.backupsDb.describeBackup(backupUser).join();
assertThat(description.mediaUsedSpace().get()).isEqualTo(total);
final StoredBackupAttributes storedAttrs = backupsDb.ttlRefresh(backupUser).join();
assertThat(description.mediaUsedSpace().orElseThrow())
.isEqualTo(total)
.isEqualTo(storedAttrs.bytesUsed());
assertThat(storedAttrs.numObjects()).isEqualTo(i + 1);
}
for (int i = 0; i < 5; i++) {
this.backupsDb.trackMedia(backupUser, -1, -i).join();
total -= i;
final BackupsDb.BackupDescription description = this.backupsDb.describeBackup(backupUser).join();
assertThat(description.mediaUsedSpace().get()).isEqualTo(total);
final StoredBackupAttributes storedAttrs = backupsDb.ttlRefresh(backupUser).join();
assertThat(description.mediaUsedSpace().orElseThrow())
.isEqualTo(total)
.isEqualTo(storedAttrs.bytesUsed());
assertThat(storedAttrs.numObjects()).isEqualTo(5 - i - 1);
}
}
@ParameterizedTest