Add new methods to When and Then (#27)

Signed-off-by: declark1 <[email protected]>

Author: declark1

book/src/concepts/mock-builder/then.md

@@ -23,5 +23,13 @@
 - `error()`
 - `ok()`
 - `bad_request()`
-- `internal_server_error()`
+- `unauthorized()`
+- `forbidden()`
 - `not_found()`
+- `unsupported_media_type()`
+- `unprocessable_content()`
+- `internal_server_error()`
+- `not_implemented()`
+- `bad_gateway()`
+- `service_unavailable()`
+- `gateway_timeout()`

mocktail/src/mock_builder/then.rs

@@ -163,31 +163,103 @@ impl Then {
         self
     }
 
+    /// 200 Ok
     pub fn ok(self) -> Self {
         self.update(|r| {
             r.status = StatusCode::OK;
         });
         self
     }
 
+    // Common client errors
+
+    /// 400 Bad Request
+    pub fn bad_request(self) -> Self {
+        self.update(|r| {
+            r.status = StatusCode::BAD_REQUEST;
+        });
+        self
+    }
+
+    /// 401 Unauthorized
+    pub fn unauthorized(self) -> Self {
+        self.update(|r| {
+            r.status = StatusCode::UNAUTHORIZED;
+        });
+        self
+    }
+
+    /// 403 Forbidden
+    pub fn forbidden(self) -> Self {
+        self.update(|r| {
+            r.status = StatusCode::FORBIDDEN;
+        });
+        self
+    }
+
+    /// 404 Not Found
     pub fn not_found(self) -> Self {
         self.update(|r| {
             r.status = StatusCode::NOT_FOUND;
         });
         self
     }
 
-    pub fn bad_request(self) -> Self {
+    /// 415 Unsupported Media Type
+    pub fn unsupported_media_type(self) -> Self {
         self.update(|r| {
-            r.status = StatusCode::BAD_REQUEST;
+            r.status = StatusCode::UNSUPPORTED_MEDIA_TYPE;
+        });
+        self
+    }
+
+    /// 422 Unprocessable Content
+    pub fn unprocessable_content(self) -> Self {
+        self.update(|r| {
+            r.status = StatusCode::UNPROCESSABLE_ENTITY;
         });
         self
     }
 
+    // Common server errors
+
+    /// 500 Internal Server Error
     pub fn internal_server_error(self) -> Self {
         self.update(|r| {
             r.status = StatusCode::INTERNAL_SERVER_ERROR;
         });
         self
     }
+
+    /// 501 Not Implemented
+    pub fn not_implemented(self) -> Self {
+        self.update(|r| {
+            r.status = StatusCode::NOT_IMPLEMENTED;
+        });
+        self
+    }
+
+    /// 502 Bad Gateway
+    pub fn bad_gateway(self) -> Self {
+        self.update(|r| {
+            r.status = StatusCode::BAD_GATEWAY;
+        });
+        self
+    }
+
+    /// 503 Service Unavailable
+    pub fn service_unavailable(self) -> Self {
+        self.update(|r| {
+            r.status = StatusCode::SERVICE_UNAVAILABLE;
+        });
+        self
+    }
+
+    /// 504 Gateway Timeout
+    pub fn gateway_timeout(self) -> Self {
+        self.update(|r| {
+            r.status = StatusCode::GATEWAY_TIMEOUT;
+        });
+        self
+    }
 }

mocktail/src/mock_builder/when.rs

@@ -213,4 +213,10 @@ impl When {
         self.push(matchers::method(Method::DELETE));
         self
     }
+
+    /// HTTP HEAD method.
+    pub fn head(self) -> Self {
+        self.push(matchers::method(Method::HEAD));
+        self
+    }
 }