X-Git-Url: http://rtime.felk.cvut.cz/gitweb/hubacji1/coffee-getter.git/blobdiff_plain/603aace8124050ba8e05cee5dfe181d20872c125..dd43805071be3fdc0db82a0d1483b48fad7235b9:/ut/test_getter.py

diff --git a/ut/test_getter.py b/ut/test_getter.py
index 1d61f7d..0324dab 100644
--- 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"]
+        )