Authenticator/AuthenticatorTests/TokenScannerTests.swift

//
//  TokenScannerTests.swift
//  Authenticator
//
//  Copyright (c) 2017-2018 Authenticator authors
//
//  Permission is hereby granted, free of charge, to any person obtaining a copy
//  of this software and associated documentation files (the "Software"), to deal
//  in the Software without restriction, including without limitation the rights
//  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
//  copies of the Software, and to permit persons to whom the Software is
//  furnished to do so, subject to the following conditions:
//
//  The above copyright notice and this permission notice shall be included in all
//  copies or substantial portions of the Software.
//
//  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
//  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
//  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
//  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
//  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
//  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
//  SOFTWARE.
//

import XCTest
@testable import Authenticator
import OneTimePassword

class TokenScannerTests: XCTestCase {
    func testCancel() {
        var tokenScanner = TokenScanner()
        XCTAssertTrue(tokenScanner.viewModel.isScanning)

        let action = TokenScanner.Action.cancel
        let effect = tokenScanner.update(with: action)
        guard let requiredEffect = effect,
            case .cancel = requiredEffect else {
                XCTFail("Expected effect .Cancel, got \(String(describing: effect))")
                return
        }

        XCTAssertTrue(tokenScanner.viewModel.isScanning)
    }

    func testCompose() {
        var tokenScanner = TokenScanner()
        XCTAssertTrue(tokenScanner.viewModel.isScanning)

        let action = TokenScanner.Action.beginManualTokenEntry
        let effect = tokenScanner.update(with: action)
        guard let requiredEffect = effect,
            case .beginManualTokenEntry = requiredEffect else {
                XCTFail("Expected effect .BeginManualTokenEntry, got \(String(describing: effect))")
                return
        }

        XCTAssertTrue(tokenScanner.viewModel.isScanning)
    }

    func testShowApplicationSettings() {
        var tokenScanner = TokenScanner()

        let action = TokenScanner.Action.showApplicationSettings
        let effect = tokenScanner.update(with: action)
        guard let requiredEffect = effect,
            case .showApplicationSettings = requiredEffect else {
                XCTFail("Expected effect .showApplicationSettings, got \(String(describing: effect))")
                return
        }
    }

    func testScannerDecodedBadText() {
        var tokenScanner = TokenScanner()
        XCTAssertTrue(tokenScanner.viewModel.isScanning)

        let action = TokenScanner.Action.scannerDecodedText("something...")
        let effect = tokenScanner.update(with: action)
        guard let requiredEffect = effect,
            case .showErrorMessage(let message) = requiredEffect else {
                XCTFail("Expected effect .ShowErrorMessage, got \(String(describing: effect))")
                return
        }
        XCTAssertEqual(message, "Invalid Token")

        XCTAssertTrue(tokenScanner.viewModel.isScanning)
    }

    func testScannerDecodedBadURL() {
        var tokenScanner = TokenScanner()
        XCTAssertTrue(tokenScanner.viewModel.isScanning)

        let urlString = "http://example.com"
        let action = TokenScanner.Action.scannerDecodedText(urlString)
        let effect = tokenScanner.update(with: action)
        guard let requiredEffect = effect,
            case .showErrorMessage(let message) = requiredEffect else {
                XCTFail("Expected effect .ShowErrorMessage, got \(String(describing: effect))")
                return
        }
        XCTAssertEqual(message, "Invalid Token")

        XCTAssertTrue(tokenScanner.viewModel.isScanning)
    }

    func testScannerDecodedGoodURL() throws {
        var tokenScanner = TokenScanner()
        XCTAssertTrue(tokenScanner.viewModel.isScanning)

        let urlString = "otpauth://totp/Authenticator?secret=ABCDEFGHIJKLMNOP"
        let action = TokenScanner.Action.scannerDecodedText(urlString)
        let effect = tokenScanner.update(with: action)
        guard let requiredEffect = effect,
            case .saveNewToken(let token) = requiredEffect else {
                XCTFail("Expected effect .SaveNewToken, got \(String(describing: effect))")
                return
        }
        // swiftlint:disable:next force_unwrapping
        let expectedToken = try Token(url: URL(string: urlString)!)
        XCTAssertEqual(token, expectedToken)

        // The scanner should stop after the first successful token capture.
        XCTAssertFalse(tokenScanner.viewModel.isScanning)
    }

    struct ScannerError: Error {}

    func testScannerError() {
        var tokenScanner = TokenScanner()
        XCTAssertTrue(tokenScanner.viewModel.isScanning)

        let action = TokenScanner.Action.scannerError(ScannerError())
        let effect = tokenScanner.update(with: action)
        guard let requiredEffect = effect,
            case .showErrorMessage(let message) = requiredEffect else {
                XCTFail("Expected effect .ShowErrorMessage, got \(String(describing: effect))")
                return
        }
        XCTAssertEqual(message, "Capture Failed")

        XCTAssertTrue(tokenScanner.viewModel.isScanning)
    }
}