From 29e3af24ab81bef30e6cb2d1f226b8a1c7c5394f Mon Sep 17 00:00:00 2001 From: Max Goedjen Date: Sun, 22 Mar 2020 22:39:45 -0700 Subject: [PATCH] Tests --- SecretAgentKit/Agent.swift | 2 +- SecretAgentKitTests/AgentTests.swift | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/SecretAgentKit/Agent.swift b/SecretAgentKit/Agent.swift index 3d48fbd..f10f882 100644 --- a/SecretAgentKit/Agent.swift +++ b/SecretAgentKit/Agent.swift @@ -27,7 +27,7 @@ extension Agent { guard !data.isEmpty else { return } let requestTypeInt = data[4] guard let requestType = SSHAgent.RequestType(rawValue: requestTypeInt) else { - writer.write(SSHAgent.ResponseType.agentFailure.data) + writer.write(OpenSSHKeyWriter().lengthAndData(of: SSHAgent.ResponseType.agentFailure.data)) os_log(.debug, "Agent returned %@", SSHAgent.ResponseType.agentFailure.debugDescription) return } diff --git a/SecretAgentKitTests/AgentTests.swift b/SecretAgentKitTests/AgentTests.swift index c570018..cc5ecbd 100644 --- a/SecretAgentKitTests/AgentTests.swift +++ b/SecretAgentKitTests/AgentTests.swift @@ -41,6 +41,7 @@ class AgentTests: XCTestCase { agent.handle(reader: stubReader, writer: stubWriter) let reader = OpenSSHReader(data: stubWriter.data) // TODO: VERIFY + XCTAssertFalse(true) print(stubWriter.data.base64EncodedString()) } @@ -87,7 +88,7 @@ class AgentTests: XCTestCase { // MARK: Exception Handling func testSignatureException() { - let stubReader = StubFileHandleReader(availableData: Constants.Requests.requestIdentities) + let stubReader = StubFileHandleReader(availableData: Constants.Requests.requestSignature) let list = storeList(with: [Constants.Secrets.ecdsa256Secret, Constants.Secrets.ecdsa384Secret]) let store = list.stores.first?.base as! Stub.Store store.shouldThrow = true