diff --git a/Sources/XXMessengerClient/Messenger/Functions/MessengerBackupParams.swift b/Sources/XXMessengerClient/Messenger/Functions/MessengerBackupParams.swift
index f5ac71f04d82b27fd3585acfbfe89b4372034ac2..4307aea58fee33ebbb195a3dbd7dd966ee077ca5 100644
--- a/Sources/XXMessengerClient/Messenger/Functions/MessengerBackupParams.swift
+++ b/Sources/XXMessengerClient/Messenger/Functions/MessengerBackupParams.swift
@@ -6,9 +6,9 @@ public struct MessengerBackupParams {
case notRunning
}
- public var run: (BackupParams) throws -> Void
+ public var run: (String) throws -> Void
- public func callAsFunction(_ params: BackupParams) throws {
+ public func callAsFunction(_ params: String) throws {
try run(params)
}
}
@@ -19,9 +19,7 @@ extension MessengerBackupParams {
guard let backup = env.backup(), backup.isRunning() else {
throw Error.notRunning
}
- let paramsData = try params.encode()
- let paramsString = String(data: paramsData, encoding: .utf8)!
- backup.addJSON(paramsString)
+ backup.addJSON(params)
}
}
}
diff --git a/Sources/XXMessengerClient/Messenger/Functions/MessengerRestoreBackup.swift b/Sources/XXMessengerClient/Messenger/Functions/MessengerRestoreBackup.swift
index 0e5387a3add343ae79d27aed5eaecf752f9fb6fc..da66d5792f095b1c6ba7befea94fc45f16298c29 100644
--- a/Sources/XXMessengerClient/Messenger/Functions/MessengerRestoreBackup.swift
+++ b/Sources/XXMessengerClient/Messenger/Functions/MessengerRestoreBackup.swift
@@ -5,14 +5,14 @@ import XCTestDynamicOverlay
public struct MessengerRestoreBackup {
public struct Result: Equatable {
public init(
- restoredParams: BackupParams,
+ restoredParams: String,
restoredContacts: [Data]
) {
self.restoredParams = restoredParams
self.restoredContacts = restoredContacts
}
- public var restoredParams: BackupParams
+ public var restoredParams: String
public var restoredContacts: [Data]
}
@@ -42,8 +42,6 @@ extension MessengerRestoreBackup {
sessionPassword: password,
backupFileContents: backupData
)
- let paramsData = report.params.data(using: .utf8)!
- let params = try BackupParams.decode(paramsData)
let cMix = try env.loadCMix(
storageDir: storageDir,
password: password,
@@ -68,7 +66,7 @@ extension MessengerRestoreBackup {
)
env.ud.set(ud)
return Result(
- restoredParams: params,
+ restoredParams: report.params,
restoredContacts: report.restoredContacts
)
}
diff --git a/Sources/XXMessengerClient/Messenger/Functions/MessengerStartBackup.swift b/Sources/XXMessengerClient/Messenger/Functions/MessengerStartBackup.swift
index 1512d36100b25a17b2b75175520edf94940ef5a2..cd28535cf9330fb4fd0898f32a399191800d8a04 100644
--- a/Sources/XXMessengerClient/Messenger/Functions/MessengerStartBackup.swift
+++ b/Sources/XXMessengerClient/Messenger/Functions/MessengerStartBackup.swift
@@ -9,11 +9,11 @@ public struct MessengerStartBackup {
case notLoggedIn
}
- public var run: (String, BackupParams) throws -> Void
+ public var run: (String, String?) throws -> Void
public func callAsFunction(
password: String,
- params: BackupParams
+ params: String? = nil
) throws {
try run(password, params)
}
@@ -31,8 +31,6 @@ extension MessengerStartBackup {
guard let ud = env.ud() else {
throw Error.notLoggedIn
}
- let paramsData = try params.encode()
- let paramsString = String(data: paramsData, encoding: .utf8)!
var didAddParams = false
var semaphore: DispatchSemaphore? = .init(value: 0)
let backup = try env.initializeBackup(
@@ -41,9 +39,9 @@ extension MessengerStartBackup {
password: password,
callback: .init { data in
semaphore?.wait()
- if !didAddParams {
+ if let params, !didAddParams {
if let backup = env.backup() {
- backup.addJSON(paramsString)
+ backup.addJSON(params)
didAddParams = true
}
} else {
diff --git a/Sources/XXMessengerClient/Utils/BackupParams.swift b/Sources/XXMessengerClient/Utils/BackupParams.swift
deleted file mode 100644
index 8bf39a5189049f234ec9ab991c63ecdf59b0f1ce..0000000000000000000000000000000000000000
--- a/Sources/XXMessengerClient/Utils/BackupParams.swift
+++ /dev/null
@@ -1,21 +0,0 @@
-import Foundation
-
-public struct BackupParams: Equatable {
- public init(
- username: String
- ) {
- self.username = username
- }
-
- public var username: String
-}
-
-extension BackupParams: Codable {
- public static func decode(_ data: Data) throws -> Self {
- try JSONDecoder().decode(Self.self, from: data)
- }
-
- public func encode() throws -> Data {
- try JSONEncoder().encode(self)
- }
-}
diff --git a/Tests/XXMessengerClientTests/Messenger/Functions/MessengerBackupParamsTests.swift b/Tests/XXMessengerClientTests/Messenger/Functions/MessengerBackupParamsTests.swift
index b65a34b9fbac97ab2a3fc629a7bce09552ae39b1..8a61b85e112ad44d74bb1a622b811dca79b49623 100644
--- a/Tests/XXMessengerClientTests/Messenger/Functions/MessengerBackupParamsTests.swift
+++ b/Tests/XXMessengerClientTests/Messenger/Functions/MessengerBackupParamsTests.swift
@@ -6,6 +6,7 @@ import XXClient
final class MessengerBackupParamsTests: XCTestCase {
func testBackupParams() throws {
var didAddJSON: [String] = []
+ let params = "test-123"
var env: MessengerEnvironment = .unimplemented
env.backup.get = {
@@ -16,11 +17,9 @@ final class MessengerBackupParamsTests: XCTestCase {
}
let backup: MessengerBackupParams = .live(env)
- try backup(.stub)
+ try backup(params)
- XCTAssertNoDifference(didAddJSON, [
- String(data: try BackupParams.stub.encode(), encoding: .utf8)!
- ])
+ XCTAssertNoDifference(didAddJSON, [params])
}
func testBackupParamsWhenNoBackup() {
@@ -28,7 +27,7 @@ final class MessengerBackupParamsTests: XCTestCase {
env.backup.get = { nil }
let backup: MessengerBackupParams = .live(env)
- XCTAssertThrowsError(try backup(.stub)) { error in
+ XCTAssertThrowsError(try backup("")) { error in
XCTAssertNoDifference(
error as NSError,
MessengerBackupParams.Error.notRunning as NSError
@@ -45,7 +44,7 @@ final class MessengerBackupParamsTests: XCTestCase {
}
let backup: MessengerBackupParams = .live(env)
- XCTAssertThrowsError(try backup(.stub)) { error in
+ XCTAssertThrowsError(try backup("")) { error in
XCTAssertNoDifference(
error as NSError,
MessengerBackupParams.Error.notRunning as NSError
diff --git a/Tests/XXMessengerClientTests/Messenger/Functions/MessengerRestoreBackupTests.swift b/Tests/XXMessengerClientTests/Messenger/Functions/MessengerRestoreBackupTests.swift
index 4f819b47393545d46caffb71097dc92ff2f08337..d097c32e137f31601b732f6502e39c87345809f8 100644
--- a/Tests/XXMessengerClientTests/Messenger/Functions/MessengerRestoreBackupTests.swift
+++ b/Tests/XXMessengerClientTests/Messenger/Functions/MessengerRestoreBackupTests.swift
@@ -10,10 +10,9 @@ final class MessengerRestoreBackupTests: XCTestCase {
let ndfData = "ndf-data".data(using: .utf8)!
let password = "password".data(using: .utf8)!
let backupContacts: [Data] = (1...3).map { "contact-\($0)" }.map { $0.data(using: .utf8)! }
- let backupParams = BackupParams.stub
let backupReport = BackupReport(
restoredContacts: backupContacts,
- params: String(data: try! JSONEncoder().encode(backupParams), encoding: .utf8)!
+ params: "backup-report-params"
)
let cMixParams = "cmix-params".data(using: .utf8)!
let e2eParams = "e2e-params".data(using: .utf8)!
@@ -157,7 +156,7 @@ final class MessengerRestoreBackupTests: XCTestCase {
])
XCTAssertNoDifference(result, MessengerRestoreBackup.Result(
- restoredParams: backupParams,
+ restoredParams: backupReport.params,
restoredContacts: backupContacts
))
}
diff --git a/Tests/XXMessengerClientTests/Messenger/Functions/MessengerStartBackupTests.swift b/Tests/XXMessengerClientTests/Messenger/Functions/MessengerStartBackupTests.swift
index b92bd1ee38948737453e973a8e4b4cd874797b84..df798488401675ca4bf8be72e1edc3a4be553068 100644
--- a/Tests/XXMessengerClientTests/Messenger/Functions/MessengerStartBackupTests.swift
+++ b/Tests/XXMessengerClientTests/Messenger/Functions/MessengerStartBackupTests.swift
@@ -4,7 +4,7 @@ import XXClient
@testable import XXMessengerClient
final class MessengerStartBackupTests: XCTestCase {
- func testStart() throws {
+ func testStartWithParams() throws {
struct InitBackupParams: Equatable {
var e2eId: Int
var udId: Int
@@ -16,6 +16,7 @@ final class MessengerStartBackupTests: XCTestCase {
var didSetBackup: [Backup?] = []
var didAddJSON: [String] = []
+ let params = "backup-params"
let password = "test-password"
let e2eId = 123
let udId = 321
@@ -49,7 +50,7 @@ final class MessengerStartBackupTests: XCTestCase {
}
let start: MessengerStartBackup = .live(env)
- try start(password: password, params: .stub)
+ try start(password: password, params: params)
XCTAssertNoDifference(didInitializeBackup, [
.init(e2eId: e2eId, udId: udId, password: password)
@@ -62,9 +63,7 @@ final class MessengerStartBackupTests: XCTestCase {
didHandleCallback.map(StringData.init),
[].map(StringData.init)
)
- XCTAssertNoDifference(didAddJSON, [
- String(data: try BackupParams.stub.encode(), encoding: .utf8)!
- ])
+ XCTAssertNoDifference(didAddJSON, [params])
backupCallbacks.forEach { $0.handle(dataWithParams) }
@@ -74,6 +73,60 @@ final class MessengerStartBackupTests: XCTestCase {
)
}
+ func testStartWithoutParams() throws {
+ struct InitBackupParams: Equatable {
+ var e2eId: Int
+ var udId: Int
+ var password: String
+ }
+ var didInitializeBackup: [InitBackupParams] = []
+ var backupCallbacks: [UpdateBackupFunc] = []
+ var didHandleCallback: [Data] = []
+ var didSetBackup: [Backup?] = []
+
+ let password = "test-password"
+ let e2eId = 123
+ let udId = 321
+ let dataWithoutParams = "backup-without-params".data(using: .utf8)!
+
+ var env: MessengerEnvironment = .unimplemented
+ env.backup.get = { didSetBackup.last as? Backup }
+ env.backup.set = { didSetBackup.append($0) }
+ env.e2e.get = {
+ var e2e: E2E = .unimplemented
+ e2e.getId.run = { e2eId }
+ return e2e
+ }
+ env.ud.get = {
+ var ud: UserDiscovery = .unimplemented
+ ud.getId.run = { udId }
+ return ud
+ }
+ env.backupCallbacks.registered = {
+ UpdateBackupFunc { didHandleCallback.append($0) }
+ }
+ env.initializeBackup.run = { e2eId, udId, password, callback in
+ didInitializeBackup.append(.init(e2eId: e2eId, udId: udId, password: password))
+ backupCallbacks.append(callback)
+ return .unimplemented
+ }
+ let start: MessengerStartBackup = .live(env)
+
+ try start(password: password)
+
+ XCTAssertNoDifference(didInitializeBackup, [
+ .init(e2eId: e2eId, udId: udId, password: password)
+ ])
+ XCTAssertNoDifference(didSetBackup.map { $0 != nil }, [true])
+
+ backupCallbacks.forEach { $0.handle(dataWithoutParams) }
+
+ XCTAssertNoDifference(
+ didHandleCallback.map(StringData.init),
+ [dataWithoutParams].map(StringData.init)
+ )
+ }
+
func testStartWhenRunning() {
var env: MessengerEnvironment = .unimplemented
env.backup.get = {
@@ -83,7 +136,7 @@ final class MessengerStartBackupTests: XCTestCase {
}
let start: MessengerStartBackup = .live(env)
- XCTAssertThrowsError(try start(password: "", params: .stub)) { error in
+ XCTAssertThrowsError(try start(password: "")) { error in
XCTAssertNoDifference(
error as NSError,
MessengerStartBackup.Error.isRunning as NSError
@@ -97,7 +150,7 @@ final class MessengerStartBackupTests: XCTestCase {
env.e2e.get = { nil }
let start: MessengerStartBackup = .live(env)
- XCTAssertThrowsError(try start(password: "", params: .stub)) { error in
+ XCTAssertThrowsError(try start(password: "")) { error in
XCTAssertNoDifference(
error as NSError,
MessengerStartBackup.Error.notConnected as NSError
@@ -112,7 +165,7 @@ final class MessengerStartBackupTests: XCTestCase {
env.ud.get = { nil }
let start: MessengerStartBackup = .live(env)
- XCTAssertThrowsError(try start(password: "", params: .stub)) { error in
+ XCTAssertThrowsError(try start(password: "")) { error in
XCTAssertNoDifference(
error as NSError,
MessengerStartBackup.Error.notLoggedIn as NSError
@@ -140,7 +193,7 @@ final class MessengerStartBackupTests: XCTestCase {
env.initializeBackup.run = { _, _, _, _ in throw failure }
let start: MessengerStartBackup = .live(env)
- XCTAssertThrowsError(try start(password: "", params: .stub)) { error in
+ XCTAssertThrowsError(try start(password: "")) { error in
XCTAssertNoDifference(error as NSError, failure as NSError)
}
}
diff --git a/Tests/XXMessengerClientTests/TestHelpers/TestDoubles.swift b/Tests/XXMessengerClientTests/TestHelpers/TestDoubles.swift
index 93110b2efb3698aec563f2f6cc9a0e51f03189a9..e43d76accbb65a1f7f58528e4f5a6c443f48e394 100644
--- a/Tests/XXMessengerClientTests/TestHelpers/TestDoubles.swift
+++ b/Tests/XXMessengerClientTests/TestHelpers/TestDoubles.swift
@@ -18,12 +18,6 @@ extension Message {
}
}
-extension BackupParams {
- static let stub = BackupParams(
- username: "stub-username"
- )
-}
-
extension ReceivedFile {
static func stub(_ id: Int) -> ReceivedFile {
ReceivedFile(