Add drunk sum UT

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

diff --git a/ut/test_getter.py b/ut/test_getter.py
index 7fe5d8eb3281d00840c4d22ef1f0da08d78f6b5c..1d61f7d5193a9710e293ec27fb030a5295abb1c0 100644 (file)
--- a/ut/test_getter.py
+++ b/ut/test_getter.py
@@ -2,6 +2,7 @@
 """Test Coffeebot Getter."""
 from requests import Response
 from unittest import TestCase
+from cbdb import ArgCountError
 from cbgetter import CoffeebotGetter
 
 class SuccessInit(TestCase):
@@ -69,3 +70,19 @@ class EqualDrinkers(TestCase):
         g.top = self.TOP
         g.computeEqualDrinkers()
         self.assertEqual(len(g.topeq), 2)
+
+class DrunkSum(TestCase):
+    def test_nof_args(self):
+        g = CoffeebotGetter()
+        self.assertRaises(ArgCountError, g.loadDrunkSum)
+        self.assertRaises(ArgCountError, g.loadDrunkSum, ["espresso"])
+        self.assertRaises(
+            ArgCountError,
+            g.loadDrunkSum,
+            ["espresso", "2010-10-20"]
+        )
+        self.assertRaises(
+            ArgCountError,
+            g.loadDrunkSum,
+            ["espresso", "lungo", "2010-10-20", "2020-10-20"]
+        )