Add support for extracting recorded requests

Author: pwalters

MockDuck/Sources/MockBundle.swift

@@ -55,32 +55,65 @@ final class MockBundle {
             os_log("Request %@ not found on disk. Expected file name: %@", log: MockDuck.log, type: .debug, "\(request)", fileName)
         }
 
-        if
-            let targetURL = targetURL,
-            let targetLoadingURL = targetLoadingURL
-        {
-            let decoder = JSONDecoder()
-
-            do {
-                let data = try Data(contentsOf: targetURL)
-
-                let loaded = try decoder.decode(MockRequestResponse.self, from: data)
-                requestResponse.responseWrapper = loaded.responseWrapper
+        if let loadedResponse = loadResponseFile(targetURL: targetURL) {
+            requestResponse.responseWrapper = loadedResponse.responseWrapper
+            requestResponse.responseData = loadedResponse.responseData
+        }
 
-                // Load the response data if the format is supported.
-                // This should be the same filename with a different extension.
-                if let dataFileName = requestResponse.fileName(for: .responseData) {
-                    let dataURL = targetLoadingURL.appendingPathComponent(dataFileName)
-                    requestResponse.responseData = try Data(contentsOf: dataURL)
+        return false
+    }
+    
+    func loadResponseFile(targetURL: URL?) -> MockRequestResponse? {
+        guard let targetURL = targetURL else { return nil }
+        let decoder = JSONDecoder()
+        
+        do {
+            let data = try Data(contentsOf: targetURL)
+            
+            var response = try decoder.decode(MockRequestResponse.self, from: data)
+            
+            // Load the response data if the format is supported.
+            // This should be the same fil  ename with a different extension.
+            if let dataFileName = response.fileName(for: .responseData) {
+                let dataURL = targetURL.deletingLastPathComponent().appendingPathComponent(dataFileName)
+                response.responseData = try? Data(contentsOf: dataURL)
+            }
+            
+            return response
+        } catch {
+            os_log("Error decoding JSON: %@", log: MockDuck.log, type: .error, "\(error)")
+        }
+        return nil
+    }
+    
+    func getResponses(for hostname: String) -> [MockRequestResponse] {
+        guard let baseURL = recordingURL else { return [] }
+        
+        var responses = [MockRequestResponse]()
+        let targetURL = baseURL.appendingPathComponent(hostname)
+        
+        let results = FileManager.default.enumerator(
+            at: targetURL,
+            includingPropertiesForKeys: [.isDirectoryKey],
+            options: [])
+        if let results = results {
+            for item in results {
+                var isDir = ObjCBool(false)
+                if
+                    let item = item as? URL,
+                    let attributes = try? FileManager.default.fileExists(
+                        atPath: item.path,
+                        isDirectory: &isDir),
+                    !isDir.boolValue,
+                    !item.lastPathComponent.contains("-response"),
+                    let response = loadResponseFile(targetURL: item)
+                {
+                    responses.append(response)
                 }
-
-                return true
-            } catch {
-                os_log("Error decoding JSON: %@", log: MockDuck.log, type: .error, "\(error)")
             }
         }
-
-        return false
+        
+        return responses
     }
 
     /// If recording is enabled, this method saves the request to the filesystem. If the request

MockDuck/Sources/MockDuck.swift

@@ -135,6 +135,13 @@ public final class MockDuck {
     public static func unregisterAllRequestHandlers() {
         mockBundle.unregisterAllRequestHandlers()
     }
+    
+    // MARK: - Fetching Response Objects
+    
+    public static func getResponses(for hostname: String) -> [MockRequestResponse] {
+        checkConfigureMockDuck()
+        return mockBundle.getResponses(for: hostname)
+    }
 
     // MARK: - Internal Use Only
 

MockDuck/Sources/MockRequestResponse.swift

@@ -9,7 +9,7 @@
 import Foundation
 
 /// A basic container for holding a request, a response, and any associated data.
-final class MockRequestResponse: Codable, CustomDebugStringConvertible {
+public final class MockRequestResponse: Codable, CustomDebugStringConvertible {
 
     enum MockFileTarget {
         case request
@@ -19,7 +19,7 @@ final class MockRequestResponse: Codable, CustomDebugStringConvertible {
 
     // MARK: - Properties
 
-    var request: URLRequest {
+    public var request: URLRequest {
         get {
             return requestWrapper.request
         }
@@ -28,11 +28,11 @@ final class MockRequestResponse: Codable, CustomDebugStringConvertible {
         }
     }
 
-    var response: URLResponse? {
+    public var response: URLResponse? {
         return responseWrapper?.response
     }
 
-    var responseData: Data? {
+    public var responseData: Data? {
         get {
             return responseWrapper?.responseData
         }
@@ -140,7 +140,7 @@ final class MockRequestResponse: Codable, CustomDebugStringConvertible {
         case responseWrapper = "response"
     }
 
-    init(from decoder: Decoder) throws {
+    public init(from decoder: Decoder) throws {
         let container = try decoder.container(keyedBy: CodingKeys.self)
         requestWrapper = try container.decode(MockRequest.self, forKey: .requestWrapper)
         responseWrapper = try container.decodeIfPresent(MockResponse.self, forKey: .responseWrapper)