 }
 
 extension Mock {
-    public func response(statusCode: Int, body: Body? = nil, header: [String: String]? = nil) -> MockResponseType {
+    public func response(statusCode: Int, body: Body? = nil, header: Header? = nil) -> MockResponseType {
         guard let url = request?.URL,
             let res = NSHTTPURLResponse(URL: url, statusCode: statusCode, HTTPVersion: nil, headerFields: header) else {
                 fatalError()

//

import Foundation

public class Mock: MockType, MockRequestType, MockResponseType {
    public typealias RequestHandler = Void -> NSURLRequest

}

public protocol MockRequestType: class {
    func response(statusCode: Int, body: Body?, header: Header?) -> MockResponseType
    func response(response: NSURLResponse, data: NSData?) -> MockResponseType
    func response(error: NSError) -> MockResponseType
    func response(handler: Mock.ResponseHandler) -> MockResponseType
