passforios/passTests/Models/QRKeyScannerTest.swift

//
//  QRKeyScannerTest.swift
//  passTests
//
//  Created by Danny Moesch on 21.08.20.
//  Copyright © 2020 Bob Sun. All rights reserved.
//

import XCTest

@testable import Pass

final class QRKeyScannerTest: XCTestCase {
    private let header = "-----BEGIN PGP PUBLIC KEY BLOCK-----"
    private let body = "key body"
    private let footer = "-----END PGP PUBLIC KEY BLOCK-----"
    private let footerWithNewline = "-----END PGP PUBLIC KEY BLOCK-----\n"
    private let privateHeader = "-----BEGIN PGP PRIVATE KEY BLOCK-----"

    private var scanner = QRKeyScanner(keyType: .pgpPublic)

    func testAddHeaderTwice() {
        XCTAssertEqual(scanner.add(segment: header), .scanned(1))
        XCTAssertEqual(scanner.add(segment: header), .scanned(1))
        XCTAssertEqual(scanner.scannedKey, header)
    }

    func testAddBodyTwice() {
        XCTAssertEqual(scanner.add(segment: header), .scanned(1))
        XCTAssertEqual(scanner.add(segment: body), .scanned(2))
        XCTAssertEqual(scanner.add(segment: body), .scanned(2))
        XCTAssertEqual(scanner.scannedKey, header + body)
    }

    func testAddCompleteBlock() {
        XCTAssertEqual(scanner.add(segment: header), .scanned(1))
        XCTAssertEqual(scanner.add(segment: footer), .completed)
        XCTAssertEqual(scanner.scannedKey, header + footer)
    }

    func testCounterKeyType() {
        XCTAssertEqual(scanner.add(segment: privateHeader), .wrongKeyType(.pgpPrivate))
        XCTAssertEqual(scanner.add(segment: privateHeader), .wrongKeyType(.pgpPrivate))
        XCTAssertTrue(scanner.scannedKey.isEmpty)
    }

    func testUnknownKeyType() {
        XCTAssertEqual(scanner.add(segment: body), .lookingForStart)
        XCTAssertEqual(scanner.add(segment: body), .lookingForStart)
        XCTAssertTrue(scanner.scannedKey.isEmpty)
    }

    func testFooterSplitIntoDifferentSegments() {
        XCTAssertEqual(scanner.add(segment: header), .scanned(1))
        XCTAssertEqual(scanner.add(segment: "-----END PGP PUBLIC KEY "), .scanned(2))
        XCTAssertEqual(scanner.add(segment: "BLOCK-----"), .completed)
        XCTAssertEqual(scanner.scannedKey, header + footer)
    }

    func testFooterWithNewlien() {
        XCTAssertEqual(scanner.add(segment: header), .scanned(1))
        XCTAssertEqual(scanner.add(segment: footerWithNewline), .completed)
        XCTAssertEqual(scanner.scannedKey, header + footerWithNewline)
    }
}