diff --git a/Sources/XXMessengerClient/Messenger/Functions/MessengerRestoreBackup.swift b/Sources/XXMessengerClient/Messenger/Functions/MessengerRestoreBackup.swift index 48a86f3eccf997ec9e1d758c65f14247de07343c..8b05e7d44dba67916bc2d0f5e0d99d005dc41e5b 100644 --- a/Sources/XXMessengerClient/Messenger/Functions/MessengerRestoreBackup.swift +++ b/Sources/XXMessengerClient/Messenger/Functions/MessengerRestoreBackup.swift @@ -48,6 +48,7 @@ extension MessengerRestoreBackup { password: password, cMixParamsJSON: env.getCMixParams() ) + try cMix.startNetworkFollower(timeoutMS: 30_000) let e2e = try env.login( cMixId: cMix.getId(), authCallbacks: env.authCallbacks.registered(), diff --git a/Tests/XXMessengerClientTests/Messenger/Functions/MessengerRestoreBackupTests.swift b/Tests/XXMessengerClientTests/Messenger/Functions/MessengerRestoreBackupTests.swift index 05b4b26e3ae30909f95c978f88d927337c17c4cf..21194303b1ee245150ff1764bfc8968f38569855 100644 --- a/Tests/XXMessengerClientTests/Messenger/Functions/MessengerRestoreBackupTests.swift +++ b/Tests/XXMessengerClientTests/Messenger/Functions/MessengerRestoreBackupTests.swift @@ -71,6 +71,9 @@ final class MessengerRestoreBackupTests: XCTestCase { caughtActions.append(.cMixDidMakeReceptionIdentity(legacy: legacy)) return receptionIdentity } + cMix.startNetworkFollower.run = { timeoutMS in + caughtActions.append(.cMixDidStartNetworkFollower(timeoutMS: timeoutMS)) + } return cMix } env.login.run = { ephemeral, cMixId, _, identity, e2eParams in @@ -130,6 +133,9 @@ final class MessengerRestoreBackupTests: XCTestCase { password: password, cMixParams: cMixParams ), + .cMixDidStartNetworkFollower( + timeoutMS: 30_000 + ), .cMixDidMakeReceptionIdentity( legacy: true ), @@ -213,6 +219,9 @@ private enum CaughtAction: Equatable { case cMixDidMakeReceptionIdentity( legacy: Bool ) + case cMixDidStartNetworkFollower( + timeoutMS: Int + ) case didNewUdManagerFromBackup( params: NewUdManagerFromBackup.Params )