From: Jiri Vlasak <hubacji1@fel.cvut.cz>
Date: Wed, 3 Apr 2019 06:02:04 +0000 (+0200)
Subject: Add drink sum UT
X-Git-Tag: v0.3.0~5^2~3
X-Git-Url: https://rtime.felk.cvut.cz/gitweb/hubacji1/coffee-getter.git/commitdiff_plain/dd4ae639c029ffae4d27f0d76fc1b5fdd37a24fc?ds=sidebyside

Add drink sum UT
---

diff --git a/ut/test_db.py b/ut/test_db.py
index ff1b652..17dc3e9 100644
--- 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)