use super::*;use codex_tools::JsonSchema;use pretty_assertions::assert_eq;use std::collections::BTreeMap;fn described_object(description: &str) -> JsonSchema { let mut schema = JsonSchema::object( BTreeMap::new(), /*required*/ None, /*additional_properties*/ None, ); schema.description = Some(description.to_string()); schema}#[test]fn spawn_agents_on_csv_tool_requires_csv_and_instruction() { assert_eq!( create_spawn_agents_on_csv_tool(), ToolSpec::Function(ResponsesApiTool { name: "spawn_agents_on_csv".to_string(), description: "Process a CSV by spawning one worker sub-agent per row. The instruction string is a template where `{column}` placeholders are replaced with row values. Each worker must call `report_agent_job_result` with a JSON object (matching `output_schema` when provided); missing reports are treated as failures. This call blocks until all rows finish and automatically exports results to `output_csv_path` (or a default path)." .to_string(), strict: false, defer_loading: None, parameters: JsonSchema::object(BTreeMap::from([ ( "csv_path".to_string(), JsonSchema::string(Some( "Path to the CSV file containing input rows.".to_string(), )), ), ( "instruction".to_string(), JsonSchema::string(Some( "Instruction template to apply to each CSV row. Use {column_name} placeholders to inject values from the row." .to_string(), )), ), ( "id_column".to_string(), JsonSchema::string(Some( "CSV column to use as stable item id. Omit to use row numbers." .to_string(), )), ), ( "output_csv_path".to_string(), JsonSchema::string(Some( "Output CSV path for exported results. Omit to create one next to the input CSV." .to_string(), )), ), ( "max_concurrency".to_string(), JsonSchema::number(Some( "Maximum concurrent workers for this job. Defaults to 16 and is capped by config." .to_string(), )), ), ( "max_workers".to_string(), JsonSchema::number(Some( "Alias for max_concurrency. Defaults to 16 and is capped by config." .to_string(), )), ), ( "max_runtime_seconds".to_string(), JsonSchema::number(Some( "Maximum runtime per worker before failure. Defaults to 1800 seconds; config may set a different default." .to_string(), )), ), ( "output_schema".to_string(), described_object( "JSON Schema for each worker result. Omit to accept any result object.", ), ), ]), Some(vec!["csv_path".to_string(), "instruction".to_string()]), Some(false.into())), output_schema: None, }) );}#[test]fn report_agent_job_result_tool_requires_result_payload() { assert_eq!( create_report_agent_job_result_tool(), ToolSpec::Function(ResponsesApiTool { name: "report_agent_job_result".to_string(), description: "Worker-only tool to report a result for an agent job item. Main agents should not call this." .to_string(), strict: false, defer_loading: None, parameters: JsonSchema::object(BTreeMap::from([ ( "job_id".to_string(), JsonSchema::string(Some("Identifier of the job.".to_string())), ), ( "item_id".to_string(), JsonSchema::string(Some("Identifier of the job item.".to_string())), ), ( "result".to_string(), described_object("Result object for this job item."), ), ( "stop".to_string(), JsonSchema::boolean(Some( "True cancels remaining job items after this result is recorded; false or omitted continues the job." .to_string(), )), ), ]), Some(vec![ "job_id".to_string(), "item_id".to_string(), "result".to_string(), ]), Some(false.into())), output_schema: None, }) );}