Merge pull request #216 from marcelofabri/ohhttpStubs_httpBody

Add OHHTTPStubs_HTTPBody to URLRequest in Swift 3.0
AliSoftware · Oct 23, 2016 · 9ce2a87 · 9ce2a87
2 parents a4c71a7 + 28e9d20
commit 9ce2a87
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/OHHTTPStubs/Sources/Swift/OHHTTPStubsSwift.swift b/OHHTTPStubs/Sources/Swift/OHHTTPStubsSwift.swift
@@ -28,6 +28,11 @@
 
 
 #if swift(>=3.0)
+    extension URLRequest {
+        public var ohhttpStubs_httpBody: Data? {
+            return (self as NSURLRequest).ohhttpStubs_HTTPBody()
+        }
+    }
 #else
 #if swift(>=2.2)
     extension OHHTTPStubs {

diff --git a/OHHTTPStubs/UnitTests/Test Suites/SwiftHelpersTests.swift b/OHHTTPStubs/UnitTests/Test Suites/SwiftHelpersTests.swift
@@ -382,4 +382,13 @@ class SwiftHelpersTests : XCTestCase {
       XCTAssert((!falseMatcher)(req) == true, "!falseMatcher should result in a trueMatcher")
     }
   }
+
+  #if swift(>=3.0)
+    func test_ohhttpStubs_httpBody {
+      var req = URLRequest(url: URL(string: "foo://bar")!)
+      req.httpBody = Data()
+
+      XCTAssert(req.ohhttpStubs_httpBody == req.httpBody)
+    }
+  #endif
 }