test regular docker runner operation

2 files changed, 21 insertions, 0 deletions

diff --git a/planetwars-matchrunner/bots/echo_bot.py b/planetwars-matchrunner/bots/echo_bot.py
new file mode 100644
index 0000000..328d914
--- /dev/null
+++ b/planetwars-matchrunner/bots/echo_bot.py
@@ -0,0 +1,5 @@
+import sys
+
+for line in sys.stdin:
+    sys.stdout.write(line)
+    sys.stdout.flush()
\ No newline at end of file
diff --git a/planetwars-matchrunner/tests/test_matchrunner.rs b/planetwars-matchrunner/tests/test_matchrunner.rs
index 5b96406..131a7b8 100644
--- a/planetwars-matchrunner/tests/test_matchrunner.rs
+++ b/planetwars-matchrunner/tests/test_matchrunner.rs
@@ -78,6 +78,22 @@ where
 }
 
 #[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]
 async fn docker_runner_timeout() {
     let bot_spec = simple_python_docker_bot_spec("./bots", "timeout_bot.py");
     with_bot_match_ctx(bot_spec, |ctx| {