async def test_watchdog_firing():
"""Test that the watchdog expiring fires the provided coroutine."""
mock_coro = CoroutineMock()
mock_coro.__name__ = "mock_coro"
watchdog = WebsocketWatchdog(mock_coro)
await watchdog.on_expire()
mock_coro.assert_called_once()
def test_main_async(mocker, event_loop):
COMMANDS = []
mocker.patch('jenkins_epo.main.COMMANDS', COMMANDS)
command = CoroutineMock()
command.__name__ = 'bot'
command.__code__ = Mock(co_varnames=(), co_argcount=0)
COMMANDS.append(command)
from jenkins_epo.main import main
main(argv=['bot'], loop=event_loop)
assert command.mock_calls
def test_main_async(mocker, event_loop):
COMMANDS = []
mocker.patch('jenkins_epo.main.COMMANDS', COMMANDS)
command = CoroutineMock()
command.__name__ = 'bot'
command.__code__ = Mock(co_varnames=(), co_argcount=0)
COMMANDS.append(command)
from jenkins_epo.main import main
main(argv=['bot'], loop=event_loop)
assert command.mock_calls
def test_main_async_exception(mocker, event_loop):
COMMANDS = []
mocker.patch('jenkins_epo.main.COMMANDS', COMMANDS)
command = CoroutineMock()
command.__name__ = 'bot'
command.__code__ = Mock(co_varnames=(), co_argcount=0)
command.side_effect = ValueError()
COMMANDS.append(command)
from jenkins_epo.main import main
with pytest.raises(ValueError):
main(argv=['bot'], loop=event_loop)
assert command.mock_calls
assert event_loop.is_closed()
def test_main_async_exception(mocker, event_loop):
COMMANDS = []
mocker.patch('jenkins_epo.main.COMMANDS', COMMANDS)
command = CoroutineMock()
command.__name__ = 'bot'
command.__code__ = Mock(co_varnames=(), co_argcount=0)
command.side_effect = ValueError()
COMMANDS.append(command)
from jenkins_epo.main import main
with pytest.raises(ValueError):
main(argv=['bot'], loop=event_loop)
assert command.mock_calls
assert event_loop.is_closed()
