replace calls to deprecated function
changes PasswordStore.encrypt behavior when .gpg-id support is off
(default):
old:
* ignores passed in keyID
* encrypt with first public key in keychain (gopenPGP), or entire
keychain (ObjectivePGP)
new:
* honor passed in keyID
* encrypt with all keys in keychain
This commit is contained in:
parent
09b0b150ce
commit
e69e590e36
7 changed files with 11 additions and 110 deletions
|
|
@ -526,28 +526,13 @@ final class PGPAgentLowLevelTests: XCTestCase {
|
|||
XCTAssertEqual(passphraseRequests, [shortID])
|
||||
}
|
||||
|
||||
/// Encrypt with short ID: when containsPublicKey matches (via suffix), the short ID is passed to interface.
|
||||
func testEncryptWithKeyID_shortIDRecognized_shortIDFlowsThrough() throws {
|
||||
let shortID = "a1024dae"
|
||||
let longFingerprint = "4712286271220db299883ea7062e678da1024dae"
|
||||
mockPGP.publicKeyIDs = [longFingerprint]
|
||||
mockPGP.keyIDs = [longFingerprint]
|
||||
|
||||
_ = try agent.encrypt(plainData: testDecryptedData, keyID: shortID)
|
||||
|
||||
XCTAssertEqual(mockPGP.containsPublicKeyCalls, [shortID])
|
||||
XCTAssertEqual(mockPGP.encryptMultiKeyCalls.count, 1)
|
||||
XCTAssertEqual(mockPGP.encryptMultiKeyCalls[0].keyIDs, [shortID])
|
||||
}
|
||||
|
||||
// MARK: - Encrypt passthrough tests (for completeness of mock interaction)
|
||||
|
||||
/// encrypt(plainData:keyID:) calls containsPublicKey and passes data through via encrypt(plainData:keyIDs:).
|
||||
func testEncryptWithKeyID_keyFound_callsInterface() throws {
|
||||
func testEncryptWithKeyIDs_passesThrough() throws {
|
||||
let longFingerprint = "4712286271220db299883ea7062e678da1024dae"
|
||||
mockPGP.publicKeyIDs = [longFingerprint]
|
||||
|
||||
let result = try agent.encrypt(plainData: testDecryptedData, keyID: longFingerprint)
|
||||
let result = try agent.encrypt(plainData: testDecryptedData, keyIDs: [longFingerprint])
|
||||
|
||||
XCTAssertEqual(result, mockPGP.encryptResult)
|
||||
XCTAssertEqual(mockPGP.encryptMultiKeyCalls.count, 1)
|
||||
|
|
@ -555,48 +540,14 @@ final class PGPAgentLowLevelTests: XCTestCase {
|
|||
XCTAssertEqual(mockPGP.encryptMultiKeyCalls[0].plainData, testDecryptedData)
|
||||
}
|
||||
|
||||
/// encrypt with unknown key and single available key falls back.
|
||||
func testEncryptWithKeyID_keyNotFound_singleKey_fallsBack() throws {
|
||||
let shortID = "e9444483"
|
||||
let longFingerprint = "4712286271220db299883ea7062e678da1024dae"
|
||||
mockPGP.publicKeyIDs = []
|
||||
mockPGP.keyIDs = [longFingerprint]
|
||||
|
||||
let result = try agent.encrypt(plainData: testDecryptedData, keyID: shortID)
|
||||
|
||||
XCTAssertEqual(result, mockPGP.encryptResult)
|
||||
XCTAssertEqual(mockPGP.containsPublicKeyCalls, [shortID])
|
||||
XCTAssertEqual(mockPGP.encryptMultiKeyCalls[0].keyIDs, [longFingerprint])
|
||||
}
|
||||
|
||||
/// encrypt with unknown key and multiple keys throws.
|
||||
func testEncryptWithKeyID_keyNotFound_multipleKeys_throws() {
|
||||
mockPGP.publicKeyIDs = []
|
||||
mockPGP.keyIDs = ["4712286271220db299883ea7062e678da1024dae", "787eae1a5fa3e749aa34cc6aa0645ebed862027e"]
|
||||
|
||||
XCTAssertThrowsError(try agent.encrypt(plainData: testDecryptedData, keyID: "a1024dae")) { error in
|
||||
XCTAssertEqual(error as? AppError, AppError.pgpPublicKeyNotFound(keyID: "a1024dae"))
|
||||
}
|
||||
XCTAssertEqual(mockPGP.encryptMultiKeyCalls.count, 0)
|
||||
}
|
||||
|
||||
/// encrypt(plainData:) without keyID passes nil to the deprecated interface method.
|
||||
func testEncryptNoKeyID_passesNilToInterface() throws {
|
||||
let result = try agent.encrypt(plainData: testDecryptedData)
|
||||
|
||||
XCTAssertEqual(result, mockPGP.encryptResult)
|
||||
XCTAssertEqual(mockPGP.encryptCalls.count, 1)
|
||||
XCTAssertNil(mockPGP.encryptCalls[0].keyID)
|
||||
}
|
||||
|
||||
/// encrypt propagates errors from interface.
|
||||
func testEncrypt_interfaceThrows_propagatesError() {
|
||||
func testEncryptWithKeyIDs_interfaceThrows_propagatesError() {
|
||||
let shortID = "a1024dae"
|
||||
let longFingerprint = "4712286271220db299883ea7062e678da1024dae"
|
||||
mockPGP.publicKeyIDs = [longFingerprint]
|
||||
mockPGP.encryptError = AppError.encryption
|
||||
|
||||
XCTAssertThrowsError(try agent.encrypt(plainData: testDecryptedData, keyID: shortID)) { error in
|
||||
XCTAssertThrowsError(try agent.encrypt(plainData: testDecryptedData, keyIDs: [shortID])) { error in
|
||||
XCTAssertEqual(error as? AppError, AppError.encryption)
|
||||
}
|
||||
}
|
||||
|
|
@ -614,7 +565,6 @@ final class PGPAgentLowLevelTests: XCTestCase {
|
|||
XCTAssertEqual(mockPGP.encryptWithAllKeysCalls[0].plainData, testDecryptedData)
|
||||
// Does not call containsPublicKey or the single/multi-key encrypt methods.
|
||||
XCTAssertEqual(mockPGP.containsPublicKeyCalls.count, 0)
|
||||
XCTAssertEqual(mockPGP.encryptCalls.count, 0)
|
||||
XCTAssertEqual(mockPGP.encryptMultiKeyCalls.count, 0)
|
||||
}
|
||||
|
||||
|
|
@ -627,7 +577,7 @@ final class PGPAgentLowLevelTests: XCTestCase {
|
|||
}
|
||||
}
|
||||
|
||||
/// encryptWithAllKeys throws encryption error when pgpInterface is nil (checkAndInit fails).
|
||||
/// encryptWithAllKeys throws keyImport when checkAndInit triggers initKeys without PGP keys.
|
||||
func testEncryptWithAllKeys_checkAndInit_requiresPGPKeyPassphraseInKeystore() throws {
|
||||
keychain.removeContent(for: Globals.pgpKeyPassphrase)
|
||||
|
||||
|
|
|
|||
|
|
@ -47,7 +47,6 @@ class MockPGPInterface: PGPInterface {
|
|||
|
||||
var decryptCalls: [DecryptCall] = []
|
||||
var resolvedPassphrases: [String] = []
|
||||
var encryptCalls: [EncryptCall] = []
|
||||
var encryptMultiKeyCalls: [EncryptMultiKeyCall] = []
|
||||
var encryptWithAllKeysCalls: [EncryptWithAllKeysCall] = []
|
||||
var containsPublicKeyCalls: [String] = []
|
||||
|
|
@ -66,14 +65,6 @@ class MockPGPInterface: PGPInterface {
|
|||
return decryptResult
|
||||
}
|
||||
|
||||
func encrypt(plainData: Data, keyID: String?) throws -> Data {
|
||||
encryptCalls.append(EncryptCall(plainData: plainData, keyID: keyID))
|
||||
if let error = encryptError {
|
||||
throw error
|
||||
}
|
||||
return encryptResult
|
||||
}
|
||||
|
||||
func encryptWithAllKeys(plainData: Data) throws -> Data {
|
||||
encryptWithAllKeysCalls.append(EncryptWithAllKeysCall(plainData: plainData))
|
||||
if let error = encryptError {
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue