zashi-ios-wallet-private/secantTests/AppReducer/AppReducerTests.swift

//
//  AppReducerTests.swift
//  secantTests
//
//  Created by Lukáš Korba on 12.04.2022.
//

import XCTest
@testable import secant_testnet
import ComposableArchitecture

class AppReducerTests: XCTestCase {
    static let testScheduler = DispatchQueue.test

    let testEnvironment = AppEnvironment(
        wrappedSDKSynchronizer: MockWrappedSDKSynchronizer(),
        databaseFiles: .throwing,
        mnemonicSeedPhraseProvider: .mock,
        scheduler: testScheduler.eraseToAnyScheduler(),
        walletStorage: .throwing,
        wrappedDerivationTool: .live(),
        zcashSDKEnvironment: .mainnet
    )

    func testWalletInitializationState_Uninitialized() throws {
        let uninitializedEnvironment = AppEnvironment(
            wrappedSDKSynchronizer: MockWrappedSDKSynchronizer(),
            databaseFiles: .throwing,
            mnemonicSeedPhraseProvider: .mock,
            scheduler: DispatchQueue.test.eraseToAnyScheduler(),
            walletStorage: .throwing,
            wrappedDerivationTool: .live(),
            zcashSDKEnvironment: .mainnet
        )

        let walletState = AppReducer.walletInitializationState(uninitializedEnvironment)
        
        XCTAssertEqual(walletState, .uninitialized)
    }

    func testWalletInitializationState_FilesPresentKeysMissing() throws {
        let wfmMock = WrappedFileManager(
            url: { _, _, _, _ in URL(fileURLWithPath: "") },
            fileExists: { _ in return true },
            removeItem: { _ in }
        )

        let keysMissingEnvironment = AppEnvironment(
            wrappedSDKSynchronizer: MockWrappedSDKSynchronizer(),
            databaseFiles: .live(databaseFiles: DatabaseFiles(fileManager: wfmMock)),
            mnemonicSeedPhraseProvider: .mock,
            scheduler: Self.testScheduler.eraseToAnyScheduler(),
            walletStorage: .throwing,
            wrappedDerivationTool: .live(),
            zcashSDKEnvironment: .mainnet
        )

        let walletState = AppReducer.walletInitializationState(keysMissingEnvironment)
        
        XCTAssertEqual(walletState, .keysMissing)
    }

    func testWalletInitializationState_FilesMissingKeysMissing() throws {
        let wfmMock = WrappedFileManager(
            url: { _, _, _, _ in URL(fileURLWithPath: "") },
            fileExists: { _ in return false },
            removeItem: { _ in }
        )

        let keysMissingEnvironment = AppEnvironment(
            wrappedSDKSynchronizer: MockWrappedSDKSynchronizer(),
            databaseFiles: .live(databaseFiles: DatabaseFiles(fileManager: wfmMock)),
            mnemonicSeedPhraseProvider: .mock,
            scheduler: Self.testScheduler.eraseToAnyScheduler(),
            walletStorage: .throwing,
            wrappedDerivationTool: .live(),
            zcashSDKEnvironment: .testnet
        )

        let walletState = AppReducer.walletInitializationState(keysMissingEnvironment)
        
        XCTAssertEqual(walletState, .uninitialized)
    }

    // TODO: - Implement testWalletInitializationState_FilesMissing when WalletStorage mock is available, issue 231 (https://github.com/zcash/secant-ios-wallet/issues/231)

    // TODO: - Implement testWalletInitializationState_Initialized when WalletStorage mock is available, issue 231 (https://github.com/zcash/secant-ios-wallet/issues/231)

    func testRespondToWalletInitializationState_Uninitialized() throws {
        let store = TestStore(
            initialState: .placeholder,
            reducer: AppReducer.default,
            environment: testEnvironment
        )
        
        store.send(.respondToWalletInitializationState(.uninitialized))
        
        Self.testScheduler.advance(by: 3)

        store.receive(.updateRoute(.onboarding)) {
            $0.route = .onboarding
            $0.appInitializationState = .uninitialized
        }
    }

    func testRespondToWalletInitializationState_KeysMissing() throws {
        let store = TestStore(
            initialState: .placeholder,
            reducer: AppReducer.default,
            environment: testEnvironment
        )
        
        store.send(.respondToWalletInitializationState(.keysMissing)) { state in
            state.appInitializationState = .keysMissing
        }
    }

    func testRespondToWalletInitializationState_FilesMissing() throws {
        let store = TestStore(
            initialState: .placeholder,
            reducer: AppReducer.default,
            environment: testEnvironment
        )
        
        store.send(.respondToWalletInitializationState(.filesMissing)) { state in
            state.appInitializationState = .filesMissing
        }
        
        store.receive(.initializeSDK) { state in
            // failed is expected because environment is throwing errors
            state.appInitializationState = .failed
        }

        store.receive(.checkBackupPhraseValidation) { state in
            // failed is expected because environment is throwing errors
            state.appInitializationState = .failed
        }
    }
    
    func testRespondToWalletInitializationState_Initialized() throws {
        let store = TestStore(
            initialState: .placeholder,
            reducer: AppReducer.default,
            environment: testEnvironment
        )
        
        store.send(.respondToWalletInitializationState(.initialized))
        
        store.receive(.initializeSDK) { state in
            // failed is expected because environment is throwing errors
            state.appInitializationState = .failed
        }
        
        store.receive(.checkBackupPhraseValidation) { state in
            // failed is expected because environment is throwing errors
            state.appInitializationState = .failed
        }
    }
}