Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
Fix flaky tests (denoland#4134)
  • Loading branch information
ry committed Feb 26, 2020
1 parent 0e37184 commit 3d03d7e
Show file tree
Hide file tree
Showing 2 changed files with 27 additions and 28 deletions.
33 changes: 13 additions & 20 deletions cli/test_util.rs
Expand Up @@ -67,39 +67,32 @@ fn kill_http_server() {

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",
"--certfile",
root_path()
.join("std/http/testdata/tls/localhost.crt")
.to_str()
.unwrap(),
"--keyfile",
root_path()
.join("std/http/testdata/tls/localhost.key")
.to_str()
.unwrap(),
])
.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 mut lines = BufReader::new(stdout).lines();
let line = lines.next().unwrap().unwrap();
assert!(line.starts_with("ready"));
server_guard.replace(child);
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());
}
}
}
}

HttpServerGuard {}
}
22 changes: 14 additions & 8 deletions tools/http_server.py
Expand Up @@ -21,23 +21,20 @@
ANOTHER_REDIRECT_PORT = 4547
DOUBLE_REDIRECTS_PORT = 4548
INF_REDIRECTS_PORT = 4549

HTTPS_PORT = 5545


def create_http_arg_parser():
parser = argparse.ArgumentParser()
parser.add_argument('--certfile')
parser.add_argument('--keyfile')
parser.add_argument('--verbose', '-v', action='store_true')
return parser


HttpArgParser = create_http_arg_parser()

args, unknown = HttpArgParser.parse_known_args(sys.argv[1:])
CERT_FILE = args.certfile
KEY_FILE = args.keyfile
CERT_FILE = os.path.join(root_path, "std/http/testdata/tls/localhost.crt")
KEY_FILE = os.path.join(root_path, "std/http/testdata/tls/localhost.key")
QUIET = not args.verbose


Expand Down Expand Up @@ -314,10 +311,19 @@ def start(s):
def spawn():
servers = (server(), redirect_server(), another_redirect_server(),
double_redirects_server(), https_server())
while any(not s.thread.is_alive() for s in servers):
sleep(0.01)
# In order to wait for each of the servers to be ready, we try connecting to
# them with a tcp socket.
for running_server in servers:
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
port = running_server.server.server_address[1]
client.connect(("127.0.0.1", port))
print "connected", port
client.close()
assert running_server.thread.is_alive()
# The following output "ready" is specificly looked for in cli/test_util.rs
# to prevent race conditions.
print "ready"
try:
print "ready"
yield servers
finally:
for s in servers:
Expand Down

0 comments on commit 3d03d7e

Please sign in to comment.