process: Share init in tests

This commit is contained in:
Alex Crichton
2019-06-24 16:56:48 -07:00
committed by Ivan Petkov
parent 124391e42b
commit 9680ecc109
3 changed files with 52 additions and 47 deletions
+43
View File
@@ -0,0 +1,43 @@
extern crate env_logger;
extern crate futures;
extern crate tokio_core;
extern crate tokio_process;
use std::env;
use std::process::Command;
use std::sync::{Once, ONCE_INIT};
use std::thread;
use std::sync::mpsc::channel;
use self::tokio_core::reactor::Core;
use self::futures::future;
use self::tokio_process::CommandExt;
pub fn init() {
static INIT: Once = ONCE_INIT;
INIT.call_once(|| {
drop(env_logger::init());
let (tx, rx) = channel();
thread::spawn(move || {
let mut lp = Core::new().unwrap();
let mut cmd = cmd("exit");
let mut child = cmd.spawn_async(&lp.handle()).unwrap();
drop(child.kill());
lp.run(child).unwrap();
tx.send(()).unwrap();
drop(lp.run(future::empty::<(), ()>()));
});
rx.recv().unwrap();
});
}
pub fn cmd(s: &str) -> Command {
let mut me = env::current_exe().unwrap();
me.pop();
if me.ends_with("deps") {
me.pop();
}
me.push(s);
Command::new(me)
}