From b7b05d2865c771d55e96d842295f17f97e1b368e Mon Sep 17 00:00:00 2001
From: Noa Aarts <noa@voorwaarts.nl>
Date: Tue, 10 Dec 2024 22:39:59 +0100
Subject: [PATCH] add tests for caps versions in text protocol

---
 src/protocols/text_protocol.rs | 42 ++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/src/protocols/text_protocol.rs b/src/protocols/text_protocol.rs
index 1c711a2..160ec75 100644
--- a/src/protocols/text_protocol.rs
+++ b/src/protocols/text_protocol.rs
@@ -209,6 +209,20 @@ mod tests {
         );
     }
 
+    #[tokio::test]
+    async fn test_px_set_w_parse_caps() {
+        let parser = TextParser::default();
+        let reader = tokio_test::io::Builder::new()
+            .read(b"PX 28283 29991 AB\n")
+            .build();
+        let mut bufreader = BufReader::new(reader);
+        let thingy = parser.parse(&mut bufreader).await;
+        assert_eq!(
+            thingy.unwrap(),
+            Command::SetPixel(0, 28283, 29991, Color::W8(0xAB))
+        );
+    }
+
     #[tokio::test]
     async fn test_px_set_rgb_parse() {
         let parser = TextParser::default();
@@ -223,6 +237,20 @@ mod tests {
         );
     }
 
+    #[tokio::test]
+    async fn test_px_set_rgb_parse_caps() {
+        let parser = TextParser::default();
+        let reader = tokio_test::io::Builder::new()
+            .read(b"PX 28283 29991 8800FA\n")
+            .build();
+        let mut bufreader = BufReader::new(reader);
+        let thingy = parser.parse(&mut bufreader).await;
+        assert_eq!(
+            thingy.unwrap(),
+            Command::SetPixel(0, 28283, 29991, Color::RGB24(0x88, 0x00, 0xfa))
+        );
+    }
+
     #[tokio::test]
     async fn test_px_set_rgba_parse() {
         let parser = TextParser::default();
@@ -237,6 +265,20 @@ mod tests {
         );
     }
 
+    #[tokio::test]
+    async fn test_px_set_rgba_parse_caps() {
+        let parser = TextParser::default();
+        let reader = tokio_test::io::Builder::new()
+            .read(b"PX 28283 29991 AB0c3F88\n")
+            .build();
+        let mut bufreader = BufReader::new(reader);
+        let thingy = parser.parse(&mut bufreader).await;
+        assert_eq!(
+            thingy.unwrap(),
+            Command::SetPixel(0, 28283, 29991, Color::RGBA32(0xab, 0x0c, 0x3f, 0x88))
+        );
+    }
+
     #[tokio::test]
     async fn test_px_get_parse() {
         let parser = TextParser::default();