botserver: Verify bot response.
This allows for more realistic testing.
This commit is contained in:
parent
4285cef29c
commit
5c062cee0d
|
@ -21,6 +21,7 @@ class BotServerTestCase(TestCase):
|
||||||
bots_config: Optional[Dict[str, Dict[str, str]]]=None,
|
bots_config: Optional[Dict[str, Dict[str, str]]]=None,
|
||||||
bot_handlers: Optional[Dict[str, Any]]=None,
|
bot_handlers: Optional[Dict[str, Any]]=None,
|
||||||
event: Optional[Dict[str, Any]]=None,
|
event: Optional[Dict[str, Any]]=None,
|
||||||
|
expected_response: Optional[str]=None,
|
||||||
check_success: bool=False,
|
check_success: bool=False,
|
||||||
third_party_bot_conf: Optional[configparser.ConfigParser]=None,
|
third_party_bot_conf: Optional[configparser.ConfigParser]=None,
|
||||||
) -> None:
|
) -> None:
|
||||||
|
@ -36,6 +37,16 @@ class BotServerTestCase(TestCase):
|
||||||
|
|
||||||
response = self.app.post(data=json.dumps(event))
|
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:
|
if check_success:
|
||||||
assert 200 <= response.status_code < 300
|
assert 200 <= response.status_code < 300
|
||||||
else:
|
else:
|
||||||
|
|
|
@ -32,6 +32,7 @@ class BotServerTests(BotServerTestCase):
|
||||||
bots_config=bots_config,
|
bots_config=bots_config,
|
||||||
event=dict(message={'content': "test message"},
|
event=dict(message={'content': "test message"},
|
||||||
bot_email='helloworld-bot@zulip.com'),
|
bot_email='helloworld-bot@zulip.com'),
|
||||||
|
expected_response="beep boop",
|
||||||
check_success=True)
|
check_success=True)
|
||||||
|
|
||||||
def test_successful_request_from_two_bots(self) -> None:
|
def test_successful_request_from_two_bots(self) -> None:
|
||||||
|
@ -51,6 +52,7 @@ class BotServerTests(BotServerTestCase):
|
||||||
self.assert_bot_server_response(available_bots=available_bots,
|
self.assert_bot_server_response(available_bots=available_bots,
|
||||||
event=dict(message={'content': "test message"},
|
event=dict(message={'content': "test message"},
|
||||||
bot_email='helloworld-bot@zulip.com'),
|
bot_email='helloworld-bot@zulip.com'),
|
||||||
|
expected_response="beep boop",
|
||||||
bots_config=bots_config,
|
bots_config=bots_config,
|
||||||
check_success=True)
|
check_success=True)
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue