remove unused test

This commit is contained in:
2026-07-13 17:13:43 +08:00
parent 3f215d352a
commit 97a1e1663e
2 changed files with 7 additions and 85 deletions
@@ -1,14 +1,14 @@
use crate::editor_agent::utils::{ImageId, ImageMetadata}; use crate::editor_agent::utils::{ImageId, ImageMetadata};
use crate::editor_generation_config::EditorGenerationPricingConfig; use crate::editor_generation_config::EditorGenerationPricingConfig;
use crate::http_error::AppError;
use crate::openai_image_generation::GPT_IMAGE_2_MODEL; use crate::openai_image_generation::GPT_IMAGE_2_MODEL;
use axum::http::StatusCode;
use module_editor_agent::agent::tool::Tool; use module_editor_agent::agent::tool::Tool;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use std::collections::HashMap;
use axum::http::StatusCode;
use serde_json::json; use serde_json::json;
use crate::http_error::AppError; use std::collections::HashMap;
#[derive(Debug, Clone, Serialize, Deserialize)] #[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct EditorToolContext { pub struct EditorToolContext {
pub images: HashMap<ImageId, ImageMetadata>, pub images: HashMap<ImageId, ImageMetadata>,
} }
@@ -53,4 +53,4 @@ pub fn image_not_found(image_id: &ImageId) -> AppError {
"provider": "editor-agent", "provider": "editor-agent",
"message": format!("asset {image_id} not found in context"), "message": format!("asset {image_id} not found in context"),
})) }))
} }
@@ -1,20 +1,11 @@
use crate::editor_agent::editor_tools::common::{ use crate::editor_agent::editor_tools::common::{
EditorAgentPricedTool, EditorToolContext, editor_agent_image_mud_points, editor_agent_image_mud_points, EditorAgentPricedTool, EditorToolContext,
}; };
use crate::editor_agent::utils::ImageId; use crate::editor_agent::utils::ImageId;
use crate::editor_generation_config::EditorGenerationPricingConfig; use crate::editor_generation_config::EditorGenerationPricingConfig;
use crate::editor_project::{
EditorGenerationCaller, EditorImageGenerationRequest, generate_editor_image_for_owner,
};
use crate::http_error::AppError;
use crate::request_context::RequestContext;
use crate::state::AppState;
use axum::http::StatusCode;
use module_editor_agent::agent::tool::{Tool, ToolFailure, ToolFailureKind}; use module_editor_agent::agent::tool::{Tool, ToolFailure, ToolFailureKind};
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use serde_json::{Value, json}; use serde_json::{json, Value};
use shared_contracts::api::ApiSuccessEnvelope;
use shared_contracts::assets::EditorCanvasGenerationCompletionPayload;
use std::error::Error; use std::error::Error;
use std::fmt::Display; use std::fmt::Display;
@@ -181,72 +172,3 @@ impl GenerateImageTool {
) )
} }
} }
#[cfg(test)]
mod tests {
use super::*;
use std::collections::HashMap;
fn tool() -> GenerateImageTool {
let image_id = ImageId::from_data_key("reference.png");
GenerateImageTool {
context: EditorToolContext {
images: HashMap::from([(
image_id,
crate::editor_agent::utils::ImageMetadata {
tag: String::new(),
data_key: "reference.png".to_string(),
},
)]),
},
}
}
#[tokio::test]
async fn proposal_accepts_prompt_and_known_references() {
let output = tool()
.call(GenerateImageToolArgs {
prompt: "雨夜的霓虹街道".to_string(),
reference_image_ids: vec![ImageId::from_data_key("reference.png")],
aspect_ratio: Some("16:9".to_string()),
image_size: Some("2K".to_string()),
})
.await
.expect("proposal should be valid");
assert!(output.message.contains("pending user confirmation"));
}
#[tokio::test]
async fn proposal_rejects_missing_prompt_and_unknown_reference() {
let prompt_error = tool()
.call(GenerateImageToolArgs {
prompt: " ".to_string(),
reference_image_ids: Vec::new(),
aspect_ratio: None,
image_size: None,
})
.await
.expect_err("blank prompt must fail");
assert!(matches!(
prompt_error,
GenerateImageError::PromptNotProvided
));
let reference_error = tool()
.call(GenerateImageToolArgs {
prompt: "一棵树".to_string(),
reference_image_ids: vec![ImageId {
id: "missing.png".to_string(),
}],
aspect_ratio: None,
image_size: None,
})
.await
.expect_err("unknown reference must fail");
assert!(matches!(
reference_error,
GenerateImageError::AssetNotFound(_)
));
}
}