From ebee3f72e6a8be59857f337501760cf9ac74315b Mon Sep 17 00:00:00 2001
From: Clark <clark@signal.org>
Date: Fri, 21 Jun 2024 10:58:35 -0400
Subject: [PATCH] Add test infrastructure for backup binprotos.

---
 .../assets/backupTests/account-data.binproto  | Bin 0 -> 218 bytes
 .../registered-blocked-contact.binproto       | Bin 0 -> 204 bytes
 .../story-distribution-list.binproto          | Bin 0 -> 454 bytes
 .../backupTests/unregistered-contact.binproto | Bin 0 -> 209 bytes
 .../backup/v2/ImportExportTestSuite.kt        |  74 ++++++++++++++++++
 5 files changed, 74 insertions(+)
 create mode 100644 app/src/androidTest/assets/backupTests/account-data.binproto
 create mode 100644 app/src/androidTest/assets/backupTests/registered-blocked-contact.binproto
 create mode 100644 app/src/androidTest/assets/backupTests/story-distribution-list.binproto
 create mode 100644 app/src/androidTest/assets/backupTests/unregistered-contact.binproto
 create mode 100644 app/src/androidTest/java/org/thoughtcrime/securesms/backup/v2/ImportExportTestSuite.kt

diff --git a/app/src/androidTest/assets/backupTests/account-data.binproto b/app/src/androidTest/assets/backupTests/account-data.binproto
new file mode 100644
index 0000000000000000000000000000000000000000..54ebb6e04ae667d9af177e3fb99b0799c6a15c9a
GIT binary patch
literal 218
zcmV<0044tk2muh-`k<fpG0p)B%K-`?VFHn>?ZZMzNS&jaG0COPna$E~^@lRzXyaGz
z0Cm~e5)5K*VqsrqWps2dHZ~eI3Ls@?TyxM4un+~~yL?JV@|MeGRQHL2BZw9071$*G
zE|U@vVZi~S0M$YliNXaK*hzZu7z82&LT_SWDg;JlbaXl?3LxyeXvZ`&8ZM(V&8J7n
zj73vrB4Jo7PtIE4-l$E^Wf>9!RZ~P50ZJ+e0T2Ng0Vn}50XUExLI?1lkB;xSk4OPf
U0ayWG0cZhm0eAs`0Re~s0{WFwlmGw#

literal 0
HcmV?d00001

diff --git a/app/src/androidTest/assets/backupTests/registered-blocked-contact.binproto b/app/src/androidTest/assets/backupTests/registered-blocked-contact.binproto
new file mode 100644
index 0000000000000000000000000000000000000000..439129a11cd9a7ccfa2015c1cae0e59cb4e916b9
GIT binary patch
literal 204
zcmV;-05ksy2muh-`k<fpF*yo13Ls$uk*n>)LP<!Sqna_vrOlbm(r@*LGU90CSMC6H
z+1MfkLT_SWDg;JlbaYAw5CL!ihyely5(EeVDgbg4a0mhtYzh!Sc9~uUF+f78<Q`|I
zee+ln5GMQrqPkd0eXxO@qB(I_3K|P&VQyb@Z)|TaG&CUhiTsE{C;>15N+6tl4PT#&
zs?r|wnLx|le4`{yAJPplac9})^;BM><N8_yNMUYb1XFKpZ)gEJ00tTa2mufQ1{wqi
G0uTaOI7%G=

literal 0
HcmV?d00001

diff --git a/app/src/androidTest/assets/backupTests/story-distribution-list.binproto b/app/src/androidTest/assets/backupTests/story-distribution-list.binproto
new file mode 100644
index 0000000000000000000000000000000000000000..369e68ab6bd517b36d999044165d032257e02ec3
GIT binary patch
literal 454
zcmd<&U=-N+Zu8kchK^kJTndZB10Q7eB(6HV{BlP9&#*;1(_@{CZ}MBl|BJ2CG_=!F
zVhb%!EmCmw66Ro(U{qlAV2of)U@Ty)VC-OIViRKFVANu86LR5T5^~@Y2#pD!E@S9@
ztWi>h^V+2^8@v=)RV=vT3~P-xtQ9NPNZXjRKU;GB&1Bmy`{u$-F{Q+VQ5+tLc?!Y#
zIr&x$CPGFW%tCrVW1>-w5fTWELKRWixNY}O%K%1*CKj-!2vj9NO_8V~3Y%b><b`B8
zSd}EX1j2eVA9MY6w_h>uhv+X2Yb$}i=WphIHw4MEf#gF&QH>Dj0?TU&X>hPBsc;D}
zpaCfXE)L&Hh2WC>qDlcq2_^+*10g*Q4kaxvftl}LZkeC!R?`&q{$BI#{<l(MTs*#s
tc`1oG`9+zDdBqUr3M{HZ${d_Z@>~L`UF;}<DaFmj>QP!!QVfg+W&o`pkYxY>

literal 0
HcmV?d00001

diff --git a/app/src/androidTest/assets/backupTests/unregistered-contact.binproto b/app/src/androidTest/assets/backupTests/unregistered-contact.binproto
new file mode 100644
index 0000000000000000000000000000000000000000..cf9413316047bdd1e5eb0dbbcf67afdb79f5fe52
GIT binary patch
literal 209
zcmV;?051Ot2muh-`k<fpF*yo13Ls$uk*n>)LP<!Sqna_vrOlbm(r@*LGU90CSMC6H
z+1MfkLT_SWDg;JlbaYAw5CL!ihyely5(EeVDgbv9bqE3yaS9MXc9~uUF+f78<Q`|I
zee+ln5GMQrqPkd0eXxO@qB(I_3K|P&VQyb@Z)|TaG&CUhiTsE{N+6tl4PT#&s?r|w
znLx|le4`{yAJPplac9})^;BM><N8nmS_4R7Zej#eZ)|U90YV1|;H9L2?lA@$1PB2T
L0R|cb2m%lSSdC5t

literal 0
HcmV?d00001

diff --git a/app/src/androidTest/java/org/thoughtcrime/securesms/backup/v2/ImportExportTestSuite.kt b/app/src/androidTest/java/org/thoughtcrime/securesms/backup/v2/ImportExportTestSuite.kt
new file mode 100644
index 0000000000..3aa19b3105
--- /dev/null
+++ b/app/src/androidTest/java/org/thoughtcrime/securesms/backup/v2/ImportExportTestSuite.kt
@@ -0,0 +1,74 @@
+/*
+ * Copyright 2024 Signal Messenger, LLC
+ * SPDX-License-Identifier: AGPL-3.0-only
+ */
+
+package org.thoughtcrime.securesms.backup.v2
+
+import androidx.test.platform.app.InstrumentationRegistry
+import org.junit.Before
+import org.junit.Test
+import org.junit.runner.RunWith
+import org.junit.runners.Parameterized
+import org.signal.core.util.Base64
+import org.signal.core.util.StreamUtil
+import org.signal.libsignal.zkgroup.profiles.ProfileKey
+import org.thoughtcrime.securesms.keyvalue.SignalStore
+import org.whispersystems.signalservice.api.kbs.MasterKey
+import org.whispersystems.signalservice.api.push.ServiceId
+import java.io.ByteArrayInputStream
+import java.util.UUID
+import kotlin.random.Random
+
+@RunWith(Parameterized::class)
+class ImportExportTestSuite(private val path: String) {
+  companion object {
+    val SELF_ACI = ServiceId.ACI.from(UUID.fromString("77770000-b477-4f35-a824-d92987a63641"))
+    val SELF_PNI = ServiceId.PNI.from(UUID.fromString("77771111-b014-41fb-bf73-05cb2ec52910"))
+    const val SELF_E164 = "+10000000000"
+    val SELF_PROFILE_KEY = ProfileKey(Random.nextBytes(32))
+    val MASTER_KEY = Base64.decode("sHuBMP4ToZk4tcNU+S8eBUeCt8Am5EZnvuqTBJIR4Do")
+
+    const val TESTS_FOLDER = "backupTests"
+
+    @JvmStatic
+    @Parameterized.Parameters(name = "{0}")
+    fun data(): Collection<Array<String>> {
+      val testFiles = InstrumentationRegistry.getInstrumentation().context.resources.assets.list(TESTS_FOLDER)
+      return testFiles?.map { arrayOf(it) }!!.toList()
+    }
+  }
+
+  @Before
+  fun setup() {
+    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()
+  }
+
+  @Test
+  fun testBinProto() {
+    val binProtoBytes: ByteArray = InstrumentationRegistry.getInstrumentation().context.resources.assets.open("${TESTS_FOLDER}/$path").use {
+      StreamUtil.readFully(it)
+    }
+    import(binProtoBytes)
+    val generatedBackupData = BackupRepository.export()
+    compare(binProtoBytes, generatedBackupData)
+  }
+
+  private fun import(importData: ByteArray) {
+    BackupRepository.import(
+      length = importData.size.toLong(),
+      inputStreamFactory = { ByteArrayInputStream(importData) },
+      selfData = BackupRepository.SelfData(SELF_ACI, SELF_PNI, SELF_E164, SELF_PROFILE_KEY),
+      plaintext = true
+    )
+  }
+
+  // TODO compare with libsignal's library
+  private fun compare(import: ByteArray, export: ByteArray) {
+  }
+}