76 lines
2.4 KiB
Rust
76 lines
2.4 KiB
Rust
use std::collections::HashMap;
|
|
use std::io::BufRead;
|
|
use std::path::PathBuf;
|
|
use std::sync::{Arc, Mutex};
|
|
use std::time::Duration;
|
|
use tokio::sync::mpsc;
|
|
|
|
use planetwars_matchrunner::docker_runner::DockerBotSpec;
|
|
use planetwars_matchrunner::match_context::{EventBus, MatchCtx, RequestError};
|
|
use planetwars_matchrunner::BotSpec;
|
|
use planetwars_matchrunner::{run_match, MatchConfig, MatchPlayer};
|
|
|
|
const PYTHON_IMAGE: &str = "python:3.10-slim-buster";
|
|
|
|
fn simple_python_docker_bot_spec(source_dir: &str, file_name: &str) -> DockerBotSpec {
|
|
let source_dir_path = std::fs::canonicalize(source_dir).unwrap();
|
|
let source_dir_path_str = source_dir_path.as_os_str().to_str().unwrap();
|
|
|
|
DockerBotSpec {
|
|
image: PYTHON_IMAGE.to_string(),
|
|
binds: Some(vec![format!("{}:{}", source_dir_path_str, "/workdir")]),
|
|
argv: Some(vec!["python".to_string(), file_name.to_string()]),
|
|
working_dir: Some("/workdir".to_string()),
|
|
pull: false,
|
|
credentials: None,
|
|
}
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn match_does_run() {
|
|
let bot = simple_python_docker_bot_spec("./bots/simplebot", "simplebot.py");
|
|
|
|
let log_file = tempfile::NamedTempFile::new().unwrap();
|
|
|
|
let config = MatchConfig {
|
|
map_name: "hex".to_string(),
|
|
map_path: PathBuf::from("maps/abc.json"),
|
|
log_path: PathBuf::from(log_file.path()),
|
|
players: vec![
|
|
MatchPlayer {
|
|
bot_spec: Box::new(bot.clone()),
|
|
},
|
|
MatchPlayer {
|
|
bot_spec: Box::new(bot.clone()),
|
|
},
|
|
],
|
|
};
|
|
|
|
run_match(config).await;
|
|
|
|
let line_count = std::io::BufReader::new(log_file.as_file()).lines().count();
|
|
assert!(line_count > 0);
|
|
|
|
tokio::time::sleep(Duration::from_secs(1)).await
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn docker_runner_timeout() {
|
|
let event_bus = Arc::new(Mutex::new(EventBus::new()));
|
|
let (logger, _rx) = mpsc::unbounded_channel();
|
|
let bot_spec = simple_python_docker_bot_spec("./bots", "timeout_bot.py");
|
|
|
|
let player_handle = bot_spec.run_bot(1, event_bus.clone(), logger.clone()).await;
|
|
|
|
let mut players = HashMap::new();
|
|
players.insert(1, player_handle);
|
|
let mut ctx = MatchCtx::new(event_bus, players, logger);
|
|
|
|
let resp = ctx
|
|
.request(1, b"sup".to_vec(), Duration::from_millis(1000))
|
|
.await;
|
|
|
|
assert_eq!(resp, Err(RequestError::Timeout));
|
|
ctx.shutdown().await;
|
|
}
|