From 412039b568e2ff506e1732bce72fe522381e3e62 Mon Sep 17 00:00:00 2001
From: Mark Joshwel <89562141+markjoshwel@users.noreply.github.com>
Date: Thu, 7 Sep 2023 01:53:35 +0800
Subject: [PATCH] s+,test: implement #25 with accompanying test (#29)

---
 surplus/surplus.py |  2 ++
 test.py            | 12 ++++++++++++
 2 files changed, 14 insertions(+)

diff --git a/surplus/surplus.py b/surplus/surplus.py
index 478f9a6..8bf0064 100644
--- a/surplus/surplus.py
+++ b/surplus/surplus.py
@@ -862,6 +862,8 @@ def parse_query(behaviour: Behaviour) -> Result[Query]:
             split_query = behaviour.query.split(" ")
 
         for word in split_query:
+            word = word.strip(",").strip()
+
             if validator.is_valid(word):
                 portion_plus_code = word
 
diff --git a/test.py b/test.py
index d3dc37c..369c4cf 100644
--- a/test.py
+++ b/test.py
@@ -121,6 +121,18 @@ tests: list[ContinuityTest] = [
             )
         ],
     ),
+    ContinuityTest(
+        query="8WWJ+4P, Singapore",  # a comma!
+        expected=[
+            (
+                "Temasek Polytechnic\n"
+                "21 Tampines Avenue 1\n"
+                "Tampines West\n"
+                "529757\n"
+                "Northeast, Singapore"
+            )
+        ],
+    ),
 ]