From: Jiri Vlasak <hubacji1@fel.cvut.cz>
Date: Thu, 2 Jan 2020 10:48:30 +0000 (+0100)
Subject: Add drink smiles to getter
X-Git-Tag: v0.4.0~1^2~3
X-Git-Url: http://rtime.felk.cvut.cz/gitweb/hubacji1/coffee-getter.git/commitdiff_plain/a3adb45915a8b56ceb245f31e710fbf4a12360b8?hp=e7ae459fa34770573665cc4034f35bddaa4e6959

Add drink smiles to getter
---

diff --git a/cbgetter.py b/cbgetter.py
index bcdde26..4c4992e 100644
--- a/cbgetter.py
+++ b/cbgetter.py
@@ -25,6 +25,23 @@ class CoffeebotGetter:
         """Send message to Slack Bot."""
         return post(self.cfg.getCoffeebotURL(), dumps({"text": self.msg}))
 
+    def get_drink_smile(self, drink):
+        """Return Slack smile for ``drink``.
+
+        Keyword arguments:
+        drink -- Drink of smile.
+        """
+        smiles = {
+            "tea": ":tea:",
+            "espresso": ":coffee:",
+            "espresso lungo": ":coffin:",
+            "cappuccino": ":flara:",
+            "latte macchiato": ":baby_bottle:",
+            "Club-Mate 0,5 l": ":club-mate:",
+            "Club-Mate 0,33 l": ":club-mate_cola:",
+        }
+        return smiles[drink]
+
     def getHeroComment(self):
         """Return comment for hero."""
         hsuff = [
diff --git a/ut/test_getter.py b/ut/test_getter.py
index 4f72f08..0ebb4e5 100644
--- a/ut/test_getter.py
+++ b/ut/test_getter.py
@@ -33,6 +33,19 @@ class TopDrinks(TestCase):
         self.assertNotEqual(self.g.msg, "")
         self.assertTrue(len(self.g.msg) >= 0)
 
+    def test_drink_smile(self):
+        smiles = {
+            "tea": ":tea:",
+            "espresso": ":coffee:",
+            "espresso lungo": ":coffin:",
+            "cappuccino": ":flara:",
+            "latte macchiato": ":baby_bottle:",
+            "Club-Mate 0,5 l": ":club-mate:",
+            "Club-Mate 0,33 l": ":club-mate_cola:",
+        }
+        for (d, s) in smiles.items():
+            assert self.g.get_drink_smile(d) == s
+
 class TopMateDrinkers(TestCase):
     def test_top_mate_drinkers(self):
         g = CoffeebotGetter()