Signal/Android
1
0
Fork 0
mirror of https://github.com/signalapp/Signal-Android.git synced 2025-12-23 20:48:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add tests to generate backup with large amount of messages and chats.

Browse Source
This commit is contained in:
Clark
2024-03-15 09:34:57 -04:00
committed by Cody Henthorne
parent 359e593481
commit 989bd662c6
1 changed files with 223 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

223
app/src/androidTest/java/org/thoughtcrime/securesms/backup/v2/ImportExportTest.kt
View File

@@ -153,6 +153,229 @@ class ImportExportTest {
importExport(*standardFrames)
}
@Test
fun largeNumberOfRecipientsAndChats() {
val recipients = ArrayList<Recipient>(5000)
val chats = ArrayList<Chat>(5000)
var id = 3L
for (i in 0..5000) {
val recipientId = id++
recipients.add(
Recipient(
id = recipientId,
contact = Contact(
aci = TestRecipientUtils.nextAci().toByteString(),
pni = TestRecipientUtils.nextPni().toByteString(),
username = "rec$i.01",
e164 = 14125550000 + i,
blocked = false,
hidden = false,
registered = Contact.Registered.REGISTERED,
unregisteredTimestamp = 0L,
profileKey = TestRecipientUtils.generateProfileKey().toByteString(),
profileSharing = true,
profileGivenName = "Test",
profileFamilyName = "Recipient$i",
hideStory = false
)
)
)
chats.add(
Chat(
id = recipientId - 2L,
recipientId = recipientId
)
)
if (i % 10 == 0) {
val groupId = id++
recipients.add(
Recipient(
id = groupId,
group = Group(
masterKey = TestRecipientUtils.generateGroupMasterKey().toByteString(),
whitelisted = true,
hideStory = false,
storySendMode = Group.StorySendMode.ENABLED,
name = "Cool Group $i"
)
)
)
chats.add(
Chat(
id = groupId - 2L,
recipientId = groupId
)
)
}
}
importExport(
*standardFrames,
*recipients.toArray()
)
}
@Test
fun largeNumberOfMessagesAndChats() {
val NUM_INDIVIDUAL_RECIPIENTS = 1000
val numIndividualMessages = 500
val numGroupMessagesPerPerson = 200
val recipients = ArrayList<Recipient>(1010)
val chats = ArrayList<Chat>(1010)
var id = 3L
for (i in 0 until NUM_INDIVIDUAL_RECIPIENTS) {
val recipientId = id++
recipients.add(
Recipient(
id = recipientId,
contact = Contact(
aci = TestRecipientUtils.nextAci().toByteString(),
pni = TestRecipientUtils.nextPni().toByteString(),
username = "rec$i.01",
e164 = 14125550000 + i,
blocked = false,
hidden = false,
registered = Contact.Registered.REGISTERED,
unregisteredTimestamp = 0L,
profileKey = TestRecipientUtils.generateProfileKey().toByteString(),
profileSharing = true,
profileGivenName = "Test",
profileFamilyName = "Recipient$i",
hideStory = false
)
)
)
chats.add(
Chat(
id = recipientId - 2L,
recipientId = recipientId
)
)
if (i % 100 == 0) {
val groupId = id++
recipients.add(
Recipient(
id = groupId,
group = Group(
masterKey = TestRecipientUtils.generateGroupMasterKey().toByteString(),
whitelisted = true,
hideStory = false,
storySendMode = Group.StorySendMode.ENABLED,
name = "Cool Group $i"
)
)
)
chats.add(
Chat(
id = groupId - 2L,
recipientId = groupId
)
)
}
}
val chatItems = ArrayList<ChatItem>()
var sentTime = 1L
val groupMembers = ArrayList<Recipient>()
var group: Recipient? = null
for (recipient in recipients) {
// Make another group and populate it with messages from these members
if (recipient.group != null) {
if (group == null) {
group = recipient
groupMembers.clear()
} else {
for (member in groupMembers) {
for (i in 0 until numGroupMessagesPerPerson) {
chatItems.add(
ChatItem(
chatId = group.id - 2L,
authorId = member.id,
dateSent = sentTime++,
sms = false,
incoming = ChatItem.IncomingMessageDetails(
dateReceived = sentTime + 1,
dateServerSent = sentTime,
read = true,
sealedSender = true
),
standardMessage = StandardMessage(
text = Text(
body = "Medium length message from ${member.contact?.profileGivenName} ${member.contact?.profileFamilyName} sent at $sentTime"
)
)
)
)
}
}
for (i in 0 until numGroupMessagesPerPerson) {
ChatItem(
chatId = group.id - 2L,
authorId = selfRecipient.id,
dateSent = sentTime++,
sms = false,
outgoing = ChatItem.OutgoingMessageDetails(
sendStatus = groupMembers.map { groupMember ->
SendStatus(recipientId = groupMember.id, deliveryStatus = SendStatus.Status.READ, sealedSender = true)
}
),
standardMessage = StandardMessage(
text = Text(
body = "Outgoing message without much text in it just because"
)
)
)
}
}
} else {
groupMembers.add(recipient)
for (i in 0 until numIndividualMessages) {
if (i % 2 == 0) {
ChatItem(
chatId = 1,
authorId = selfRecipient.id,
dateSent = sentTime++,
sms = false,
outgoing = ChatItem.OutgoingMessageDetails(
sendStatus = listOf(
SendStatus(recipient.id, deliveryStatus = SendStatus.Status.READ, sealedSender = true)
)
),
standardMessage = StandardMessage(
text = Text(
body = "Outgoing message without much text in it just because"
)
)
)
} else {
ChatItem(
chatId = 1,
authorId = selfRecipient.id,
dateSent = sentTime++,
sms = false,
incoming = ChatItem.IncomingMessageDetails(
dateReceived = sentTime + 1,
dateServerSent = sentTime,
read = true,
sealedSender = true
),
standardMessage = StandardMessage(
text = Text(
body = "Outgoing message without much text in it just because"
)
)
)
}
}
}
}
val import = exportFrames(
*standardFrames,
*recipients.toArray(),
*chatItems.toArray()
)
outputFile(import)
}
@Test
fun individualRecipients() {
importExport(