Add drink sum UT

diff --git a/ut/test_db.py b/ut/test_db.py
index ff1b65272c22117648b2a12c09a9a03cce22fad8..17dc3e925758ed9f24c517414c0e447659979712 100644 (file)
--- a/ut/test_db.py
+++ b/ut/test_db.py
@@ -230,3 +230,20 @@ class DrunkSum(TestCase):
             db.getDrunkSum,
             "espresso", "lungo", "2010-10-20", "2020-10-20"
         )
+
+    def test_all_mate_sum(self):
+        create_test_db(TESTDB_FILENAME)
+        try:
+            db = Db(TESTDB_FILENAME)
+            drunk = db.getDrunkSum(
+                "Club-Mate 0,5 l;Club-Mate 0,33 l",
+                self.lwstr,
+                self.tdystr
+            )
+            self.assertEqual(drunk, [
+                ("tester", 8),
+                ("newer", 2),
+                ("clone", 2),
+            ])
+        finally:
+            delete_test_db(TESTDB_FILENAME)