Add drunk list getter UT

[hubacji1/coffee-getter.git] / ut / test_getter.py

diff --git a/ut/test_getter.py b/ut/test_getter.py
index 1d61f7d5193a9710e293ec27fb030a5295abb1c0..0324dabfcd39648f36e2b8a6a7371a3ed7c051a1 100644 (file)
--- a/ut/test_getter.py
+++ b/ut/test_getter.py
@@ -86,3 +86,19 @@ class DrunkSum(TestCase):
             g.loadDrunkSum,
             ["espresso", "lungo", "2010-10-20", "2020-10-20"]
         )
+
+class DrunkList(TestCase):
+    def test_nof_args(self):
+        g = CoffeebotGetter()
+        self.assertRaises(ArgCountError, g.loadDrunkList)
+        self.assertRaises(ArgCountError, g.loadDrunkList, ["espresso"])
+        self.assertRaises(
+            ArgCountError,
+            g.loadDrunkList,
+            ["espresso", "2010-10-20"]
+        )
+        self.assertRaises(
+            ArgCountError,
+            g.loadDrunkList,
+            ["espresso", "lungo", "2010-10-20", "2020-10-20"]
+        )