def test_commandline(monkeypatch, config_file, args, cold_start): server = Mock() Server = Mock(return_value=server) server.is_alive.return_value = False import spalloc_server.server monkeypatch.setattr(spalloc_server.server, "Server", Server) main(args.format(config_file).split()) Server.assert_called_once_with(config_filename=config_file, cold_start=cold_start)
def test_keyboard_interrupt(monkeypatch, config_file): s = Mock() Server = Mock(return_value=s) import spalloc_server.server monkeypatch.setattr(spalloc_server.server, "Server", Server) s.is_alive.side_effect = KeyboardInterrupt main([config_file]) Server.assert_called_once_with(config_filename=config_file, cold_start=False) s.is_alive.assert_called_once_with() s.stop_and_join.assert_called_once_with()