From bb1de0e0cc5714153aa80e66f0576b86da58db01 Mon Sep 17 00:00:00 2001
From: Solene Rapenne <solene@perso.pw>
Date: Sat, 13 Aug 2022 14:00:29 +0200
Subject: [PATCH] more unit tests

---
 unit.hs | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/unit.hs b/unit.hs
index 4e1b2ff..d4f09a3 100644
--- a/unit.hs
+++ b/unit.hs
@@ -37,6 +37,17 @@ regex_7 = TestCase (assertEqual
   [["gemini://perso.pw/directory/\r\n", "gemini://", "perso.pw", "directory/", "", "", "\r\n"]]
   (parse_url "gemini://perso.pw/../../directory/\r\n"))
 
+regex_8 = TestCase (assertEqual
+  "directory requested with a query"
+  [["gemini://perso.pw/directory/?query=value\r\n", "gemini://", "perso.pw", "directory/", "?", "query=value", "\r\n"]]
+  (parse_url "gemini://perso.pw/directory/?query=value\r\n"))
+
+regex_9 = TestCase (assertEqual
+  "directory requested with a query without a trailing slash"
+  [["gemini://perso.pw/directory?query=value\r\n", "gemini://", "perso.pw", "directory", "?", "query=value", "\r\n"]]
+  (parse_url "gemini://perso.pw/directory?query=value\r\n"))
+
+
 tests = TestList
   [ regex_1
   , regex_2
@@ -45,6 +56,8 @@ tests = TestList
   , regex_5
   , regex_6
   , regex_7
+  , regex_8
+  , regex_9
   ]
 
 main :: IO Counts