1 files changed, 50 insertions, 0 deletions
diff --git a/third_party/rust/warp/tests/cookie.rs b/third_party/rust/warp/tests/cookie.rs
new file mode 100644
index 0000000000..4cf286a3e1
--- /dev/null
+++ b/third_party/rust/warp/tests/cookie.rs
@@ -0,0 +1,50 @@
+#![deny(warnings)]
+
+#[tokio::test]
+async fn cookie() {
+    let foo = warp::cookie::<String>("foo");
+
+    let req = warp::test::request().header("cookie", "foo=bar");
+    assert_eq!(req.filter(&foo).await.unwrap(), "bar");
+
+    let req = warp::test::request().header("cookie", "abc=def; foo=baz");
+    assert_eq!(req.filter(&foo).await.unwrap(), "baz");
+
+    let req = warp::test::request().header("cookie", "abc=def");
+    assert!(!req.matches(&foo).await);
+
+    let req = warp::test::request().header("cookie", "foobar=quux");
+    assert!(!req.matches(&foo).await);
+}
+
+#[tokio::test]
+async fn optional() {
+    let foo = warp::cookie::optional::<String>("foo");
+
+    let req = warp::test::request().header("cookie", "foo=bar");
+    assert_eq!(req.filter(&foo).await.unwrap().unwrap(), "bar");
+
+    let req = warp::test::request().header("cookie", "abc=def; foo=baz");
+    assert_eq!(req.filter(&foo).await.unwrap().unwrap(), "baz");
+
+    let req = warp::test::request().header("cookie", "abc=def");
+    assert!(req.matches(&foo).await);
+
+    let req = warp::test::request().header("cookie", "foobar=quux");
+    assert!(req.matches(&foo).await);
+}
+
+#[tokio::test]
+async fn missing() {
+    let _ = pretty_env_logger::try_init();
+
+    let cookie = warp::cookie::<String>("foo");
+
+    let res = warp::test::request()
+        .header("cookie", "not=here")
+        .reply(&cookie)
+        .await;
+
+    assert_eq!(res.status(), 400);
+    assert_eq!(res.body(), "Missing request cookie \"foo\"");
+}