Decouple mock_request_exception from StubBotTestCase.

Tests that need mock_request_exception can just import it
directly now.
This commit is contained in:
Steve Howell 2017-12-11 09:26:36 -06:00
parent 1dbb73a1c8
commit 971b48cbfc
2 changed files with 2 additions and 6 deletions

View file

@ -1,6 +1,7 @@
#!/usr/bin/env python
from zulip_bots.test_lib import StubBotTestCase
from zulip_bots.request_test_lib import mock_request_exception
class TestWikipediaBot(StubBotTestCase):
bot_name = "wikipedia"
@ -65,5 +66,5 @@ class TestWikipediaBot(StubBotTestCase):
# Request Exception
bot_request = 'Z'
bot_response = None
with self.mock_request_exception():
with mock_request_exception():
self.verify_reply(bot_request, bot_response)

View file

@ -8,7 +8,6 @@ from typing import List, Dict, Any, Tuple
from zulip_bots.request_test_lib import (
mock_http_conversation,
mock_request_exception,
)
from zulip_bots.simple_lib import (
@ -164,10 +163,6 @@ class StubBotTestCase(TestCase):
http_data = read_bot_fixture_data(self.bot_name, test_name)
return mock_http_conversation(http_data)
def mock_request_exception(self):
# type: () -> Any
return mock_request_exception()
def mock_config_info(self, config_info):
# type: (Dict[str, str]) -> Any
return patch('zulip_bots.test_lib.StubBotHandler.get_config_info', return_value=config_info)