def test_trivia_command_callback(self):
query = BotQuery()
query.has_callback_query = True
query.callback_query = BotQuery()
query.callback_query.callback_data = "payload here/trivia"
result = CommandParser.parse_command(query)
self.assertIsInstance(result, TriviaCommand)
self.assertIs(query, result.query)
def test_trivia_callback_query(self):
query = BotQuery()
query.has_callback_query = True
trivia_command = TriviaCommand(query)
trivia_command.handle_callback_query = Mock()
trivia_command.execute()
trivia_command.handle_callback_query.assert_called_once()
def test_trivia_callback_query_correct(self):
query = BotQuery()
query.has_callback_query = True
query.callback_query = BotQuery()
query.callback_query.callback_data = "CC/trivia"
trivia_command = TriviaCommand(query)
with patch("commands.trivia_command.requests.post") as mock_post:
trivia_command.execute()
response_text = mock_post.call_args[1]["json"]["text"]
self.assertTrue(response_text.endswith("*C* is correct!"))