@@ -20,6 +20,7 @@ import android.util.Log
2020import org.amshove.kluent.fail
2121import org.junit.Assert.assertEquals
2222import org.junit.Assert.assertNotNull
23+ import org.junit.Assert.assertNull
2324import org.junit.Assert.assertTrue
2425import org.matrix.android.sdk.api.auth.UIABaseAuth
2526import org.matrix.android.sdk.api.auth.UserInteractiveAuthInterceptor
@@ -43,7 +44,9 @@ import org.matrix.android.sdk.api.session.crypto.verification.IncomingSasVerific
4344import org.matrix.android.sdk.api.session.crypto.verification.OutgoingSasVerificationTransaction
4445import org.matrix.android.sdk.api.session.crypto.verification.VerificationMethod
4546import org.matrix.android.sdk.api.session.crypto.verification.VerificationTxState
47+ import org.matrix.android.sdk.api.session.events.model.Event
4648import org.matrix.android.sdk.api.session.events.model.EventType
49+ import org.matrix.android.sdk.api.session.events.model.toContent
4750import org.matrix.android.sdk.api.session.events.model.toModel
4851import org.matrix.android.sdk.api.session.getRoom
4952import org.matrix.android.sdk.api.session.room.model.Membership
@@ -53,6 +56,7 @@ import org.matrix.android.sdk.api.session.room.model.message.MessageContent
5356import org.matrix.android.sdk.api.session.room.roomSummaryQueryParams
5457import org.matrix.android.sdk.api.session.securestorage.EmptyKeySigner
5558import org.matrix.android.sdk.api.session.securestorage.KeyRef
59+ import org.matrix.android.sdk.api.util.awaitCallback
5660import org.matrix.android.sdk.api.util.toBase64NoPadding
5761import java.util.UUID
5862import kotlin.coroutines.Continuation
@@ -300,10 +304,10 @@ class CryptoTestHelper(val testHelper: CommonTestHelper) {
300304 )
301305
302306 // set up megolm backup
303- val creationInfo = testHelper.waitForCallback <MegolmBackupCreationInfo > {
304- session.cryptoService().keysBackupService().prepareKeysBackupVersion(null , null , it)
307+ val creationInfo = awaitCallback <MegolmBackupCreationInfo > {
308+ session.cryptoService().keysBackupService().prepareKeysBackupVersion(null , null , null , it)
305309 }
306- val version = testHelper.waitForCallback <KeysVersion > {
310+ val version = awaitCallback <KeysVersion > {
307311 session.cryptoService().keysBackupService().createKeysBackupVersion(creationInfo, it)
308312 }
309313 // Save it for gossiping
@@ -315,24 +319,6 @@ class CryptoTestHelper(val testHelper: CommonTestHelper) {
315319 secret,
316320 listOf (KeyRef (keyInfo.keyId, keyInfo.keySpec))
317321 )
318-
319- // set up megolm backup
320- val creationInfo = awaitCallback<MegolmBackupCreationInfo > {
321- session.cryptoService().keysBackupService().prepareKeysBackupVersion(null , null , null , it)
322- }
323- val version = awaitCallback<KeysVersion > {
324- session.cryptoService().keysBackupService().createKeysBackupVersion(creationInfo, it)
325- }
326- // Save it for gossiping
327- session.cryptoService().keysBackupService().saveBackupRecoveryKey(creationInfo.recoveryKey, version = version.version)
328-
329- extractCurveKeyFromRecoveryKey(creationInfo.recoveryKey)?.toBase64NoPadding()?.let { secret ->
330- ssssService.storeSecret(
331- KEYBACKUP_SECRET_SSSS_NAME ,
332- secret,
333- listOf (KeyRef (keyInfo.keyId, keyInfo.keySpec))
334- )
335- }
336322 }
337323 }
338324
0 commit comments