Files
claw/tests/scene_generator_test.rs
木炎 67fe17302e feat: add SceneKind::Monitoring and scene_kind_hint param to analyzer
- Add SceneKind::Monitoring enum variant with from_str/as_str helpers
- Add analyze_scene_source_with_hint function accepting optional scene kind hint
- User hint takes priority over meta tag, defaults to ReportCollection
- ReportCollection requires target_url, expected_domain, entry_script
- Monitoring type has optional fields
- Add test cases for hint parameter behavior
- Update non_report fixture with required meta tags for ReportCollection

🤖 Generated with [Qoder][https://qoder.com]
2026-04-16 23:33:24 +08:00

143 lines
5.1 KiB
Rust

use std::fs;
use std::path::{Path, PathBuf};
use std::time::{SystemTime, UNIX_EPOCH};
use sgclaw::compat::scene_platform::registry::load_scene_registry;
use sgclaw::generated_scene::analyzer::{analyze_scene_source, analyze_scene_source_with_hint, SceneKind, ToolKind};
use sgclaw::generated_scene::generator::{generate_scene_package, GenerateSceneRequest};
#[test]
fn analyzer_classifies_supported_report_collection_source() {
let analysis = analyze_scene_source(Path::new(
"tests/fixtures/generated_scene/report_collection",
))
.unwrap();
assert_eq!(analysis.scene_kind, SceneKind::ReportCollection);
assert_eq!(analysis.tool_kind, ToolKind::BrowserScript);
assert_eq!(
analysis.bootstrap.target_url.as_deref(),
Some("http://20.76.57.61:18080/gsllys/tqLinelossStatis/tqQualifyRateMonitor")
);
assert_eq!(
analysis.bootstrap.expected_domain.as_deref(),
Some("20.76.57.61")
);
assert_eq!(
analysis.collection_entry_script.as_deref(),
Some("js/report.js")
);
}
#[test]
fn generator_writes_registration_ready_package_with_scene_toml() {
let output_root = temp_workspace("sgclaw-scene-generator");
generate_scene_package(GenerateSceneRequest {
source_dir: PathBuf::from("tests/fixtures/generated_scene/report_collection"),
scene_id: "sample-report-scene".to_string(),
scene_name: "示例报表场景".to_string(),
output_root: output_root.clone(),
lessons_path: PathBuf::from("docs/superpowers/references/tq-lineloss-lessons-learned.toml"),
})
.unwrap();
let skill_root = output_root.join("skills/sample-report-scene");
assert!(skill_root.join("SKILL.toml").exists());
assert!(skill_root.join("SKILL.md").exists());
assert!(skill_root.join("scene.toml").exists());
assert!(skill_root
.join("scripts/collect_sample_report_scene.js")
.exists());
assert!(skill_root
.join("scripts/collect_sample_report_scene.test.js")
.exists());
assert!(skill_root.join("references/generation-lessons.md").exists());
assert!(skill_root.join("references/org-dictionary.json").exists());
let generated_script =
fs::read_to_string(skill_root.join("scripts/collect_sample_report_scene.js")).unwrap();
assert!(generated_script.contains("return buildBrowserEntrypointResult(args);"));
let generated_manifest = fs::read_to_string(skill_root.join("scene.toml")).unwrap();
assert!(generated_manifest.contains("resolver = \"dictionary_entity\""));
assert!(generated_manifest.contains("dictionary_ref = \"references/org-dictionary.json\""));
assert!(generated_manifest.contains("required = true"));
let registry = load_scene_registry(&output_root.join("skills")).unwrap();
let entry = registry
.iter()
.find(|entry| entry.manifest.scene.id == "sample-report-scene")
.expect("generated package should be registration-ready");
assert_eq!(entry.manifest.scene.kind, "browser_script");
assert_eq!(entry.manifest.scene.category, "report_collection");
assert_eq!(entry.manifest.scene.tool, "collect_sample_report_scene");
assert_eq!(entry.manifest.bootstrap.expected_domain, "20.76.57.61");
}
#[test]
fn generator_rejects_non_report_source_with_explicit_reason() {
let err =
analyze_scene_source(Path::new("tests/fixtures/generated_scene/non_report")).unwrap_err();
assert!(err
.to_string()
.contains("report/collection browser_script only"));
}
fn temp_workspace(prefix: &str) -> PathBuf {
let nanos = SystemTime::now()
.duration_since(UNIX_EPOCH)
.unwrap()
.as_nanos();
let path = std::env::temp_dir().join(format!("{prefix}-{nanos}"));
fs::create_dir_all(&path).unwrap();
path
}
#[test]
fn analyzer_accepts_missing_meta_with_scene_kind_hint() {
// non_report fixture has no scene-kind meta tag
let analysis = analyze_scene_source_with_hint(
Path::new("tests/fixtures/generated_scene/non_report"),
Some(SceneKind::ReportCollection),
)
.unwrap();
// should succeed, using hint parameter as type
assert_eq!(analysis.scene_kind, SceneKind::ReportCollection);
}
#[test]
fn analyzer_uses_hint_when_meta_missing() {
let analysis = analyze_scene_source_with_hint(
Path::new("tests/fixtures/generated_scene/non_report"),
Some(SceneKind::Monitoring),
)
.unwrap();
assert_eq!(analysis.scene_kind, SceneKind::Monitoring);
}
#[test]
fn analyzer_uses_meta_when_present_and_no_hint() {
// report_collection fixture has correct meta tag
let analysis = analyze_scene_source_with_hint(
Path::new("tests/fixtures/generated_scene/report_collection"),
None,
)
.unwrap();
assert_eq!(analysis.scene_kind, SceneKind::ReportCollection);
}
#[test]
fn analyzer_hint_overrides_meta() {
// user choice takes priority over meta tag
let analysis = analyze_scene_source_with_hint(
Path::new("tests/fixtures/generated_scene/report_collection"),
Some(SceneKind::Monitoring),
)
.unwrap();
assert_eq!(analysis.scene_kind, SceneKind::Monitoring);
}