This repository has been archived by the owner on Jan 10, 2025. It is now read-only.
forked from mollyim/mollyim-android
-
Notifications
You must be signed in to change notification settings - Fork 5
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
564 changed files
with
37,789 additions
and
20,419 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
1 change: 1 addition & 0 deletions
1
app/src/androidTest/assets/inAppPaymentsTests/configuration.json
Large diffs are not rendered by default.
Oops, something went wrong.
287 changes: 287 additions & 0 deletions
287
app/src/androidTest/java/org/thoughtcrime/securesms/backup/v2/ArchiveImportExportTests.kt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,287 @@ | ||
| /* | ||
| * Copyright 2024 Signal Messenger, LLC | ||
| * SPDX-License-Identifier: AGPL-3.0-only | ||
| */ | ||
|
|
||
| package org.thoughtcrime.securesms.backup.v2 | ||
|
|
||
| import androidx.test.ext.junit.runners.AndroidJUnit4 | ||
| import androidx.test.platform.app.InstrumentationRegistry | ||
| import com.github.difflib.DiffUtils | ||
| import com.github.difflib.UnifiedDiffUtils | ||
| import junit.framework.Assert.assertTrue | ||
| import org.junit.Ignore | ||
| import org.junit.Test | ||
| import org.junit.runner.RunWith | ||
| import org.signal.core.util.Base64 | ||
| import org.signal.core.util.SqlUtil | ||
| import org.signal.core.util.logging.Log | ||
| import org.signal.core.util.readFully | ||
| import org.signal.libsignal.messagebackup.ComparableBackup | ||
| import org.signal.libsignal.messagebackup.MessageBackup | ||
| import org.signal.libsignal.zkgroup.profiles.ProfileKey | ||
| import org.thoughtcrime.securesms.backup.v2.proto.Frame | ||
| import org.thoughtcrime.securesms.backup.v2.stream.PlainTextBackupReader | ||
| import org.thoughtcrime.securesms.database.DistributionListTables | ||
| import org.thoughtcrime.securesms.database.KeyValueDatabase | ||
| import org.thoughtcrime.securesms.database.SignalDatabase | ||
| import org.thoughtcrime.securesms.dependencies.AppDependencies | ||
| import org.thoughtcrime.securesms.keyvalue.SignalStore | ||
| import org.thoughtcrime.securesms.recipients.RecipientId | ||
| import org.whispersystems.signalservice.api.kbs.MasterKey | ||
| import org.whispersystems.signalservice.api.push.ServiceId | ||
| import java.io.ByteArrayInputStream | ||
| import java.util.UUID | ||
|
|
||
| @RunWith(AndroidJUnit4::class) | ||
| class ArchiveImportExportTests { | ||
|
|
||
| companion object { | ||
| const val TAG = "ImportExport" | ||
| const val TESTS_FOLDER = "backupTests" | ||
|
|
||
| val SELF_ACI = ServiceId.ACI.from(UUID(100, 100)) | ||
| val SELF_PNI = ServiceId.PNI.from(UUID(101, 101)) | ||
| val SELF_E164 = "+10000000000" | ||
| val SELF_PROFILE_KEY: ByteArray = Base64.decode("YQKRq+3DQklInaOaMcmlzZnN0m/1hzLiaONX7gB12dg=") | ||
| val MASTER_KEY = Base64.decode("sHuBMP4ToZk4tcNU+S8eBUeCt8Am5EZnvuqTBJIR4Do") | ||
| } | ||
|
|
||
| @Test | ||
| fun all() { | ||
| runTests() | ||
| } | ||
|
|
||
| @Ignore("Just for debugging") | ||
| @Test | ||
| fun accountData() { | ||
| runTests { it.startsWith("account_data_") } | ||
| } | ||
|
|
||
| @Ignore("Just for debugging") | ||
| @Test | ||
| fun recipientContacts() { | ||
| runTests { it.startsWith("recipient_contacts_") } | ||
| } | ||
|
|
||
| @Ignore("Just for debugging") | ||
| @Test | ||
| fun recipientDistributionLists() { | ||
| runTests { it.startsWith("recipient_distribution_list_") } | ||
| } | ||
|
|
||
| @Ignore("Just for debugging") | ||
| @Test | ||
| fun recipientGroups() { | ||
| runTests { it.startsWith("recipient_groups_") } | ||
| } | ||
|
|
||
| @Ignore("Just for debugging") | ||
| @Test | ||
| fun chatStandardMessageTextOnly() { | ||
| runTests { it.startsWith("chat_standard_message_text_only_") } | ||
| } | ||
|
|
||
| @Ignore("Just for debugging") | ||
| @Test | ||
| fun chatStandardMessageFormattedText() { | ||
| runTests { it.startsWith("chat_standard_message_formatted_text_") } | ||
| } | ||
|
|
||
| @Ignore("Just for debugging") | ||
| @Test | ||
| fun chatStandardMessageLongText() { | ||
| runTests { it.startsWith("chat_standard_message_long_text_") } | ||
| } | ||
|
|
||
| @Ignore("Just for debugging") | ||
| @Test | ||
| fun chatStandardMessageStandardAttachments() { | ||
| runTests { it.startsWith("chat_standard_message_standard_attachments_") } | ||
| } | ||
|
|
||
| @Ignore("Just for debugging") | ||
| @Test | ||
| fun chatStandardMessageSpecialAttachments() { | ||
| runTests { it.startsWith("chat_standard_message_special_attachments_") } | ||
| } | ||
|
|
||
| @Ignore("Just for debugging") | ||
| @Test | ||
| fun chatSimpleUpdates() { | ||
| runTests { it.startsWith("chat_simple_updates_") } | ||
| } | ||
|
|
||
| @Ignore("Just for debugging") | ||
| @Test | ||
| fun chatContactMessage() { | ||
| runTests { it.startsWith("chat_contact_message_") } | ||
| } | ||
|
|
||
| private fun runTests(predicate: (String) -> Boolean = { true }) { | ||
| val testFiles = InstrumentationRegistry.getInstrumentation().context.resources.assets.list(TESTS_FOLDER)!!.filter(predicate) | ||
| val results: MutableList<TestResult> = mutableListOf() | ||
|
|
||
| Log.d(TAG, "About to run ${testFiles.size} tests.") | ||
|
|
||
| for (filename in testFiles) { | ||
| Log.d(TAG, "> $filename") | ||
| val startTime = System.currentTimeMillis() | ||
| val result = test(filename) | ||
| results += result | ||
|
|
||
| if (result is TestResult.Success) { | ||
| Log.d(TAG, " \uD83D\uDFE2 Passed in ${System.currentTimeMillis() - startTime} ms") | ||
| } else { | ||
| Log.d(TAG, " \uD83D\uDD34 Failed in ${System.currentTimeMillis() - startTime} ms") | ||
| } | ||
| } | ||
|
|
||
| results | ||
| .filterIsInstance<TestResult.Failure>() | ||
| .forEach { | ||
| Log.e(TAG, "Failure: ${it.name}\n${it.message}") | ||
| Log.e(TAG, "----------------------------------") | ||
| Log.e(TAG, "----------------------------------") | ||
| Log.e(TAG, "----------------------------------") | ||
| } | ||
|
|
||
| if (results.any { it is TestResult.Failure }) { | ||
| val successCount = results.count { it is TestResult.Success } | ||
| val failingTestNames = results.filterIsInstance<TestResult.Failure>().joinToString(separator = "\n") { " \uD83D\uDD34 ${it.name}" } | ||
| val message = "Some tests failed! Only $successCount/${results.size} passed. Failure details are above. Failing tests:\n$failingTestNames" | ||
|
|
||
| Log.d(TAG, message) | ||
| throw AssertionError(message) | ||
| } else { | ||
| Log.d(TAG, "All ${results.size} tests passed!") | ||
| } | ||
| } | ||
|
|
||
| private fun test(filename: String): TestResult { | ||
| resetAllData() | ||
|
|
||
| val inputFileBytes: ByteArray = InstrumentationRegistry.getInstrumentation().context.resources.assets.open("$TESTS_FOLDER/$filename").readFully(true) | ||
|
|
||
| val importResult = import(inputFileBytes) | ||
| assertTrue(importResult is ImportResult.Success) | ||
| val success = importResult as ImportResult.Success | ||
|
|
||
| val generatedBackupData = BackupRepository.debugExport(plaintext = true, currentTime = success.backupTime) | ||
| checkEquivalent(filename, inputFileBytes, generatedBackupData)?.let { return it } | ||
|
|
||
| return TestResult.Success(filename) | ||
| } | ||
|
|
||
| private fun resetAllData() { | ||
| // Need to delete these first to prevent foreign key crash | ||
| SignalDatabase.rawDatabase.execSQL("DELETE FROM ${DistributionListTables.ListTable.TABLE_NAME}") | ||
| SignalDatabase.rawDatabase.execSQL("DELETE FROM ${DistributionListTables.MembershipTable.TABLE_NAME}") | ||
|
|
||
| SqlUtil.getAllTables(SignalDatabase.rawDatabase) | ||
| .filterNot { it.contains("sqlite") || it.contains("fts") || it.startsWith("emoji_search_") } // If we delete these we'll corrupt the DB | ||
| .sorted() | ||
| .forEach { table -> | ||
| SignalDatabase.rawDatabase.execSQL("DELETE FROM $table") | ||
| SqlUtil.resetAutoIncrementValue(SignalDatabase.rawDatabase, table) | ||
| } | ||
|
|
||
| AppDependencies.recipientCache.clear() | ||
| AppDependencies.recipientCache.clearSelf() | ||
| RecipientId.clearCache() | ||
|
|
||
| KeyValueDatabase.getInstance(AppDependencies.application).clear() | ||
| SignalStore.resetCache() | ||
|
|
||
| SignalStore.svr.setMasterKey(MasterKey(MASTER_KEY), "1234") | ||
| SignalStore.account.setE164(SELF_E164) | ||
| SignalStore.account.setAci(SELF_ACI) | ||
| SignalStore.account.setPni(SELF_PNI) | ||
| SignalStore.account.generateAciIdentityKeyIfNecessary() | ||
| SignalStore.account.generatePniIdentityKeyIfNecessary() | ||
| SignalStore.backup.backupTier = MessageBackupTier.PAID | ||
| } | ||
|
|
||
| private fun import(importData: ByteArray): ImportResult { | ||
| return BackupRepository.import( | ||
| length = importData.size.toLong(), | ||
| inputStreamFactory = { ByteArrayInputStream(importData) }, | ||
| selfData = BackupRepository.SelfData(SELF_ACI, SELF_PNI, SELF_E164, ProfileKey(SELF_PROFILE_KEY)), | ||
| plaintext = true | ||
| ) | ||
| } | ||
|
|
||
| private fun assertPassesValidator(testName: String, generatedBackupData: ByteArray): TestResult.Failure? { | ||
| try { | ||
| BackupRepository.validate( | ||
| length = generatedBackupData.size.toLong(), | ||
| inputStreamFactory = { ByteArrayInputStream(generatedBackupData) }, | ||
| selfData = BackupRepository.SelfData(SELF_ACI, SELF_PNI, SELF_E164, ProfileKey(SELF_PROFILE_KEY)) | ||
| ) | ||
| } catch (e: Exception) { | ||
| return TestResult.Failure(testName, "Generated backup failed validation: ${e.message}") | ||
| } | ||
|
|
||
| return null | ||
| } | ||
|
|
||
| private fun checkEquivalent(testName: String, import: ByteArray, export: ByteArray): TestResult.Failure? { | ||
| val importComparable = try { | ||
| ComparableBackup.readUnencrypted(MessageBackup.Purpose.REMOTE_BACKUP, import.inputStream(), import.size.toLong()) | ||
| } catch (e: Exception) { | ||
| return TestResult.Failure(testName, "Imported backup hit a validation error: ${e.message}") | ||
| } | ||
|
|
||
| val exportComparable = try { | ||
| ComparableBackup.readUnencrypted(MessageBackup.Purpose.REMOTE_BACKUP, export.inputStream(), import.size.toLong()) | ||
| } catch (e: Exception) { | ||
| return TestResult.Failure(testName, "Exported backup hit a validation error: ${e.message}") | ||
| } | ||
|
|
||
| if (importComparable.unknownFieldMessages.isNotEmpty()) { | ||
| return TestResult.Failure(testName, "Imported backup contains unknown fields: ${importComparable.unknownFieldMessages}") | ||
| } | ||
|
|
||
| if (exportComparable.unknownFieldMessages.isNotEmpty()) { | ||
| return TestResult.Failure(testName, "Imported backup contains unknown fields: ${importComparable.unknownFieldMessages}") | ||
| } | ||
|
|
||
| val canonicalImport = importComparable.comparableString | ||
| val canonicalExport = exportComparable.comparableString | ||
|
|
||
| if (canonicalImport != canonicalExport) { | ||
| val importLines = canonicalImport.lines() | ||
| val exportLines = canonicalExport.lines() | ||
|
|
||
| val patch = DiffUtils.diff(importLines, exportLines) | ||
| val diff = UnifiedDiffUtils.generateUnifiedDiff("Import", "Export", importLines, patch, 3).joinToString(separator = "\n") | ||
|
|
||
| val importFrames = import.toFrames() | ||
| val exportFrames = export.toFrames() | ||
|
|
||
| val importGroupFramesByMasterKey = importFrames.mapNotNull { it.recipient?.group }.associateBy { it.masterKey } | ||
| val exportGroupFramesByMasterKey = exportFrames.mapNotNull { it.recipient?.group }.associateBy { it.masterKey } | ||
|
|
||
| val groupErrorMessage = StringBuilder() | ||
|
|
||
| for ((importKey, importValue) in importGroupFramesByMasterKey) { | ||
| if (exportGroupFramesByMasterKey[importKey]?.let { it.snapshot != importValue.snapshot } == true) { | ||
| groupErrorMessage.append("[$importKey] Snapshot mismatch.\nImport:\n${importValue}\n\nExport:\n${exportGroupFramesByMasterKey[importKey]}\n\n") | ||
| } | ||
| } | ||
|
|
||
| return TestResult.Failure(testName, "Imported backup does not match exported backup. Diff:\n$diff\n$groupErrorMessage") | ||
| } | ||
|
|
||
| return null | ||
| } | ||
|
|
||
| fun ByteArray.toFrames(): List<Frame> { | ||
| return PlainTextBackupReader(this.inputStream(), this.size.toLong()).use { it.asSequence().toList() } | ||
| } | ||
|
|
||
| private sealed class TestResult(val name: String) { | ||
| class Success(name: String) : TestResult(name) | ||
| class Failure(name: String, val message: String) : TestResult(name) | ||
| } | ||
| } |
Oops, something went wrong.