Skip to content

Commit

Permalink
tests: alter integration tests for stdio to not use a handrolled cat …
Browse files Browse the repository at this point in the history
…implementation

This fixes #4801, where, as a result of rust-lang/rust#95469, our implementation of cat used for this test no longer works, as stdio functions on windows now can abort the process if the pipe is set to nonblocking mode.

Unfortunately in windows, setting one end of the pipe to be nonblocking makes the whole thing nonblocking, so when, in tokio::process we set the child pipes to nonblocking mode, it causes serious problems for any rust program at the other end.

Fixing this issue is for another day, but fixing the tests is for today.
  • Loading branch information
Noah-Kennedy committed Jul 2, 2022
1 parent 8ed06ef commit 5e271d7
Show file tree
Hide file tree
Showing 3 changed files with 23 additions and 17 deletions.
2 changes: 2 additions & 0 deletions tests-integration/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ publish = false

[[bin]]
name = "test-cat"
required-features = ["rt-process-io-util"]

[[bin]]
name = "test-mem"
Expand All @@ -17,6 +18,7 @@ name = "test-process-signal"
required-features = ["rt-process-signal"]

[features]
rt-process-io-util = ["tokio/rt", "tokio/macros", "tokio/process", "tokio/io-util", "tokio/io-std"]
# For mem check
rt-net = ["tokio/rt", "tokio/rt-multi-thread", "tokio/net"]
# For test-process-signal
Expand Down
24 changes: 9 additions & 15 deletions tests-integration/src/bin/test-cat.rs
Original file line number Diff line number Diff line change
@@ -1,20 +1,14 @@
//! A cat-like utility that can be used as a subprocess to test I/O
//! stream communication.

use std::io;
use std::io::Write;
use tokio::io::AsyncWriteExt;

fn main() {
let stdin = io::stdin();
let mut stdout = io::stdout();
let mut line = String::new();
loop {
line.clear();
stdin.read_line(&mut line).unwrap();
if line.is_empty() {
break;
}
stdout.write_all(line.as_bytes()).unwrap();
}
stdout.flush().unwrap();
#[tokio::main(flavor = "current_thread")]
async fn main() {
let mut stdin = tokio::io::stdin();
let mut stdout = tokio::io::stdout();

tokio::io::copy(&mut stdin, &mut stdout).await.unwrap();

stdout.flush().await.unwrap();
}
14 changes: 12 additions & 2 deletions tests-integration/tests/process_stdio.rs
Original file line number Diff line number Diff line change
Expand Up @@ -8,12 +8,22 @@ use tokio_test::assert_ok;

use futures::future::{self, FutureExt};
use std::convert::TryInto;
use std::env;
use std::io;
use std::process::{ExitStatus, Stdio};

// so, we need to change this back as a test, but for now this doesn't work because of:
// https://github.com/rust-lang/rust/pull/95469
//
// undo when this is closed: https://github.com/tokio-rs/tokio/issues/4802

// fn cat() -> Command {
// let mut cmd = Command::new(std::env!("CARGO_BIN_EXE_test-cat"));
// cmd.stdin(Stdio::piped()).stdout(Stdio::piped());
// cmd
// }

fn cat() -> Command {
let mut cmd = Command::new(env!("CARGO_BIN_EXE_test-cat"));
let mut cmd = Command::new("cat");
cmd.stdin(Stdio::piped()).stdout(Stdio::piped());
cmd
}
Expand Down

0 comments on commit 5e271d7

Please sign in to comment.