ut/test_getter.py

   1 # -*- coding: utf-8 -*-
   2 """Test Coffeebot Getter."""
   3 from requests import Response
   4 from unittest import TestCase
   5 from cbgetter import CoffeebotGetter
   6
   7 class SuccessInit(TestCase):
   8     def test_not_blank(self):
   9         g = CoffeebotGetter()
  10         self.assertIsNotNone(g.cfg)
  11         self.assertIsNotNone(g.db)
  12         self.assertEqual(g.msg, "")
  13         self.assertEqual(g.top, [])
  14
  15     def test_msg_getter_is_blank_by_default(self):
  16         g = CoffeebotGetter()
  17         self.assertEqual(g.getMsg(), "")
  18
  19     def test_msg_sent_to_slack(self):
  20         g = CoffeebotGetter()
  21         r = g.sendMsgToSlack()
  22         self.assertIsInstance(r, Response)
  23
  24 class TopMateDrinkers(TestCase):
  25     def test_top_mate_drinkers(self):
  26         g = CoffeebotGetter()
  27         g.loadTopMateDrinkers()
  28         self.assertIsNotNone(g.top)
  29         self.assertTrue(len(g.top) >= 0)
  30         self.assertNotEqual(g.msg, "")
  31         self.assertTrue(len(g.msg) >= 0)
  32
  33     def test_get_hero_comment(self):
  34         g = CoffeebotGetter()
  35         self.assertNotEqual(g.getHeroComment(), "")
  36
  37     def test_get_some_smile(self):
  38         g = CoffeebotGetter()
  39         self.assertNotEqual(g.getSomeSmile(), "")
  40         self.assertEqual(g.getSomeSmile()[0], ":")
  41         self.assertEqual(g.getSomeSmile()[-1], ":")