Add all mate count in list UT

diff --git a/ut/test_db.py b/ut/test_db.py
index 0c82bd164cabc56902b9b66dae445e0b8a9c0111..f059c2ba0a1184a7662360e9c0b776d26f976012 100644 (file)
--- a/ut/test_db.py
+++ b/ut/test_db.py
@@ -249,3 +249,20 @@ class DrunkSum(TestCase):
             ])
         finally:
             delete_test_db(TESTDB_FILENAME)
+
+    def test_all_mate_list(self):
+        create_test_db(TESTDB_FILENAME)
+        try:
+            db = Db(TESTDB_FILENAME)
+            drunk = db.getDrunkList(
+                "Club-Mate 0,5 l;Club-Mate 0,33 l",
+                self.lwstr,
+                self.tdystr
+            )
+            self.assertEqual(drunk, {
+                "tester": [4, 4],
+                "newer": [2],
+                "clone": [1, 2],
+            })
+        finally:
+            delete_test_db(TESTDB_FILENAME)