mirror of
https://github.com/denoland/deno.git
synced 2025-01-10 16:11:13 -05:00
This reverts commit e6167c7813
.
This commit is contained in:
parent
1cb1ab6c00
commit
9b6a9b769e
1 changed files with 46 additions and 57 deletions
103
cli/test_util.rs
103
cli/test_util.rs
|
@ -8,13 +8,11 @@ use std::path::PathBuf;
|
||||||
use std::process::Child;
|
use std::process::Child;
|
||||||
use std::process::Command;
|
use std::process::Command;
|
||||||
use std::process::Stdio;
|
use std::process::Stdio;
|
||||||
use std::sync::atomic::AtomicUsize;
|
|
||||||
use std::sync::atomic::Ordering;
|
|
||||||
use std::sync::Mutex;
|
use std::sync::Mutex;
|
||||||
|
use std::sync::MutexGuard;
|
||||||
|
|
||||||
lazy_static! {
|
lazy_static! {
|
||||||
static ref SERVER: Mutex<Option<Child>> = Mutex::new(None);
|
static ref GUARD: Mutex<()> = Mutex::new(());
|
||||||
static ref SERVER_COUNT: AtomicUsize = AtomicUsize::new(0);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn root_path() -> PathBuf {
|
pub fn root_path() -> PathBuf {
|
||||||
|
@ -37,62 +35,53 @@ pub fn deno_exe_path() -> PathBuf {
|
||||||
p
|
p
|
||||||
}
|
}
|
||||||
|
|
||||||
pub struct HttpServerGuard {}
|
pub struct HttpServerGuard<'a> {
|
||||||
|
#[allow(dead_code)]
|
||||||
|
g: MutexGuard<'a, ()>,
|
||||||
|
child: Child,
|
||||||
|
}
|
||||||
|
|
||||||
impl Drop for HttpServerGuard {
|
impl<'a> Drop for HttpServerGuard<'a> {
|
||||||
fn drop(&mut self) {
|
fn drop(&mut self) {
|
||||||
let count = SERVER_COUNT.fetch_sub(1, Ordering::SeqCst);
|
match self.child.try_wait() {
|
||||||
// If no more tests hold guard we can kill the server
|
Ok(None) => {
|
||||||
|
self.child.kill().expect("failed to kill http_server.py");
|
||||||
if count == 1 {
|
|
||||||
kill_http_server();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn kill_http_server() {
|
|
||||||
let mut server_guard = SERVER.lock().unwrap();
|
|
||||||
let mut child = server_guard
|
|
||||||
.take()
|
|
||||||
.expect("Trying to kill server but already killed");
|
|
||||||
match child.try_wait() {
|
|
||||||
Ok(None) => {
|
|
||||||
child.kill().expect("failed to kill http_server.py");
|
|
||||||
}
|
|
||||||
Ok(Some(status)) => panic!("http_server.py exited unexpectedly {}", status),
|
|
||||||
Err(e) => panic!("http_server.py error: {}", e),
|
|
||||||
}
|
|
||||||
drop(server_guard);
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn http_server() -> HttpServerGuard {
|
|
||||||
SERVER_COUNT.fetch_add(1, Ordering::SeqCst);
|
|
||||||
{
|
|
||||||
let mut server_guard = SERVER.lock().unwrap();
|
|
||||||
if server_guard.is_none() {
|
|
||||||
println!("tools/http_server.py starting...");
|
|
||||||
let mut child = Command::new("python")
|
|
||||||
.current_dir(root_path())
|
|
||||||
.args(&["-u", "tools/http_server.py"])
|
|
||||||
.stdout(Stdio::piped())
|
|
||||||
.spawn()
|
|
||||||
.expect("failed to execute child");
|
|
||||||
|
|
||||||
let stdout = child.stdout.as_mut().unwrap();
|
|
||||||
use std::io::{BufRead, BufReader};
|
|
||||||
let lines = BufReader::new(stdout).lines();
|
|
||||||
// Wait for "ready" on stdout. See tools/http_server.py
|
|
||||||
for maybe_line in lines {
|
|
||||||
if let Ok(line) = maybe_line {
|
|
||||||
if line.starts_with("ready") {
|
|
||||||
server_guard.replace(child);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
panic!(maybe_line.unwrap_err());
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
Ok(Some(status)) => {
|
||||||
|
panic!("http_server.py exited unexpectedly {}", status)
|
||||||
|
}
|
||||||
|
Err(e) => panic!("http_server.py err {}", e),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
HttpServerGuard {}
|
}
|
||||||
|
|
||||||
|
/// Starts tools/http_server.py when the returned guard is dropped, the server
|
||||||
|
/// will be killed.
|
||||||
|
pub fn http_server<'a>() -> HttpServerGuard<'a> {
|
||||||
|
// TODO(bartlomieju) Allow tests to use the http server in parallel.
|
||||||
|
let g = GUARD.lock().unwrap();
|
||||||
|
|
||||||
|
println!("tools/http_server.py starting...");
|
||||||
|
let mut child = Command::new("python")
|
||||||
|
.current_dir(root_path())
|
||||||
|
.args(&["-u", "tools/http_server.py"])
|
||||||
|
.stdout(Stdio::piped())
|
||||||
|
.spawn()
|
||||||
|
.expect("failed to execute child");
|
||||||
|
|
||||||
|
let stdout = child.stdout.as_mut().unwrap();
|
||||||
|
use std::io::{BufRead, BufReader};
|
||||||
|
let lines = BufReader::new(stdout).lines();
|
||||||
|
// Wait for "ready" on stdout. See tools/http_server.py
|
||||||
|
for maybe_line in lines {
|
||||||
|
if let Ok(line) = maybe_line {
|
||||||
|
if line.starts_with("ready") {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
panic!(maybe_line.unwrap_err());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
HttpServerGuard { child, g }
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue