From aca64f3ef9158d56a1d53a8264aed0e91dc62aa1 Mon Sep 17 00:00:00 2001
From: CJ Holmes <cholmes@lear.com>
Date: Sun, 24 Mar 2024 19:43:09 -0700
Subject: [PATCH] 106 Add a unit test class; add tests for the parser

---
 README.md |  2 ++
 test.py   | 54 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 56 insertions(+)
 create mode 100644 test.py

diff --git a/README.md b/README.md
index ad69a24..a4ac1c9 100644
--- a/README.md
+++ b/README.md
@@ -28,3 +28,5 @@ https://discord.com/api/oauth2/authorize?client_id=1168083075515826186&permissio
 10. Use the OAuth2 URL to authorize the bot to talk to your server.
 
 You should be able to run the bot with `./run.sh`
+
+You can test the dice parsing and rolling code with `python test.py`
diff --git a/test.py b/test.py
new file mode 100644
index 0000000..1d8e9f0
--- /dev/null
+++ b/test.py
@@ -0,0 +1,54 @@
+#!.venv/bin/python3
+# MvKDiceBot: Discord bot for rolling dice for Mecha Vs Kaiju
+# Copyright (C) 2023  Eric Eisenhart
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <https://www.gnu.org/licenses/>.
+#
+# https://github.com/freiheit/MvKDiceBot
+"""Test module for MvKDiceBot"""
+
+import unittest
+import mvkroller as roller
+
+class TestRoller(unittest.TestCase):
+    """Test cases for the mvkroller functions."""
+
+    def test_parser_good(self):
+        """Test the parser with valid strings, including some that produce no dice."""
+        strings = {
+            "": {20:0, 12:0, 10:0, 8:0, 6:0, 4:0},
+            "there are no matching dice strings here": {20:0, 12:0, 10:0, 8:0, 6:0, 4:0},
+            "messyd20what?": {20:1, 12:0, 10:0, 8:0, 6:0, 4:0},
+            "d20 3d6 5d10": {20:1, 12:0, 10:5, 8:0, 6:3, 4:0},
+            "4d12 1d10 2d4": {20:0, 12:4, 10:1, 8:0, 6:0, 4:2},
+            " d4 d6 d8 d12 ": {20:0, 12:1, 10:0, 8:1, 6:1, 4:1},
+            "1024d20, 500d4": {20:1024, 12:0, 10:0, 8:0, 6:0, 4:500},
+        }
+        for (dstring, dspec) in strings.items():
+            with self.subTest(dstring=dstring):
+                self.assertEqual(roller.parse_dice(dstring), dspec)
+
+    def test_parser_bad(self):
+        """Ensure the parser will throw exceptions when bad dice sizes are given."""
+        strings = [
+            "50d3nothing",
+            "50d17",
+            "1d20 2d7",
+        ]
+        for dstring in strings:
+            with self.subTest(dstring=dstring), self.assertRaises(roller.RollError):
+                roller.parse_dice(dstring)
+
+if __name__ == "__main__":
+    unittest.main()