Cookie support

Package.resolved

@@ -28,6 +28,15 @@
           "version": "4.2.4"
         }
       },
+      {
+        "package": "multipart-kit",
+        "repositoryURL": "https://github.com/vapor/multipart-kit.git",
+        "state": {
+          "branch": null,
+          "revision": "4fcc81d2ce9210a0debceb015a210c2d2eb1da71",
+          "version": "4.0.2"
+        }
+      },
       {
         "package": "routing-kit",
         "repositoryURL": "https://github.com/vapor/routing-kit.git",
@@ -123,8 +132,8 @@
         "repositoryURL": "https://github.com/vapor/vapor.git",
         "state": {
           "branch": null,
-          "revision": "5148f02e42149f73f6616452062cf96b64afbda6",
-          "version": "4.36.2"
+          "revision": "338c4be2362cac6a465bcda2ef73a189d06b18a6",
+          "version": "4.44.2"
         }
       },
       {

Sources/Spec/Spec.swift

@@ -16,8 +16,16 @@ public final class Spec {
     private var bearerToken: String? = nil
     private var headers: HTTPHeaders = [:]
     private var buffer: ByteBuffer? = nil
-    private var beforeRequest: (inout XCTHTTPRequest) throws -> () = { _ in }
+    private var beforeRequests: [(inout XCTHTTPRequest) throws -> ()] = []
     private var expectations: [(XCTHTTPResponse) throws -> ()] = []
+
+    private func beforeRequest() -> (inout XCTHTTPRequest) throws -> () {
+        { [unowned self] req in
+            for item in beforeRequests {
+                try item(&req)
+            }
+        }
+    }
 
     internal init(name: String, app: Application) {
         self.name = name
@@ -60,11 +68,19 @@ public final class Spec {
         return self
     }
 
+    ///set a content as the request body
+    public func cookie(_ cookie: HTTPCookies) -> Self {
+        self.beforeRequests.append({ req in
+            req.headers.cookie = cookie
+        })
+        return self
+    }
+
     ///set a content as the request body
     public func body<T: Content>(_ body: T) -> Self {
-        self.beforeRequest = { req in
+        self.beforeRequests.append({ req in
             try req.content.encode(body)
-        }
+        })
         return self
     }
 
@@ -127,7 +143,7 @@ public final class Spec {
                   body: self.buffer,
                   file: file,
                   line: line,
-                  beforeRequest: self.beforeRequest,
+                  beforeRequest: self.beforeRequest(),
                   afterResponse: afterRequest)
     }
 }