From b64a71e2fa34557011f6eb52ea323da56173e540 Mon Sep 17 00:00:00 2001
From: Maxime Augier <max@xolus.net>
Date: Tue, 4 May 2021 19:35:36 +0200
Subject: [PATCH] Add tests for messages with a leading colon in the suffix

---
 irc-proto/src/message.rs | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/irc-proto/src/message.rs b/irc-proto/src/message.rs
index 82ab7cc..6ef9d8f 100644
--- a/irc-proto/src/message.rs
+++ b/irc-proto/src/message.rs
@@ -547,4 +547,29 @@ mod test {
         let message = "@tag=\\:\\s\\\\\\r\\na :test PRIVMSG #test test\r\n";
         assert_eq!(msg, message);
     }
+
+    #[test]
+    fn to_message_with_colon_in_suffix() {
+        let msg = "PRIVMSG #test ::test"
+            .parse::<Message>()
+            .unwrap();
+        let message = Message {
+            tags: None,
+            prefix: None,
+            command: PRIVMSG("#test".to_string(), ":test".to_string())
+        };
+        assert_eq!(msg, message);
+    }
+
+    #[test]
+    fn to_string_with_colon_in_suffix() {
+        let msg = Message {
+            tags: None,
+            prefix: None,
+            command: PRIVMSG("#test".to_string(), ":test".to_string()),
+        }
+        .to_string();
+        let message = "PRIVMSG #test ::test\r\n";
+        assert_eq!(msg, message);
+    }
 }