Add tests for directory error responses

Author: spacebear21

payjoin-directory/src/lib.rs

@@ -628,3 +628,87 @@ fn empty() -> BoxBody<Bytes, hyper::Error> {
 fn full<T: Into<Bytes>>(chunk: T) -> BoxBody<Bytes, hyper::Error> {
     Full::new(chunk.into()).map_err(|never| match never {}).boxed()
 }
+
+#[cfg(test)]
+mod tests {
+    use std::time::Duration;
+
+    use http_body_util::BodyExt;
+    use hyper::body::Bytes;
+    use hyper::{Method, Request, StatusCode};
+    use ohttp_relay::SentinelTag;
+    use payjoin::directory::ShortId;
+
+    use super::*;
+
+    async fn test_service(enable_v1: bool) -> Service<FilesDb> {
+        let dir = tempfile::tempdir().expect("tempdir");
+        let db = FilesDb::init(Duration::from_millis(100), dir.keep()).await.expect("db init");
+        let ohttp: ohttp::Server =
+            key_config::gen_ohttp_server_config().expect("ohttp config").into();
+        Service::new(db, ohttp, SentinelTag::new([0u8; 32]), enable_v1)
+    }
+
+    /// A valid ShortId encoded as bech32 for use in URL paths.
+    fn valid_short_id_path() -> String {
+        let id = ShortId([0u8; 8]);
+        id.to_string()
+    }
+
+    async fn collect_body(res: Response<BoxBody<Bytes, hyper::Error>>) -> (StatusCode, String) {
+        let (parts, body) = res.into_parts();
+        let bytes = body.collect().await.unwrap().to_bytes();
+        (parts.status, String::from_utf8(bytes.to_vec()).unwrap())
+    }
+
+    #[tokio::test]
+    async fn post_v1_when_disabled_returns_version_unsupported() {
+        let mut svc = test_service(false).await;
+        let id = valid_short_id_path();
+        let req = Request::builder()
+            .method(Method::POST)
+            .uri(format!("http://localhost/{id}"))
+            .body(Full::new(Bytes::from("base64-psbt")))
+            .unwrap();
+
+        let res = tower::Service::call(&mut svc, req).await.unwrap();
+        let (status, body) = collect_body(res).await;
+
+        assert_eq!(status, StatusCode::BAD_REQUEST);
+        assert_eq!(body, V1_VERSION_UNSUPPORTED_RES_JSON);
+    }
+
+    #[tokio::test]
+    async fn post_v1_with_invalid_body_returns_reject() {
+        let mut svc = test_service(true).await;
+        let id = valid_short_id_path();
+        let req = Request::builder()
+            .method(Method::POST)
+            .uri(format!("http://localhost/{id}"))
+            .body(Full::new(Bytes::from(vec![0xFF, 0xFE])))
+            .unwrap();
+
+        let res = tower::Service::call(&mut svc, req).await.unwrap();
+        let (status, body) = collect_body(res).await;
+
+        assert_eq!(status, StatusCode::BAD_REQUEST);
+        assert_eq!(body, V1_REJECT_RES_JSON);
+    }
+
+    #[tokio::test]
+    async fn post_v1_with_no_receiver_returns_unavailable() {
+        let mut svc = test_service(true).await;
+        let id = valid_short_id_path();
+        let req = Request::builder()
+            .method(Method::POST)
+            .uri(format!("http://localhost/{id}"))
+            .body(Full::new(Bytes::from("base64-psbt")))
+            .unwrap();
+
+        let res = tower::Service::call(&mut svc, req).await.unwrap();
+        let (status, body) = collect_body(res).await;
+
+        assert_eq!(status, StatusCode::SERVICE_UNAVAILABLE);
+        assert_eq!(body, V1_UNAVAILABLE_RES_JSON);
+    }
+}