mod common;
use common::{output_to_string, test_dir_file, test_dir_with_subfolder, wash};
use std::fs::{remove_dir_all, File};
use std::io::prelude::*;
const LIB: &str = "wasmcloudcache";
const OCI: &str = "wasmcloud_ocicache";
#[test]
fn integration_drain_comprehensive() {
integration_drain_lib();
integration_drain_oci();
integration_drain_all();
}
fn integration_drain_lib() {
let test_dir = test_dir_with_subfolder("drain_lib");
let lib_subdir = &format!("drain_lib/{}", LIB);
let lib_dir = test_dir_with_subfolder(lib_subdir);
let _nested_dir = test_dir_with_subfolder(&format!("{}/a/b/c/d/e", lib_subdir));
let wasm = test_dir_file(lib_subdir, "hello.wasm");
let mut wasm_file = File::create(wasm).unwrap();
wasm_file.write_all(b"bytes_or_something_idk").unwrap();
let provider = test_dir_file(lib_subdir, "world.par.gz");
let mut provider_file = File::create(provider).unwrap();
provider_file.write_all(b"parcheesi").unwrap();
std::env::set_var("TMPDIR", test_dir.clone());
std::env::set_var("TMP", test_dir.clone());
let drain_basic = wash()
.args(&["drain", "lib", "-o", "json"])
.output()
.unwrap_or_else(|_| panic!("failed to drain {:?}", lib_dir.clone()));
assert!(drain_basic.status.success());
assert_eq!(
output_to_string(drain_basic),
format!("{{\"drained\":[\"{}\"]}}\n", lib_dir.to_str().unwrap())
);
assert!(lib_dir.read_dir().unwrap().next().is_none());
remove_dir_all(test_dir).unwrap();
}
fn integration_drain_oci() {
let test_dir = test_dir_with_subfolder("drain_oci");
let oci_subdir = &format!("drain_oci/{}", OCI);
let oci_dir = test_dir_with_subfolder(oci_subdir);
let _nested_dir = test_dir_with_subfolder(&format!("{}/a/b/c/d/e", oci_subdir));
let wasm = test_dir_file(oci_subdir, "hello.wasm");
let mut wasm_file = File::create(wasm).unwrap();
wasm_file.write_all(b"bytes_or_something_idk").unwrap();
let provider = test_dir_file(oci_subdir, "world.par.gz");
let mut provider_file = File::create(provider).unwrap();
provider_file.write_all(b"parcheesi").unwrap();
std::env::set_var("TMPDIR", test_dir.clone());
std::env::set_var("TMP", test_dir.clone());
let drain_basic = wash()
.args(&["drain", "oci", "-o", "json"])
.output()
.unwrap_or_else(|_| panic!("failed to drain {:?}", oci_dir.clone()));
assert!(drain_basic.status.success());
assert_eq!(
output_to_string(drain_basic),
format!("{{\"drained\":[\"{}\"]}}\n", oci_dir.to_str().unwrap())
);
assert!(oci_dir.read_dir().unwrap().next().is_none());
remove_dir_all(test_dir).unwrap();
}
fn integration_drain_all() {
let test_dir = test_dir_with_subfolder("drain_all");
let oci_subdir = &format!("drain_all/{}", OCI);
let oci_dir = test_dir_with_subfolder(oci_subdir);
let lib_subdir = &format!("drain_all/{}", LIB);
let lib_dir = test_dir_with_subfolder(lib_subdir);
let _nested_dir = test_dir_with_subfolder(&format!("{}/a/b/c/d/e", oci_subdir));
let _nested_dir = test_dir_with_subfolder(&format!("{}/a/b/c/d/e", lib_subdir));
let wasm = test_dir_file(oci_subdir, "hello.wasm");
let mut wasm_file = File::create(wasm).unwrap();
wasm_file.write_all(b"bytes_or_something_idk").unwrap();
let provider = test_dir_file(lib_subdir, "world.par.gz");
let mut provider_file = File::create(provider).unwrap();
provider_file.write_all(b"parcheesi").unwrap();
std::env::set_var("TMPDIR", test_dir.clone());
std::env::set_var("TMP", test_dir.clone());
let drain_basic = wash()
.args(&["drain", "all", "-o", "json"])
.output()
.unwrap_or_else(|_| panic!("failed to drain {:?}", oci_dir.clone()));
assert!(drain_basic.status.success());
assert_eq!(
output_to_string(drain_basic),
format!(
"{{\"drained\":[\"{}\",\"{}\"]}}\n",
lib_dir.to_str().unwrap(),
oci_dir.to_str().unwrap()
)
);
assert!(lib_dir.read_dir().unwrap().next().is_none());
assert!(oci_dir.read_dir().unwrap().next().is_none());
remove_dir_all(test_dir).unwrap();
}