fabro/lib/crates/fabro-workflow/tests/it/cp_integration.rs
Bryan Helmkamp 72924ba611
Merge remote-tracking branch 'origin/main'
# Conflicts:
#	lib/crates/fabro-agent/src/cli.rs
#	lib/crates/fabro-cli/src/commands/run/run_progress/mod.rs
#	lib/crates/fabro-cli/tests/it/cmd/exec.rs
#	lib/crates/fabro-cli/tests/it/scenario/lifecycle.rs
#	lib/crates/fabro-telemetry/src/spawn.rs
#	lib/crates/fabro-workflow/tests/it/attractor_compat.rs
#	lib/crates/fabro-workflow/tests/it/cp_integration.rs
2026-04-19 20:32:02 -04:00

256 lines
8.3 KiB
Rust

//! E2E tests for `fabro cp` against local and Docker sandbox backends.
//!
//! Local tests run without `#[ignore]` (no external dependencies).
//! Docker tests require a Docker daemon and are marked `#[ignore]`.
//! Run Docker tests with: `cargo test --package arc-workflows --test
//! cp_integration -- --ignored`
#![allow(
clippy::ignore_without_reason,
reason = "This integration module intentionally uses concise ignored-test markers."
)]
#![expect(
clippy::disallowed_methods,
reason = "This integration test stages sandbox fixtures with sync std::fs."
)]
use fabro_sandbox::SandboxRecord;
use fabro_sandbox::reconnect::reconnect;
// ---------------------------------------------------------------------------
// Local sandbox
// ---------------------------------------------------------------------------
fn local_record(working_directory: &std::path::Path) -> SandboxRecord {
SandboxRecord {
provider: "local".to_string(),
working_directory: working_directory.to_string_lossy().to_string(),
identifier: None,
host_working_directory: None,
container_mount_point: None,
}
}
#[tokio::test]
async fn local_cp_upload_download_round_trip() {
let sandbox_dir = tempfile::tempdir().unwrap();
let scratch = tempfile::tempdir().unwrap();
let record = local_record(sandbox_dir.path());
let sandbox = reconnect(&record, None).await.expect("reconnect local");
// Upload a text file
let content = b"hello from local cp test\n";
let local_src = scratch.path().join("upload.txt");
std::fs::write(&local_src, content).unwrap();
sandbox
.upload_file_from_local(&local_src, "cp_test.txt")
.await
.expect("upload text");
// Verify it landed in the sandbox working directory
assert!(sandbox_dir.path().join("cp_test.txt").exists());
// Download it back
let local_dst = scratch.path().join("download.txt");
sandbox
.download_file_to_local("cp_test.txt", &local_dst)
.await
.expect("download text");
assert_eq!(std::fs::read(&local_dst).unwrap(), content);
}
#[tokio::test]
async fn local_cp_binary_round_trip() {
let sandbox_dir = tempfile::tempdir().unwrap();
let scratch = tempfile::tempdir().unwrap();
let record = local_record(sandbox_dir.path());
let sandbox = reconnect(&record, None).await.expect("reconnect local");
// All 256 byte values
let binary: Vec<u8> = (0..=255).collect();
let local_src = scratch.path().join("binary.bin");
std::fs::write(&local_src, &binary).unwrap();
sandbox
.upload_file_from_local(&local_src, "binary.bin")
.await
.expect("upload binary");
let local_dst = scratch.path().join("binary_dl.bin");
sandbox
.download_file_to_local("binary.bin", &local_dst)
.await
.expect("download binary");
assert_eq!(std::fs::read(&local_dst).unwrap(), binary);
}
#[tokio::test]
async fn local_cp_creates_parent_dirs() {
let sandbox_dir = tempfile::tempdir().unwrap();
let scratch = tempfile::tempdir().unwrap();
let record = local_record(sandbox_dir.path());
let sandbox = reconnect(&record, None).await.expect("reconnect local");
let content = b"nested file\n";
let local_src = scratch.path().join("nested.txt");
std::fs::write(&local_src, content).unwrap();
// Upload to a nested path that doesn't exist yet
sandbox
.upload_file_from_local(&local_src, "a/b/c/nested.txt")
.await
.expect("upload to nested path");
assert!(sandbox_dir.path().join("a/b/c/nested.txt").exists());
// Download to a nested local path that doesn't exist yet
let local_dst = scratch.path().join("x/y/z/nested.txt");
sandbox
.download_file_to_local("a/b/c/nested.txt", &local_dst)
.await
.expect("download to nested path");
assert_eq!(std::fs::read(&local_dst).unwrap(), content);
}
// ---------------------------------------------------------------------------
// Docker sandbox
// ---------------------------------------------------------------------------
fn docker_record(host_dir: &std::path::Path, mount_point: &str) -> SandboxRecord {
SandboxRecord {
provider: "docker".to_string(),
working_directory: mount_point.to_string(),
identifier: None,
host_working_directory: Some(host_dir.to_string_lossy().to_string()),
container_mount_point: Some(mount_point.to_string()),
}
}
#[tokio::test]
#[ignore] // requires Docker daemon
async fn docker_cp_upload_download_round_trip() {
let host_dir = tempfile::tempdir().unwrap();
let scratch = tempfile::tempdir().unwrap();
let record = docker_record(host_dir.path(), "/workspace");
let sandbox = reconnect(&record, None).await.expect("reconnect docker");
// Upload a text file
let content = b"hello from docker cp test\n";
let local_src = scratch.path().join("upload.txt");
std::fs::write(&local_src, content).unwrap();
sandbox
.upload_file_from_local(&local_src, "cp_test.txt")
.await
.expect("upload text");
// Verify it landed on the host filesystem (bind mount path)
assert!(host_dir.path().join("cp_test.txt").exists());
assert_eq!(
std::fs::read(host_dir.path().join("cp_test.txt")).unwrap(),
content
);
// Download it back
let local_dst = scratch.path().join("download.txt");
sandbox
.download_file_to_local("cp_test.txt", &local_dst)
.await
.expect("download text");
assert_eq!(std::fs::read(&local_dst).unwrap(), content);
}
#[tokio::test]
#[ignore] // requires Docker daemon
async fn docker_cp_binary_round_trip() {
let host_dir = tempfile::tempdir().unwrap();
let scratch = tempfile::tempdir().unwrap();
let record = docker_record(host_dir.path(), "/workspace");
let sandbox = reconnect(&record, None).await.expect("reconnect docker");
let binary: Vec<u8> = (0..=255).collect();
let local_src = scratch.path().join("binary.bin");
std::fs::write(&local_src, &binary).unwrap();
sandbox
.upload_file_from_local(&local_src, "binary.bin")
.await
.expect("upload binary");
let local_dst = scratch.path().join("binary_dl.bin");
sandbox
.download_file_to_local("binary.bin", &local_dst)
.await
.expect("download binary");
assert_eq!(std::fs::read(&local_dst).unwrap(), binary);
}
#[tokio::test]
#[ignore] // requires Docker daemon
async fn docker_cp_creates_parent_dirs() {
let host_dir = tempfile::tempdir().unwrap();
let scratch = tempfile::tempdir().unwrap();
let record = docker_record(host_dir.path(), "/workspace");
let sandbox = reconnect(&record, None).await.expect("reconnect docker");
let content = b"nested docker file\n";
let local_src = scratch.path().join("nested.txt");
std::fs::write(&local_src, content).unwrap();
sandbox
.upload_file_from_local(&local_src, "deep/nested/file.txt")
.await
.expect("upload to nested path");
assert!(host_dir.path().join("deep/nested/file.txt").exists());
let local_dst = scratch.path().join("p/q/file.txt");
sandbox
.download_file_to_local("deep/nested/file.txt", &local_dst)
.await
.expect("download to nested path");
assert_eq!(std::fs::read(&local_dst).unwrap(), content);
}
#[tokio::test]
#[ignore] // requires Docker daemon
async fn docker_cp_custom_mount_point() {
let host_dir = tempfile::tempdir().unwrap();
let scratch = tempfile::tempdir().unwrap();
// Use a non-default mount point
let record = docker_record(host_dir.path(), "/app");
let sandbox = reconnect(&record, None).await.expect("reconnect docker");
let content = b"custom mount\n";
let local_src = scratch.path().join("mount.txt");
std::fs::write(&local_src, content).unwrap();
sandbox
.upload_file_from_local(&local_src, "mount.txt")
.await
.expect("upload with custom mount");
assert!(host_dir.path().join("mount.txt").exists());
let local_dst = scratch.path().join("mount_dl.txt");
sandbox
.download_file_to_local("mount.txt", &local_dst)
.await
.expect("download with custom mount");
assert_eq!(std::fs::read(&local_dst).unwrap(), content);
}