def test_trivia_get_category_id_valid(self): return_val = { "trivia_categories": [{ "name": "cat1", "id": 1 }, { "name": "cat2", "id": 2 }, { "name": "cat3", "id": 3 }] } with patch("trivia.Trivia.get_category_dict", return_value=return_val) as mock_dict: self.assertEqual(Trivia.get_category_id("cat1"), 1) self.assertEqual(Trivia.get_category_id("cat2"), 2) self.assertEqual(Trivia.get_category_id("cat3"), 3) self.assertEqual(Trivia.get_category_id("cAT1"), 1) self.assertEqual(Trivia.get_category_id("CaT2"), 2) self.assertEqual(Trivia.get_category_id("cAt3"), 3) self.assertEqual(Trivia.get_category_id("Any"), 0) self.assertEqual(Trivia.get_category_id(None), 0) self.assertEqual(Trivia.get_category_id(""), 0)
def test_trivia_get_category_id_invalid(self): return_val = {"trivia_categories": [{"name": "science", "id": 1}]} with patch("trivia.Trivia.get_category_dict", return_value=return_val) as mock_dict: self.assertEqual(Trivia.get_category_id("Science"), 1) self.assertRaises(BotError, Trivia.get_category_id, "bad")