Add unit tests for "key: value" syntax in Password class
This commit is contained in:
Danny Moesch
Bob Sun
parent
b7a0cbaef6
commit
213234f57e
2 changed files with 275 additions and 0 deletions
271
passKitTests/PasswordTests.swift
Normal file
271
passKitTests/PasswordTests.swift
Normal file
|
|
@ -0,0 +1,271 @@
|
|||
//
|
||||
// PasswordTests.swift
|
||||
// passKitTests
|
||||
//
|
||||
// Created by Danny Mösch on 02.05.18.
|
||||
// Copyright © 2018 Bob Sun. All rights reserved.
|
||||
//
|
||||
|
||||
import XCTest
|
||||
@testable import passKit
|
||||
|
||||
class PasswordTest: XCTestCase {
|
||||
static let EMPTY_STRING = ""
|
||||
static let PASSWORD_NAME = "password"
|
||||
static let PASSWORD_PATH = "/path/to/\(PASSWORD_NAME)"
|
||||
static let PASSWORD_URL = URL(fileURLWithPath: PASSWORD_PATH)
|
||||
static let PASSWORD_STRING = "abcd1234"
|
||||
static let OTP_TOKEN = "otpauth://totp/email@email.com?secret=abcd1234"
|
||||
|
||||
static let SECURE_URL_FIELD = AdditionField(title: "url", content: "https://secure.com")
|
||||
static let INSECURE_URL_FIELD = AdditionField(title: "url", content: "http://insecure.com")
|
||||
static let LOGIN_FIELD = AdditionField(title: "login", content: "login name")
|
||||
static let USERNAME_FIELD = AdditionField(title: "username", content: "some username")
|
||||
static let NOTE_FIELD = AdditionField(title: "note", content: "A NOTE")
|
||||
|
||||
func testUrl() {
|
||||
let password1 = getPasswordObjectWith(content: PasswordTest.EMPTY_STRING)
|
||||
XCTAssertEqual(password1.namePath, PasswordTest.PASSWORD_PATH)
|
||||
|
||||
let password2 = getPasswordObjectWith(content: PasswordTest.EMPTY_STRING, url: nil)
|
||||
XCTAssertEqual(password2.namePath, PasswordTest.EMPTY_STRING)
|
||||
}
|
||||
|
||||
func testLooksLikeOTP() {
|
||||
XCTAssertTrue(Password.LooksLikeOTP(line: PasswordTest.OTP_TOKEN))
|
||||
XCTAssertFalse(Password.LooksLikeOTP(line: "no_auth://totp/blabla"))
|
||||
}
|
||||
|
||||
func testEmptyFile() {
|
||||
let fileContent = PasswordTest.EMPTY_STRING
|
||||
let password = getPasswordObjectWith(content: fileContent)
|
||||
|
||||
XCTAssertEqual(password.password, PasswordTest.EMPTY_STRING)
|
||||
XCTAssertEqual(password.getPlainData(), fileContent.data(using: .utf8))
|
||||
|
||||
XCTAssertEqual(password.getAdditionsPlainText(), PasswordTest.EMPTY_STRING)
|
||||
XCTAssertTrue(password.getFilteredAdditions().isEmpty)
|
||||
|
||||
XCTAssertNil(password.getUsername())
|
||||
XCTAssertNil(password.getURLString())
|
||||
XCTAssertNil(password.getLogin())
|
||||
}
|
||||
|
||||
func testOneEmptyLine() {
|
||||
let fileContent = """
|
||||
|
||||
"""
|
||||
let password = getPasswordObjectWith(content: fileContent)
|
||||
|
||||
XCTAssertEqual(password.password, PasswordTest.EMPTY_STRING)
|
||||
XCTAssertEqual(password.getPlainData(), fileContent.data(using: .utf8))
|
||||
|
||||
XCTAssertEqual(password.getAdditionsPlainText(), PasswordTest.EMPTY_STRING)
|
||||
XCTAssertTrue(password.getFilteredAdditions().isEmpty)
|
||||
|
||||
XCTAssertNil(password.getUsername())
|
||||
XCTAssertNil(password.getURLString())
|
||||
XCTAssertNil(password.getLogin())
|
||||
}
|
||||
|
||||
func testSimplePasswordFile() {
|
||||
let passwordString = PasswordTest.PASSWORD_STRING
|
||||
let urlField = PasswordTest.SECURE_URL_FIELD
|
||||
let loginField = PasswordTest.LOGIN_FIELD
|
||||
let usernameField = PasswordTest.USERNAME_FIELD
|
||||
let noteField = PasswordTest.NOTE_FIELD
|
||||
let fileContent = """
|
||||
\(passwordString)
|
||||
\(urlField.asString)
|
||||
\(loginField.asString)
|
||||
\(usernameField.asString)
|
||||
\(noteField.asString)
|
||||
"""
|
||||
let password = getPasswordObjectWith(content: fileContent)
|
||||
|
||||
XCTAssertEqual(password.password, passwordString)
|
||||
XCTAssertEqual(password.getPlainData(), fileContent.data(using: .utf8))
|
||||
|
||||
XCTAssertEqual(password.getAdditionsPlainText(), asPlainText(urlField, loginField, usernameField, noteField))
|
||||
XCTAssertTrue(does(password: password, contain: urlField))
|
||||
XCTAssertFalse(does(password: password, contain: loginField))
|
||||
XCTAssertFalse(does(password: password, contain: usernameField))
|
||||
XCTAssertTrue(does(password: password, contain: noteField))
|
||||
|
||||
XCTAssertEqual(password.getURLString(), urlField.content)
|
||||
XCTAssertEqual(password.getLogin(), loginField.content)
|
||||
XCTAssertEqual(password.getUsername(), usernameField.content)
|
||||
}
|
||||
|
||||
func testTwoPasswords() {
|
||||
let firstPasswordString = PasswordTest.PASSWORD_STRING
|
||||
let secondPasswordString = "efgh5678"
|
||||
let urlField = PasswordTest.INSECURE_URL_FIELD
|
||||
let fileContent = """
|
||||
\(firstPasswordString)
|
||||
\(secondPasswordString)
|
||||
\(urlField.asString)
|
||||
"""
|
||||
let password = getPasswordObjectWith(content: fileContent)
|
||||
|
||||
XCTAssertEqual(password.password, firstPasswordString)
|
||||
XCTAssertEqual(password.getPlainData(), fileContent.data(using: .utf8))
|
||||
XCTAssertEqual(password.getAdditionsPlainText(), asPlainText(secondPasswordString, urlField.asString))
|
||||
|
||||
XCTAssertTrue(does(password: password, contain: urlField))
|
||||
XCTAssertTrue(does(password: password, contain: AdditionField(title: "unknown 1", content: secondPasswordString)))
|
||||
|
||||
XCTAssertNil(password.getUsername())
|
||||
XCTAssertEqual(password.getURLString(), urlField.content)
|
||||
XCTAssertNil(password.getLogin())
|
||||
}
|
||||
|
||||
func testNoPassword() {
|
||||
let urlField = PasswordTest.SECURE_URL_FIELD
|
||||
let noteField = PasswordTest.NOTE_FIELD
|
||||
let fileContent = """
|
||||
\(urlField.asString)
|
||||
\(noteField.asString)
|
||||
"""
|
||||
let password = getPasswordObjectWith(content: fileContent)
|
||||
|
||||
XCTAssertEqual(password.password, urlField.asString)
|
||||
XCTAssertEqual(password.getPlainData(), fileContent.data(using: .utf8))
|
||||
|
||||
XCTAssertEqual(password.getAdditionsPlainText(), asPlainText(noteField))
|
||||
XCTAssertTrue(does(password: password, contain: noteField))
|
||||
|
||||
XCTAssertNil(password.getUsername())
|
||||
XCTAssertNil(password.getURLString())
|
||||
XCTAssertNil(password.getLogin())
|
||||
}
|
||||
|
||||
func testDuplicateKeys() {
|
||||
let passwordString = PasswordTest.PASSWORD_STRING
|
||||
let urlField1 = PasswordTest.SECURE_URL_FIELD
|
||||
let urlField2 = PasswordTest.INSECURE_URL_FIELD
|
||||
let fileContent = """
|
||||
\(passwordString)
|
||||
\(urlField1.asString)
|
||||
\(urlField2.asString)
|
||||
"""
|
||||
let password = getPasswordObjectWith(content: fileContent)
|
||||
|
||||
XCTAssertEqual(password.password, passwordString)
|
||||
XCTAssertEqual(password.getPlainData(), fileContent.data(using: .utf8))
|
||||
|
||||
XCTAssertEqual(password.getAdditionsPlainText(), asPlainText(urlField1, urlField2))
|
||||
XCTAssertTrue(does(password: password, contain: urlField1))
|
||||
XCTAssertTrue(does(password: password, contain: urlField2))
|
||||
|
||||
XCTAssertNil(password.getUsername())
|
||||
XCTAssertEqual(password.getURLString(), urlField1.content)
|
||||
XCTAssertNil(password.getLogin())
|
||||
}
|
||||
|
||||
func testUnknownKeys() {
|
||||
let passwordString = PasswordTest.PASSWORD_STRING
|
||||
let value1 = "value 1"
|
||||
let value2 = "value 2"
|
||||
let value3 = "value 3"
|
||||
let value4 = "value 4"
|
||||
let noteField = PasswordTest.NOTE_FIELD
|
||||
let urlField = PasswordTest.SECURE_URL_FIELD
|
||||
let fileContent = """
|
||||
\(passwordString)
|
||||
\(value1)
|
||||
\(noteField.asString)
|
||||
\(value2)
|
||||
\(value3)
|
||||
\(urlField.asString)
|
||||
\(value4)
|
||||
"""
|
||||
let password = getPasswordObjectWith(content: fileContent)
|
||||
|
||||
XCTAssertEqual(password.password, passwordString)
|
||||
XCTAssertEqual(password.getPlainData(), fileContent.data(using: .utf8))
|
||||
|
||||
XCTAssertEqual(password.getAdditionsPlainText(), asPlainText(value1, noteField.asString, value2, value3, urlField.asString, value4))
|
||||
XCTAssertTrue(does(password: password, contain: AdditionField(title: "unknown 1", content: value1)))
|
||||
XCTAssertTrue(does(password: password, contain: noteField))
|
||||
XCTAssertTrue(does(password: password, contain: AdditionField(title: "unknown 2", content: value2)))
|
||||
XCTAssertTrue(does(password: password, contain: AdditionField(title: "unknown 3", content: value3)))
|
||||
XCTAssertTrue(does(password: password, contain: urlField))
|
||||
XCTAssertTrue(does(password: password, contain: AdditionField(title: "unknown 4", content: value4)))
|
||||
|
||||
XCTAssertNil(password.getUsername())
|
||||
XCTAssertEqual(password.getURLString(), urlField.content)
|
||||
XCTAssertNil(password.getLogin())
|
||||
}
|
||||
|
||||
func testPasswordFileWithOtpToken() {
|
||||
let passwordString = PasswordTest.PASSWORD_STRING
|
||||
let noteField = PasswordTest.NOTE_FIELD
|
||||
let otpToken = PasswordTest.OTP_TOKEN
|
||||
let fileContent = """
|
||||
\(passwordString)
|
||||
\(noteField.asString)
|
||||
\(otpToken)
|
||||
"""
|
||||
let password = getPasswordObjectWith(content: fileContent)
|
||||
|
||||
XCTAssertEqual(password.password, passwordString)
|
||||
XCTAssertEqual(password.getPlainData(), fileContent.data(using: .utf8))
|
||||
|
||||
XCTAssertEqual(password.getAdditionsPlainText(), asPlainText(noteField.asString, otpToken))
|
||||
|
||||
XCTAssertEqual(password.otpType, Password.OtpType.totp)
|
||||
XCTAssertNotNil(password.getOtp())
|
||||
}
|
||||
|
||||
func testFirstLineIsOtpToken() {
|
||||
let otpToken = PasswordTest.OTP_TOKEN
|
||||
let fileContent = """
|
||||
\(otpToken)
|
||||
"""
|
||||
let password = getPasswordObjectWith(content: fileContent)
|
||||
|
||||
XCTAssertEqual(password.password, otpToken)
|
||||
XCTAssertEqual(password.getPlainData(), fileContent.data(using: .utf8))
|
||||
|
||||
XCTAssertEqual(password.getAdditionsPlainText(), PasswordTest.EMPTY_STRING)
|
||||
|
||||
XCTAssertNil(password.getUsername())
|
||||
XCTAssertNil(password.getURLString())
|
||||
XCTAssertNil(password.getLogin())
|
||||
|
||||
XCTAssertEqual(password.otpType, Password.OtpType.totp)
|
||||
XCTAssertNotNil(password.getOtp())
|
||||
}
|
||||
|
||||
func testWrongOtpToken() {
|
||||
let otpToken = "otpauth://htop/blabla"
|
||||
let fileContent = """
|
||||
\(otpToken)
|
||||
"""
|
||||
let password = getPasswordObjectWith(content: fileContent)
|
||||
|
||||
XCTAssertEqual(password.password, otpToken)
|
||||
XCTAssertEqual(password.getPlainData(), fileContent.data(using: .utf8))
|
||||
|
||||
XCTAssertEqual(password.getAdditionsPlainText(), PasswordTest.EMPTY_STRING)
|
||||
|
||||
XCTAssertEqual(password.otpType, Password.OtpType.none)
|
||||
XCTAssertNil(password.getOtp())
|
||||
}
|
||||
|
||||
private func getPasswordObjectWith(content: String, url: URL? = PasswordTest.PASSWORD_URL) -> Password {
|
||||
return Password(name: PasswordTest.PASSWORD_NAME, url: url, plainText: content)
|
||||
}
|
||||
|
||||
private func does(password: Password, contain field: AdditionField) -> Bool {
|
||||
return password.getFilteredAdditions().contains(field)
|
||||
}
|
||||
|
||||
private func asPlainText(_ strings: String...) -> String {
|
||||
return strings.joined(separator: "\n")
|
||||
}
|
||||
private func asPlainText(_ fields: AdditionField...) -> String {
|
||||
return fields.map { $0.asString }.joined(separator: "\n")
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue