def test_can_auto_restart_pass(monkeypatch, caplog): """Test can_auto_restart for successful host selection.""" def select_suite_host(**_): return ('localhost', 'localhost') monkeypatch.setattr('cylc.flow.main_loop.auto_restart.select_suite_host', select_suite_host) assert _can_auto_restart() assert caplog.record_tuples == []
def test_can_auto_restart_fail_horribly(monkeypatch, caplog): """Test can_auto_restart for really unsuccessful host selection.""" def select_suite_host(**_): raise Exception('Unexpected error in host selection') monkeypatch.setattr('cylc.flow.main_loop.auto_restart.select_suite_host', select_suite_host) with caplog.at_level(level=logging.DEBUG, logger=CYLC_LOG): assert not _can_auto_restart() [(_, level, msg)] = caplog.record_tuples assert level == logging.CRITICAL assert 'Error in host selection' in msg
def test_can_auto_restart_fail(monkeypatch, caplog): """Test can_auto_restart for unsuccessful host selection.""" def select_suite_host(**_): raise HostSelectException({}) monkeypatch.setattr('cylc.flow.main_loop.auto_restart.select_suite_host', select_suite_host) with caplog.at_level(level=logging.DEBUG, logger=CYLC_LOG): assert not _can_auto_restart() [(_, level, msg)] = caplog.record_tuples assert level == logging.CRITICAL assert 'No alternative host to restart suite on' in msg