xenofem/python-zulip-api
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

test_lib.py: Add mock_request_exception.

Browse source
This commit is contained in:
sColin16 2017-12-10 16:25:25 +00:00 committed by showell
parent 0a05a28e34
commit 1221c7107f
1 changed files with 17 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
zulip_bots/zulip_bots/test_lib.py
View file

@ -174,6 +174,10 @@ class StubBotTestCase(TestCase):
http_data = read_bot_fixture_data(self.bot_name, test_name) http_data = read_bot_fixture_data(self.bot_name, test_name)
return mock_http_conversation(http_data) return mock_http_conversation(http_data)
def mock_request_exception(self):
# type: () -> Any
return mock_request_exception()
def mock_config_info(self, config_info): def mock_config_info(self, config_info):
# type: (Dict[str, str]) -> Any # type: (Dict[str, str]) -> Any
return patch('zulip_bots.test_lib.StubBotHandler.get_config_info', return_value=config_info) return patch('zulip_bots.test_lib.StubBotHandler.get_config_info', return_value=config_info)
@ -258,6 +262,19 @@ def mock_http_conversation(http_data):
['params', 'headers', 'json'] ['params', 'headers', 'json']
) )
@contextmanager
def mock_request_exception():
# type: () -> Any
def assert_mock_called(mock_result):
# type: (Any) -> None
mock_result.assert_called()
with patch('requests.get') as mock_get:
mock_get.return_value = True
mock_get.side_effect = requests.exceptions.RequestException
yield
assert_mock_called(mock_get)
class BotTestCase(StubBotTestCase): class BotTestCase(StubBotTestCase):
"""Test class for common Bot test helper methods""" """Test class for common Bot test helper methods"""
bot_name = '' # type: str bot_name = '' # type: str