Add top drinks ut

diff --git a/ut/test_db.py b/ut/test_db.py
index 7f58aa5cb3b81c390cffb440e527e16698b4b7e3..505855d3be58a530077e2b70c42b277fb07ba0a6 100644 (file)
--- 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 04f1ee1b2687d94c49165ce5bb0081136fb175c3..4f72f084197084f1806b9201655b169e9b3daefe 100644 (file)
--- 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()