use std::fs;
use std::path::Path;
use std::thread;
use assert_cmd::Command;
use predicates::str::contains;
use tempfile::tempdir;
use tiny_http::{Header, Response, Server, StatusCode};
fn write_file(path: &Path, content: &str) {
if let Some(parent) = path.parent() {
fs::create_dir_all(parent).expect("mkdir");
}
fs::write(path, content).expect("write");
}
fn create_workspace(root: &Path) {
write_file(
&root.join("Cargo.toml"),
r#"
[workspace]
members = ["demo"]
resolver = "2"
"#,
);
write_file(
&root.join("demo/Cargo.toml"),
r#"
[package]
name = "demo"
version = "0.1.0"
edition = "2021"
"#,
);
write_file(&root.join("demo/src/lib.rs"), "pub fn demo() {}\n");
}
fn shipper_cmd() -> Command {
Command::new(assert_cmd::cargo::cargo_bin!("shipper-cli"))
}
struct TestRegistry {
base_url: String,
handle: thread::JoinHandle<()>,
}
impl TestRegistry {
fn join(self) {
self.handle.join().expect("join server");
}
}
fn spawn_registry(expected_requests: usize) -> TestRegistry {
let server = Server::http("127.0.0.1:0").expect("server");
let base_url = format!("http://{}", server.server_addr());
let handle = thread::spawn(move || {
for _ in 0..expected_requests {
let req = match server.recv_timeout(std::time::Duration::from_secs(30)) {
Ok(Some(r)) => r,
_ => break,
};
let resp = Response::from_string(r#"{"crate":{"id":"serde"}}"#)
.with_status_code(StatusCode(200))
.with_header(
Header::from_bytes("Content-Type", "application/json").expect("header"),
);
req.respond(resp).expect("respond");
}
});
TestRegistry { base_url, handle }
}
mod environment_info {
use super::*;
#[test]
fn given_clean_environment_when_running_doctor_then_reports_environment_info() {
let td = tempdir().expect("tempdir");
create_workspace(td.path());
fs::create_dir_all(td.path().join("cargo-home")).expect("mkdir");
let registry = spawn_registry(1);
let mut cmd = shipper_cmd();
cmd.arg("--manifest-path")
.arg(td.path().join("Cargo.toml"))
.arg("--api-base")
.arg(®istry.base_url)
.arg("doctor")
.env("CARGO_HOME", td.path().join("cargo-home"))
.env_remove("CARGO_REGISTRY_TOKEN")
.env_remove("CARGO_REGISTRIES_CRATES_IO_TOKEN");
cmd.assert()
.success()
.stdout(contains("Shipper Doctor - Diagnostics Report"))
.stdout(contains("workspace_root:"))
.stdout(contains("Diagnostics complete."));
registry.join();
}
}
mod cargo_detection {
use super::*;
#[test]
fn given_cargo_is_installed_when_running_doctor_then_cargo_version_is_detected() {
let td = tempdir().expect("tempdir");
create_workspace(td.path());
fs::create_dir_all(td.path().join("cargo-home")).expect("mkdir");
let registry = spawn_registry(1);
let mut cmd = shipper_cmd();
let output = cmd
.arg("--manifest-path")
.arg(td.path().join("Cargo.toml"))
.arg("--api-base")
.arg(®istry.base_url)
.arg("doctor")
.env("CARGO_HOME", td.path().join("cargo-home"))
.env_remove("CARGO_REGISTRY_TOKEN")
.env_remove("CARGO_REGISTRIES_CRATES_IO_TOKEN")
.assert()
.success()
.get_output()
.stdout
.clone();
let stdout = String::from_utf8(output).expect("utf8");
assert!(
stdout.contains("cargo: cargo"),
"Expected cargo version line in output, got:\n{stdout}"
);
registry.join();
}
}
mod missing_token {
use super::*;
#[test]
fn given_no_cargo_registry_token_when_running_doctor_then_warns_about_missing_token() {
let td = tempdir().expect("tempdir");
create_workspace(td.path());
let cargo_home = td.path().join("cargo-home");
fs::create_dir_all(&cargo_home).expect("mkdir");
let registry = spawn_registry(1);
let mut cmd = shipper_cmd();
cmd.arg("--manifest-path")
.arg(td.path().join("Cargo.toml"))
.arg("--api-base")
.arg(®istry.base_url)
.arg("doctor")
.env("CARGO_HOME", &cargo_home)
.env_remove("CARGO_REGISTRY_TOKEN")
.env_remove("CARGO_REGISTRIES_CRATES_IO_TOKEN");
cmd.assert().success().stdout(contains("NONE FOUND"));
registry.join();
}
}
mod workspace_health {
use super::*;
#[test]
fn given_valid_workspace_when_running_doctor_then_checks_workspace_health() {
let td = tempdir().expect("tempdir");
create_workspace(td.path());
fs::create_dir_all(td.path().join("cargo-home")).expect("mkdir");
let registry = spawn_registry(1);
let mut cmd = shipper_cmd();
cmd.arg("--manifest-path")
.arg(td.path().join("Cargo.toml"))
.arg("--api-base")
.arg(®istry.base_url)
.arg("doctor")
.env("CARGO_HOME", td.path().join("cargo-home"))
.env_remove("CARGO_REGISTRY_TOKEN")
.env_remove("CARGO_REGISTRIES_CRATES_IO_TOKEN");
cmd.assert()
.success()
.stdout(contains("registry:"))
.stdout(contains("state_dir:"))
.stdout(contains("registry_reachable: true"))
.stdout(contains("git:"));
registry.join();
}
#[test]
fn given_nonexistent_state_dir_when_running_doctor_then_reports_will_be_created() {
let td = tempdir().expect("tempdir");
create_workspace(td.path());
fs::create_dir_all(td.path().join("cargo-home")).expect("mkdir");
let registry = spawn_registry(1);
let mut cmd = shipper_cmd();
cmd.arg("--manifest-path")
.arg(td.path().join("Cargo.toml"))
.arg("--api-base")
.arg(®istry.base_url)
.arg("doctor")
.env("CARGO_HOME", td.path().join("cargo-home"))
.env_remove("CARGO_REGISTRY_TOKEN")
.env_remove("CARGO_REGISTRIES_CRATES_IO_TOKEN");
cmd.assert()
.success()
.stdout(contains("state_dir_exists: false (will be created)"));
registry.join();
}
}