test regular docker runner operation

This commit is contained in:
Ilion Beyst 2022-09-24 22:21:38 +02:00
parent 6cc629a3b7
commit e86fbc7275
2 changed files with 21 additions and 0 deletions

View file

@ -0,0 +1,5 @@
import sys
for line in sys.stdin:
sys.stdout.write(line)
sys.stdout.flush()

View file

@ -77,6 +77,22 @@ where
ctx.shutdown().await; ctx.shutdown().await;
} }
#[tokio::test]
async fn docker_runner_success() {
let bot_spec = simple_python_docker_bot_spec("./bots", "echo_bot.py");
with_bot_match_ctx(bot_spec, |ctx| {
async move {
let resp = ctx
.request(1, b"sup".to_vec(), Duration::from_millis(200))
.await;
assert_eq!(resp, Ok(b"sup\n".to_vec()));
}
.boxed()
})
.await;
}
#[tokio::test] #[tokio::test]
async fn docker_runner_timeout() { async fn docker_runner_timeout() {
let bot_spec = simple_python_docker_bot_spec("./bots", "timeout_bot.py"); let bot_spec = simple_python_docker_bot_spec("./bots", "timeout_bot.py");