X-Git-Url: http://rtime.felk.cvut.cz/gitweb/hubacji1/coffee-getter.git/blobdiff_plain/cd206e9cb965ad61bece6ffbc4866da4c7d7ac1e..45c1679a10b8d23adfd398eb53ec2f091a2398d8:/ut/test_db.py

diff --git a/ut/test_db.py b/ut/test_db.py
index 0c82bd1..f059c2b 100644
--- 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)