Adding Form Body

MATATAT · MATATAT · commit f705eb63d4be · 2025-08-24T10:36:33.000-07:00
diff --git a/mocktail/Cargo.toml b/mocktail/Cargo.toml
@@ -28,6 +28,7 @@ prost = "0.14"
 rand = "0.9"
 serde = "1"
 serde_json = "1"
+serde_urlencoded = "0.7"
 thiserror = "2"
 tokio = "1"
 tokio-stream = "0"
diff --git a/mocktail/src/form.rs b/mocktail/src/form.rs
@@ -0,0 +1,41 @@
+use crate::body::Body;
+
+/// The body of a mock request or response as a form.
+#[derive(Debug, Default, Clone, PartialEq, PartialOrd)]
+pub struct FormBody {
+    fields: Vec<(String, String)>,
+}
+
+impl FormBody {
+    /// Creates an empty form.
+    pub fn empty() -> Self {
+        Self::default()
+    }
+
+    /// Adds a field to the form.
+    pub fn add_field(&mut self, name: impl Into<String>, value: impl Into<String>) {
+        self.fields.push((name.into(), value.into()));
+    }
+
+    /// Converts the form body to a URL-encoded string.
+    pub fn url_encoded(&self) -> String {
+        self.fields
+            .iter()
+            .map(|(k, v)| format!("{k}={v}"))
+            .collect::<Vec<_>>()
+            .join("&")
+    }
+}
+
+impl PartialEq<Body> for FormBody {
+    fn eq(&self, other: &Body) -> bool {
+        let mut body_copy = other.clone();
+        let other_form_body =
+            serde_urlencoded::from_bytes::<Vec<(String, String)>>(&body_copy.as_bytes());
+
+        match other_form_body {
+            Ok(other) => self.fields == other,
+            Err(_) => false,
+        }
+    }
+}
diff --git a/mocktail/src/lib.rs b/mocktail/src/lib.rs
@@ -30,6 +30,7 @@ pub mod prelude {
 }
 mod ext;
 mod service;
+mod form;
 
 /// Represents errors that can occur while serving mocks.
 #[derive(thiserror::Error, Debug)]
diff --git a/mocktail/src/matchers.rs b/mocktail/src/matchers.rs
@@ -2,7 +2,7 @@
 use std::{any::Any, borrow::Cow, cmp::Ordering};
 
 use super::{body::Body, headers::Headers, request::Request};
-use crate::request::Method;
+use crate::{form::FormBody, request::Method};
 
 /// A matcher.
 pub trait Matcher: std::fmt::Debug + Send + Sync + 'static + AsMatcherEq {
@@ -97,6 +97,23 @@ pub fn body(body: Body) -> BodyMatcher {
     BodyMatcher(body)
 }
 
+/// Form body matcher.
+#[derive(Debug, PartialEq, PartialOrd)]
+pub struct FormMatcher(FormBody);
+
+impl Matcher for FormMatcher {
+    fn name(&self) -> &str {
+        "form_body"
+    }
+    fn matches(&self, req: &Request) -> bool {
+        self.0 == req.body
+    }
+}
+
+pub fn form(body: FormBody) -> FormMatcher {
+    FormMatcher(body)
+}
+
 /// Headers matcher.
 #[derive(Debug, PartialEq, PartialOrd)]
 pub struct HeadersMatcher(Headers);
diff --git a/mocktail/src/mock_builder/when.rs b/mocktail/src/mock_builder/when.rs
@@ -5,9 +5,9 @@ use bytes::Bytes;
 
 use crate::{
     body::Body,
+    form::FormBody,
     headers::{HeaderName, HeaderValue, Headers},
-    matchers,
-    matchers::Matcher,
+    matchers::{self, Matcher},
     request::Method,
 };
 
@@ -66,6 +66,12 @@ impl When {
         self
     }
 
+    /// Form Body.
+    pub fn form(self, body: FormBody) -> Self {
+        self.push(matchers::form(body));
+        self
+    }
+
     /// Headers.
     ///
     /// Cannonicalizes passed in header name and values to ensure matching is done in a

Original file line number	Diff line number	Diff line change
`@@ -30,6 +30,7 @@ pub mod prelude {`
`30`	`30`	`}`
`31`	`31`	`mod ext;`
`32`	`32`	`mod service;`
	`33`	`+mod form;`
`33`	`34`
`34`	`35`	`/// Represents errors that can occur while serving mocks.`
`35`	`36`	`#[derive(thiserror::Error, Debug)]`