github/kata-containers
1
0
Fork 1
mirror of https://github.com/kata-containers/kata-containers.git synced 2025-07-17 17:02:42 +00:00
Code Issues Projects Releases Wiki Activity

genpolicy: refactor policy test workdir setup

Browse Source 
This aligns the workdir preparation more closely with the workdir
preparation for the generate integration test. Most notably, we
clean up the temporary directory before we execute the tests in it.
This way we better isolate different runs.

Signed-off-by: Leonard Cohnen <leonard.cohnen@gmail.com>
Signed-off-by: Markus Rudy <mr@edgeless.systems>
This commit is contained in:
Leonard Cohnen 2025-03-12 20:11:22 +01:00 committed by Markus Rudy
parent bad0cd0003
commit b23ff6fc68
1 changed files with 46 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
src/tools/genpolicy/tests/policy/main.rs
View File

@ -5,6 +5,7 @@
#[cfg(test)]
mod tests {
use anyhow::Context;
use base64::prelude::*;
use std::fmt::{self, Display};
use std::fs::{self, File};
@ -63,23 +64,7 @@ mod tests {
/// with the tag `type` listing the exact type of request.
async fn runtests(test_case_dir: &str) {
// Prepare temp dir for running genpolicy.
let workdir = path::PathBuf::from(env!("CARGO_TARGET_TMPDIR")).join(test_case_dir);
fs::create_dir_all(&workdir)
.expect("should be able to create directories under CARGO_TARGET_TMPDIR");
let testdata_dir = path::PathBuf::from(env!("CARGO_MANIFEST_DIR"))
.join("tests/policy/testdata")
.join(test_case_dir);
fs::copy(testdata_dir.join("pod.yaml"), workdir.join("pod.yaml"))
.expect("copying files around should not fail");
let genpolicy_dir =
path::PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("../../tools/genpolicy");
for base in ["rules.rego", "genpolicy-settings.json"] {
fs::copy(genpolicy_dir.join(base), workdir.join(base))
.expect("copying files around should not fail");
}
let (workdir, testdata_dir) = prepare_workdir(test_case_dir, &["pod.yaml"]);
// Run the command and return the generated policy.
@ -153,6 +138,50 @@ mod tests {
}
}
fn prepare_workdir(
test_case_dir: &str,
files_to_copy: &[&str],
) -> (path::PathBuf, path::PathBuf) {
// Prepare temp dir for running genpolicy.
let workdir = path::PathBuf::from(env!("CARGO_TARGET_TMPDIR")).join(test_case_dir);
fs::create_dir_all(&workdir)
.expect("should be able to create directories under CARGO_TARGET_TMPDIR");
let testdata_dir = path::PathBuf::from(env!("CARGO_MANIFEST_DIR"))
.join("tests/policy/testdata")
.join(test_case_dir);
// Make sure that workdir is empty.
for entry in fs::read_dir(&workdir).expect("should be able to read directories") {
let entry = entry.expect("should be able to read directory entries");
fs::remove_file(entry.path()).expect("should be able to remove files");
}
for file in files_to_copy {
fs::copy(testdata_dir.join(file), workdir.join(file))
.context(format!(
"{:?} --> {:?}",
testdata_dir.join(file),
workdir.join(file)
))
.expect("copying files around should not fail");
}
let genpolicy_dir = path::PathBuf::from(env!("CARGO_MANIFEST_DIR"));
for base in ["rules.rego", "genpolicy-settings.json"] {
fs::copy(genpolicy_dir.join(base), workdir.join(base))
.context(format!(
"{:?} --> {:?}",
genpolicy_dir.join(base),
workdir.join(base)
))
.expect("copying files around should not fail");
}
(workdir, testdata_dir)
}
#[tokio::test]
async fn test_copyfile() {
runtests("copyfile").await;