attune-system/attune
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

formatting

Browse Source
This commit is contained in:
David Culbreth
2026-03-04 22:42:23 -06:00
parent 67a1c02543
commit 95765f50a8
30 changed files with 5172 additions and 336 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
crates/worker/src/runtime/local.rs
View File

@@ -177,7 +177,7 @@ mod tests {
parameter_delivery: ParameterDelivery::default(),
parameter_format: ParameterFormat::default(),
output_format: OutputFormat::default(),
cancel_token: None,
cancel_token: None,
};
assert!(runtime.can_execute(&context));
@@ -210,7 +210,7 @@ mod tests {
parameter_delivery: ParameterDelivery::default(),
parameter_format: ParameterFormat::default(),
output_format: OutputFormat::default(),
cancel_token: None,
cancel_token: None,
};
assert!(!runtime.can_execute(&context));

16
crates/worker/src/runtime/process.rs
View File

@@ -906,7 +906,7 @@ mod tests {
parameter_delivery: ParameterDelivery::default(),
parameter_format: ParameterFormat::default(),
output_format: OutputFormat::default(),
cancel_token: None,
cancel_token: None,
};
assert!(runtime.can_execute(&context));
@@ -941,7 +941,7 @@ mod tests {
parameter_delivery: ParameterDelivery::default(),
parameter_format: ParameterFormat::default(),
output_format: OutputFormat::default(),
cancel_token: None,
cancel_token: None,
};
assert!(runtime.can_execute(&context));
@@ -976,7 +976,7 @@ mod tests {
parameter_delivery: ParameterDelivery::default(),
parameter_format: ParameterFormat::default(),
output_format: OutputFormat::default(),
cancel_token: None,
cancel_token: None,
};
assert!(!runtime.can_execute(&context));
@@ -1067,7 +1067,7 @@ mod tests {
parameter_delivery: ParameterDelivery::default(),
parameter_format: ParameterFormat::default(),
output_format: OutputFormat::default(),
cancel_token: None,
cancel_token: None,
};
let result = runtime.execute(context).await.unwrap();
@@ -1125,7 +1125,7 @@ mod tests {
parameter_delivery: ParameterDelivery::default(),
parameter_format: ParameterFormat::default(),
output_format: OutputFormat::default(),
cancel_token: None,
cancel_token: None,
};
let result = runtime.execute(context).await.unwrap();
@@ -1164,7 +1164,7 @@ mod tests {
parameter_delivery: ParameterDelivery::default(),
parameter_format: ParameterFormat::default(),
output_format: OutputFormat::default(),
cancel_token: None,
cancel_token: None,
};
let result = runtime.execute(context).await.unwrap();
@@ -1215,7 +1215,7 @@ mod tests {
parameter_delivery: ParameterDelivery::default(),
parameter_format: ParameterFormat::default(),
output_format: OutputFormat::default(),
cancel_token: None,
cancel_token: None,
};
let result = runtime.execute(context).await.unwrap();
@@ -1324,7 +1324,7 @@ mod tests {
parameter_delivery: ParameterDelivery::default(),
parameter_format: ParameterFormat::default(),
output_format: OutputFormat::default(),
cancel_token: None,
cancel_token: None,
};
let result = runtime.execute(context).await.unwrap();

14
crates/worker/src/runtime/shell.rs
View File

