1 files changed, 46 insertions, 0 deletions
diff --git a/tests/binary_response.rs b/tests/binary_response.rs
new file mode 100644
index 0000000..47d3935
--- /dev/null
+++ b/tests/binary_response.rs
@@ -0,0 +1,46 @@
+use windmark::response::Response;
+
+// PNG magic + bytes invalid as UTF-8 (lone continuation bytes, lone 4-byte
+// lead, reserved C0/C1 leads, DEL).
+const NON_UTF8: &[u8] = &[
+  0x89, 0x50, 0x4E, 0x47, 0x0D, 0x0A, 0x1A, 0x0A, 0x80, 0x81, 0xFE, 0xFF, 0xC0,
+  0xC1, 0xF5, 0x00, 0x7F,
+];
+
+#[test]
+fn binary_success_preserves_non_utf8_bytes() {
+  let response = Response::binary_success(NON_UTF8, "image/png");
+
+  assert_eq!(response.status, 21);
+  assert_eq!(response.mime.as_deref(), Some("image/png"));
+  assert_eq!(response.binary_content.as_deref(), Some(NON_UTF8));
+}
+
+#[test]
+fn serialize_body_writes_binary_bytes_verbatim() {
+  let response = Response::binary_success(NON_UTF8, "image/png");
+  let body = response.serialize_body("ignored-header", "ignored-footer");
+
+  assert_eq!(body, NON_UTF8);
+}
+
+#[cfg(feature = "auto-deduce-mime")]
+#[test]
+fn binary_success_auto_preserves_non_utf8_bytes() {
+  let response = Response::binary_success_auto(NON_UTF8);
+
+  assert_eq!(response.status, 22);
+  assert_eq!(response.binary_content.as_deref(), Some(NON_UTF8));
+  assert_eq!(
+    response.serialize_body("ignored-header", "ignored-footer"),
+    NON_UTF8,
+  );
+}
+
+#[test]
+fn serialize_body_text_success_wraps_with_header_and_footer() {
+  let response = Response::success("body");
+  let body = response.serialize_body("HEADER\n", "FOOTER");
+
+  assert_eq!(body, b"HEADER\nbody\nFOOTER");
+}