From: Jiri Vlasak <hubacji1@fel.cvut.cz>
Date: Thu, 2 Jan 2020 10:13:41 +0000 (+0100)
Subject: Add top drinks ut
X-Git-Tag: v0.4.0~1^2~6
X-Git-Url: http://rtime.felk.cvut.cz/gitweb/hubacji1/coffee-getter.git/commitdiff_plain/6f7ef2c0a6bcc96b2777c7b65ed8ff2bca767661

Add top drinks ut
---

diff --git a/ut/test_db.py b/ut/test_db.py
index 7f58aa5..505855d 100644
--- a/ut/test_db.py
+++ b/ut/test_db.py
@@ -108,6 +108,23 @@ class FailCoffeeDb(TestCase):
     def test_db_path_not_set(self):
         self.assertRaises(FileNotSetError, Db)
 
+class TopDrinks(TestCase):
+    def setUp(self):
+        create_test_db(TESTDB_FILENAME)
+        self.db = Db(TESTDB_FILENAME)
+
+    def tearDown(self):
+        delete_test_db(TESTDB_FILENAME)
+
+    def test_top_drinks(self):
+        top = self.db.get_top_drinks()
+        assert top == [
+            ("espresso lungo", 8),
+            ("Club-Mate 0,33 l", 8),
+            ("Club-Mate 0,5 l", 5),
+            ("tea", 3),
+        ]
+
 class TopMateDrinkers(TestCase):
     def test_test_top_mate_drinkers(self):
         create_test_db(TESTDB_FILENAME)
diff --git a/ut/test_getter.py b/ut/test_getter.py
index 04f1ee1..4f72f08 100644
--- a/ut/test_getter.py
+++ b/ut/test_getter.py
@@ -22,6 +22,17 @@ class SuccessInit(TestCase):
         r = g.sendMsgToSlack()
         self.assertIsInstance(r, Response)
 
+class TopDrinks(TestCase):
+    def setUp(self):
+        self.g = CoffeebotGetter()
+
+    def test_top_drinks(self):
+        top = self.g.load_top_drinks()
+        self.assertIsNotNone(self.g.top)
+        self.assertTrue(len(self.g.top) >= 0)
+        self.assertNotEqual(self.g.msg, "")
+        self.assertTrue(len(self.g.msg) >= 0)
+
 class TopMateDrinkers(TestCase):
     def test_top_mate_drinkers(self):
         g = CoffeebotGetter()