@@ -623,7 +623,7 @@ mod tests {
parameter_delivery: attune_common::models::ParameterDelivery::default(),
parameter_format: attune_common::models::ParameterFormat::default(),
output_format: attune_common::models::OutputFormat::default(),
cancel_token: None,
cancel_token: None,
};
let result = runtime.execute(context).await.unwrap();
@@ -660,7 +660,7 @@ mod tests {
parameter_delivery: attune_common::models::ParameterDelivery::default(),
parameter_format: attune_common::models::ParameterFormat::default(),
output_format: attune_common::models::OutputFormat::default(),
cancel_token: None,
cancel_token: None,
};
let result = runtime.execute(context).await.unwrap();
@@ -692,7 +692,7 @@ mod tests {
parameter_delivery: attune_common::models::ParameterDelivery::default(),
parameter_format: attune_common::models::ParameterFormat::default(),
output_format: attune_common::models::OutputFormat::default(),
cancel_token: None,
cancel_token: None,
};
let result = runtime.execute(context).await.unwrap();
@@ -726,7 +726,7 @@ mod tests {
parameter_delivery: attune_common::models::ParameterDelivery::default(),
parameter_format: attune_common::models::ParameterFormat::default(),
output_format: attune_common::models::OutputFormat::default(),
cancel_token: None,
cancel_token: None,
};
let result = runtime.execute(context).await.unwrap();
@@ -775,7 +775,7 @@ echo "missing=$missing"
parameter_delivery: attune_common::models::ParameterDelivery::default(),
parameter_format: attune_common::models::ParameterFormat::default(),
output_format: attune_common::models::OutputFormat::default(),
cancel_token: None,
cancel_token: None,
};
let result = runtime.execute(context).await.unwrap();
@@ -819,7 +819,7 @@ echo '{"id": 3, "name": "Charlie"}'
parameter_delivery: attune_common::models::ParameterDelivery::default(),
parameter_format: attune_common::models::ParameterFormat::default(),
output_format: attune_common::models::OutputFormat::Jsonl,
cancel_token: None,
cancel_token: None,
};
let result = runtime.execute(context).await.unwrap();
@@ -942,7 +942,7 @@ echo '{"result": "success", "count": 42}'
parameter_delivery: attune_common::models::ParameterDelivery::default(),
parameter_format: attune_common::models::ParameterFormat::default(),
output_format: attune_common::models::OutputFormat::Json,
cancel_token: None,
cancel_token: None,
};
let result = runtime.execute(context).await.unwrap();

14
crates/worker/tests/log_truncation_test.rs
View File

@@ -21,7 +21,12 @@ fn make_python_process_runtime(packs_base_dir: PathBuf) -> ProcessRuntime {
dependencies: None,
env_vars: std::collections::HashMap::new(),
};
ProcessRuntime::new("python".to_string(), config, packs_base_dir.clone(), packs_base_dir.join("../runtime_envs"))
ProcessRuntime::new(
"python".to_string(),
config,
packs_base_dir.clone(),
packs_base_dir.join("../runtime_envs"),
)
}
fn make_python_context(
@@ -269,7 +274,12 @@ async fn test_shell_process_runtime_truncation() {
dependencies: None,
env_vars: std::collections::HashMap::new(),
};
let runtime = ProcessRuntime::new("shell".to_string(), config, tmp.path().to_path_buf(), tmp.path().join("runtime_envs"));
let runtime = ProcessRuntime::new(
"shell".to_string(),
config,
tmp.path().to_path_buf(),
tmp.path().join("runtime_envs"),
);
let context = ExecutionContext {
execution_id: 8,

26
crates/worker/tests/security_tests.rs
View File

@@ -22,8 +22,16 @@ fn make_python_process_runtime(packs_base_dir: PathBuf) -> ProcessRuntime {
dependencies: None,
env_vars: std::collections::HashMap::new(),
};
let runtime_envs_dir = packs_base_dir.parent().unwrap_or(&packs_base_dir).join("runtime_envs");
ProcessRuntime::new("python".to_string(), config, packs_base_dir, runtime_envs_dir)
let runtime_envs_dir = packs_base_dir
.parent()
.unwrap_or(&packs_base_dir)
.join("runtime_envs");
ProcessRuntime::new(
"python".to_string(),
config,
packs_base_dir,
runtime_envs_dir,
)
}
#[tokio::test]
@@ -436,7 +444,12 @@ echo "PASS: No secrets in environment"
dependencies: None,
env_vars: std::collections::HashMap::new(),
};
let runtime = ProcessRuntime::new("shell".to_string(), config, tmp.path().to_path_buf(), tmp.path().join("runtime_envs"));
let runtime = ProcessRuntime::new(
"shell".to_string(),
config,
tmp.path().to_path_buf(),
tmp.path().join("runtime_envs"),
);
let context = ExecutionContext {
execution_id: 7,
@@ -508,7 +521,12 @@ print(json.dumps({"leaked": leaked}))
dependencies: None,
env_vars: std::collections::HashMap::new(),
};
let runtime = ProcessRuntime::new("python".to_string(), config, tmp.path().to_path_buf(), tmp.path().join("runtime_envs"));
let runtime = ProcessRuntime::new(
"python".to_string(),
config,
tmp.path().to_path_buf(),
tmp.path().join("runtime_envs"),
);
let context = ExecutionContext {
execution_id: 8,