58 lines
2 KiB
Swift
58 lines
2 KiB
Swift
//
|
|
// QRKeyScannerTest.swift
|
|
// passTests
|
|
//
|
|
// Created by Danny Moesch on 21.08.20.
|
|
// Copyright © 2020 Bob Sun. All rights reserved.
|
|
//
|
|
|
|
import XCTest
|
|
|
|
@testable import Pass
|
|
|
|
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 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)
|
|
}
|
|
}
|