Update wallet sync logic and add onboarding tests

Features/Onboarding/Package.swift

@@ -54,6 +54,9 @@ let package = Package(
             dependencies: [
                 "Onboarding",
                 .product(name: "WalletServiceTestKit", package: "FeatureServices"),
+                .product(name: "KeystoreTestKit", package: "Keystore"),
+                .product(name: "NameServiceTestKit", package: "ChainServices"),
+                .product(name: "StoreTestKit", package: "Store"),
             ],
             path: "Tests"
         )

Features/Onboarding/Sources/ViewModels/CreateWalletModel.swift

@@ -53,6 +53,7 @@ extension CreateWalletModel {
             source: .create
         )
         walletService.acceptTerms()
+        WalletPreferences(walletId: wallet.walletId).completeInitialSynchronization()
         return wallet
     }
 

Features/Onboarding/Sources/ViewModels/ImportWalletViewModel.swift

@@ -6,7 +6,6 @@ import Primitives
 import WalletService
 import AvatarService
 import PrimitivesComponents
-import Preferences
 import enum Keystore.KeystoreImportType
 
 @Observable
@@ -51,7 +50,6 @@ extension ImportWalletViewModel {
     func importWallet(data: WalletImportData) async throws -> Wallet {
         let wallet = try await walletService.loadOrCreateWallet(name: data.name, type: data.keystoreType, source: .import)
         walletService.acceptTerms()
-        WalletPreferences(walletId: wallet.walletId).completeInitialSynchronization()
         try await walletService.setCurrent(wallet: wallet)
         return wallet
     }

Features/Onboarding/Tests/CreateWalletModelTests.swift

@@ -0,0 +1,30 @@
+// Copyright (c). Gem Wallet. All rights reserved.
+
+import Foundation
+import Testing
+import Preferences
+import Keystore
+import KeystoreTestKit
+import StoreTestKit
+import WalletServiceTestKit
+
+@testable import Onboarding
+
+@MainActor
+struct CreateWalletModelTests {
+
+    @Test
+    func createWalletSetsAddressStatus() async throws {
+        let model = CreateWalletModel(
+            walletService: .mock(keystore: KeystoreMock()),
+            avatarService: .init(store: .mock()),
+            onComplete: nil
+        )
+
+        let wallet = try await model.createWallet(words: LocalKeystore.words)
+        let preferences = WalletPreferences(walletId: wallet.walletId)
+
+        #expect(preferences.completeInitialAddressStatus)
+        #expect(preferences.completeInitialLoadAssets)
+    }
+}

Features/Onboarding/Tests/ImportWalletViewModelTests.swift

@@ -0,0 +1,35 @@
+// Copyright (c). Gem Wallet. All rights reserved.
+
+import Foundation
+import Testing
+import Preferences
+import Keystore
+import KeystoreTestKit
+import NameServiceTestKit
+import StoreTestKit
+import WalletServiceTestKit
+
+@testable import Onboarding
+
+@MainActor
+struct ImportWalletViewModelTests {
+
+    @Test
+    func importWalletDoesNotSetAddressStatus() async throws {
+        let model = ImportWalletViewModel(
+            walletService: .mock(keystore: KeystoreMock()),
+            avatarService: .init(store: .mock()),
+            nameService: .mock(),
+            onComplete: nil
+        )
+
+        let wallet = try await model.importWallet(data: WalletImportData(
+            name: "Test",
+            keystoreType: .phrase(words: LocalKeystore.words, chains: [.tron])
+        ))
+        let preferences = WalletPreferences(walletId: wallet.walletId)
+
+        #expect(preferences.completeInitialAddressStatus == false)
+        #expect(preferences.completeInitialLoadAssets == false)
+    }
+}

Packages/Preferences/Sources/WalletPreferences.swift

@@ -44,10 +44,6 @@ public final class WalletPreferences: @unchecked Sendable {
         get { defaults.bool(forKey: Keys.completeInitialAddressStatus) }
     }
 
-    public var isCompleteInitialSynchronization: Bool {
-        completeInitialAddressStatus && completeInitialLoadAssets
-    }
-
     public func completeInitialSynchronization() {
         completeInitialAddressStatus = true
         completeInitialLoadAssets = true