botserver: Verify bot response.

This allows for more realistic testing.
This commit is contained in:
Robert Hönig 2018-05-28 18:47:18 +02:00 committed by Tim Abbott
parent 4285cef29c
commit 5c062cee0d
2 changed files with 13 additions and 0 deletions

View file

@ -21,6 +21,7 @@ class BotServerTestCase(TestCase):
bots_config: Optional[Dict[str, Dict[str, str]]]=None,
bot_handlers: Optional[Dict[str, Any]]=None,
event: Optional[Dict[str, Any]]=None,
expected_response: Optional[str]=None,
check_success: bool=False,
third_party_bot_conf: Optional[configparser.ConfigParser]=None,
) -> None:
@ -36,6 +37,16 @@ class BotServerTestCase(TestCase):
response = self.app.post(data=json.dumps(event))
# NOTE: Currently, assert_bot_server_response can only check the expected_response
# for bots that use send_reply. However, the vast majority of bots use send_reply.
# Therefore, the Botserver can be still be effectively tested.
bot_send_reply_call = mock_ExternalBotHandler.return_value.send_reply
if expected_response is not None:
self.assertTrue(bot_send_reply_call.called)
self.assertEqual(expected_response, bot_send_reply_call.call_args[0][1])
else:
self.assertFalse(bot_send_reply_call.called)
if check_success:
assert 200 <= response.status_code < 300
else: