Mock handler

andrewhickman · andrewhickman · commit 417522ed4344 · 2020-07-01T21:49:56.000+01:00
diff --git a/src/config.rs b/src/config.rs
@@ -52,6 +52,7 @@ pub enum RouteKind {
     File(FileRoute),
     Proxy(ProxyRoute),
     Json(JsonRoute),
+    Mock(MockRoute),
 }
 
 #[derive(Debug, Deserialize)]
@@ -71,6 +72,13 @@ pub struct JsonRoute {
     pub pretty: bool,
 }
 
+#[derive(Debug, Deserialize)]
+pub struct MockRoute {
+    #[serde(with = "http_serde::status_code")]
+    pub status: http::StatusCode,
+    pub body: Option<serde_json::Value>,
+}
+
 #[derive(Debug, Deserialize)]
 #[serde(rename_all = "kebab-case")]
 pub struct ProxyRoute {
@@ -95,6 +103,7 @@ impl Route {
             RouteKind::File(file) => file.validate(),
             RouteKind::Proxy(proxy) => proxy.validate(),
             RouteKind::Json(json) => json.validate(),
+            RouteKind::Mock(_) => Ok(()),
         }
     }
 }
diff --git a/src/handler/mock.rs b/src/handler/mock.rs
@@ -0,0 +1,28 @@
+use hyper::Body;
+
+use crate::{config, response};
+
+#[derive(Debug)]
+pub struct MockHandler {
+    config: config::MockRoute,
+}
+
+impl MockHandler {
+    pub fn new(config: config::MockRoute) -> Self {
+        MockHandler { config }
+    }
+
+    pub async fn handle(
+        &self,
+        _: http::Request<Body>,
+    ) -> Result<http::Response<Body>, (http::Request<Body>, http::Response<Body>)> {
+        match &self.config.body {
+            Some(value) => {
+                let mut response = response::json(value);
+                *response.status_mut() = self.config.status;
+                Ok(response)
+            }
+            None => Ok(response::from_status(self.config.status)),
+        }
+    }
+}
diff --git a/src/handler/mod.rs b/src/handler/mod.rs
@@ -1,5 +1,6 @@
 mod fs;
 mod json;
+mod mock;
 mod proxy;
 
 use anyhow::Result;
@@ -8,6 +9,7 @@ use regex::Regex;
 
 use self::fs::{DirHandler, FileHandler};
 use self::json::JsonHandler;
+use self::mock::MockHandler;
 use self::proxy::ProxyHandler;
 use crate::config;
 
@@ -24,6 +26,7 @@ pub enum HandlerKind {
     Dir(DirHandler),
     Proxy(ProxyHandler),
     Json(JsonHandler),
+    Mock(MockHandler),
 }
 
 #[derive(Debug)]
@@ -50,6 +53,7 @@ impl Handler {
             config::RouteKind::Dir(dir) => HandlerKind::Dir(DirHandler::new(dir)),
             config::RouteKind::Proxy(proxy) => HandlerKind::Proxy(ProxyHandler::new(proxy)),
             config::RouteKind::Json(json) => HandlerKind::Json(JsonHandler::new(json).await?),
+            config::RouteKind::Mock(mock) => HandlerKind::Mock(MockHandler::new(mock)),
         };
 
         Ok(Handler {
@@ -73,6 +77,7 @@ impl Handler {
             HandlerKind::Dir(dir) => dir.handle(request, &path).await,
             HandlerKind::Proxy(proxy) => proxy.handle(request, &path).await,
             HandlerKind::Json(json) => json.handle(request, &path).await,
+            HandlerKind::Mock(mock) => mock.handle(request).await,
         };
 
         if let Ok(response) = &mut result {

Original file line number	Diff line number	Diff line change
`@@ -52,6 +52,7 @@ pub enum RouteKind {`
`52`	`52`	`File(FileRoute),`
`53`	`53`	`Proxy(ProxyRoute),`
`54`	`54`	`Json(JsonRoute),`
	`55`	`+ Mock(MockRoute),`
`55`	`56`	`}`
`56`	`57`
`57`	`58`	`#[derive(Debug, Deserialize)]`
`@@ -71,6 +72,13 @@ pub struct JsonRoute {`
`71`	`72`	`pub pretty: bool,`
`72`	`73`	`}`
`73`	`74`
	`75`	`+#[derive(Debug, Deserialize)]`
	`76`	`+pub struct MockRoute {`
	`77`	`+ #[serde(with = "http_serde::status_code")]`
	`78`	`+ pub status: http::StatusCode,`
	`79`	`+ pub body: Option<serde_json::Value>,`
	`80`	`+}`
	`81`	`+`
`74`	`82`	`#[derive(Debug, Deserialize)]`
`75`	`83`	`#[serde(rename_all = "kebab-case")]`
`76`	`84`	`pub struct ProxyRoute {`
`@@ -95,6 +103,7 @@ impl Route {`
`95`	`103`	`RouteKind::File(file) => file.validate(),`
`96`	`104`	`RouteKind::Proxy(proxy) => proxy.validate(),`
`97`	`105`	`RouteKind::Json(json) => json.validate(),`
	`106`	`+ RouteKind::Mock(_) => Ok(()),`
`98`	`107`	`}`
`99`	`108`	`}`
`100`	`109`	`}`