* Generate a script to rerun the failed sub-command in cargo-test-bpf * Generate a script to rerun the failed sub-command in cargo-build-bpf * Add cargo-build-bpf test for generate-child-script-on-failure option
		
			
				
	
	
		
			96 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Rust
		
	
	
	
	
	
			
		
		
	
	
			96 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Rust
		
	
	
	
	
	
use std::{
 | 
						|
    env, fs,
 | 
						|
    io::{self, Write},
 | 
						|
    process::{Command, Output},
 | 
						|
};
 | 
						|
 | 
						|
#[macro_use]
 | 
						|
extern crate serial_test;
 | 
						|
 | 
						|
fn run_cargo_build(crate_name: &str, extra_args: &[&str]) -> Output {
 | 
						|
    let cwd = env::current_dir().expect("Unable to get current working directory");
 | 
						|
    let root = cwd
 | 
						|
        .parent()
 | 
						|
        .expect("Unable to get parent directory of current working dir")
 | 
						|
        .parent()
 | 
						|
        .expect("Unable to get ../.. of current working dir");
 | 
						|
    let toml = cwd
 | 
						|
        .join("tests")
 | 
						|
        .join("crates")
 | 
						|
        .join(crate_name)
 | 
						|
        .join("Cargo.toml");
 | 
						|
    let toml = format!("{}", toml.display());
 | 
						|
    let mut args = vec!["--bpf-sdk", "../bpf", "--manifest-path", &toml];
 | 
						|
    for arg in extra_args {
 | 
						|
        args.push(arg);
 | 
						|
    }
 | 
						|
    let cargo_build_bpf = root.join("target").join("debug").join("cargo-build-bpf");
 | 
						|
    let output = Command::new(cargo_build_bpf)
 | 
						|
        .args(&args)
 | 
						|
        .output()
 | 
						|
        .expect("Error running cargo-build-bpf");
 | 
						|
    if !output.status.success() {
 | 
						|
        eprintln!("--- stdout ---");
 | 
						|
        io::stderr().write_all(&output.stdout).unwrap();
 | 
						|
        eprintln!("--- stderr ---");
 | 
						|
        io::stderr().write_all(&output.stderr).unwrap();
 | 
						|
        eprintln!("--------------");
 | 
						|
    }
 | 
						|
    output
 | 
						|
}
 | 
						|
 | 
						|
#[test]
 | 
						|
#[serial]
 | 
						|
fn test_build() {
 | 
						|
    let output = run_cargo_build("noop", &[]);
 | 
						|
    assert!(output.status.success());
 | 
						|
}
 | 
						|
 | 
						|
#[test]
 | 
						|
#[serial]
 | 
						|
fn test_dump() {
 | 
						|
    // This test requires rustfilt.
 | 
						|
    assert!(Command::new("cargo")
 | 
						|
        .args(&["install", "rustfilt"])
 | 
						|
        .status()
 | 
						|
        .expect("Unable to install rustfilt required for --dump option")
 | 
						|
        .success());
 | 
						|
    let output = run_cargo_build("noop", &["--dump"]);
 | 
						|
    assert!(output.status.success());
 | 
						|
    let cwd = env::current_dir().expect("Unable to get current working directory");
 | 
						|
    let dump = cwd
 | 
						|
        .join("tests")
 | 
						|
        .join("crates")
 | 
						|
        .join("noop")
 | 
						|
        .join("target")
 | 
						|
        .join("deploy")
 | 
						|
        .join("noop-dump.txt");
 | 
						|
    assert!(dump.exists());
 | 
						|
}
 | 
						|
 | 
						|
#[test]
 | 
						|
#[serial]
 | 
						|
fn test_out_dir() {
 | 
						|
    let output = run_cargo_build("noop", &["--bpf-out-dir", "tmp_out"]);
 | 
						|
    assert!(output.status.success());
 | 
						|
    let cwd = env::current_dir().expect("Unable to get current working directory");
 | 
						|
    let dir = cwd.join("tmp_out");
 | 
						|
    assert!(dir.exists());
 | 
						|
    fs::remove_dir_all("tmp_out").expect("Failed to remove tmp_out dir");
 | 
						|
}
 | 
						|
 | 
						|
#[test]
 | 
						|
#[serial]
 | 
						|
fn test_generate_child_script_on_failre() {
 | 
						|
    let output = run_cargo_build("fail", &["--generate-child-script-on-failure"]);
 | 
						|
    assert!(!output.status.success());
 | 
						|
    let cwd = env::current_dir().expect("Unable to get current working directory");
 | 
						|
    let scr = cwd
 | 
						|
        .join("tests")
 | 
						|
        .join("crates")
 | 
						|
        .join("fail")
 | 
						|
        .join("cargo-build-bpf-child-script-cargo.sh");
 | 
						|
    assert!(scr.exists());
 | 
						|
    fs::remove_file(scr).expect("Failed to remove script");
 | 
						|
}
 |