//! Snapshot test pinning the `Vec` JSON for a //! representative C code `CanonicalDocument`. //! //! This is an integration test. `kebab-parse-code` is intentionally NOT //! a dev-dep (design §6.3 / §8 boundary: AST extraction is parser-side). //! The `CanonicalDocument` is built inline from hand-crafted `Block::Code` //! units, which is the same pattern used in `code_go_ast_v1.rs`'s //! internal `code_doc` test helper. //! //! Set `UPDATE_SNAPSHOTS=1` to re-bake the baseline. use std::path::PathBuf; use kebab_chunk::CodeCAstV1Chunker; use kebab_core::{ AssetId, Block, CanonicalDocument, ChunkPolicy, Chunker, ChunkerVersion, CodeBlock, CommonBlock, Lang, Metadata, ParserVersion, Provenance, SourceSpan, SourceType, TrustLevel, WorkspacePath, id_for_block, id_for_doc, }; use serde_json::Value; use time::OffsetDateTime; fn fixtures_dir() -> PathBuf { PathBuf::from(env!("CARGO_MANIFEST_DIR")) .join("tests") .join("fixtures") } fn fixed_doc() -> CanonicalDocument { let wp = WorkspacePath("projects/record.c".into()); let aid = AssetId("c".repeat(64)); // Pin parser_version so doc_id / block_ids are reproducible. let pv = ParserVersion("code-c-v1".into()); let doc_id = id_for_doc(&wp, &aid, &pv); // Representative units: // 0. imports + defines (lines 1–4, ≤200) // 1. status_t enum typedef (lines 6–9, ≤200) // 2. record_t struct typedef (lines 11–16, ≤200) // 3. static counter decl glue (line 18, ≤200) // 4. parse_record fn (lines 20–23, ≤200) // 5. print_record fn (lines 25–27, ≤200) // 6. main fn (lines 29–33, ≤200) let raw_units: Vec<(&str, u32, u32, String)> = vec![ ( "", 1, 18, "#include \n#include \n\n#define MAX_BUF 4096\n\ntypedef enum {\n OK = 0,\n ERR_PARSE,\n ERR_IO,\n} status_t;\n\ntypedef struct {\n int id;\n char name[64];\n status_t status;\n} record_t;\n\nstatic int counter = 0;".to_string(), ), ( "parse_record", 20, 23, "int parse_record(const char *line, record_t *out) {\n if (line == NULL || out == NULL) return ERR_PARSE;\n return OK;\n}".to_string(), ), ( "print_record", 25, 27, "void print_record(const record_t *r) {\n printf(\"[%d] %s (status=%d)\\n\", r->id, r->name, r->status);\n}".to_string(), ), ( "main", 29, 33, "int main(void) {\n record_t r = { .id = 1, .name = \"foo\", .status = OK };\n print_record(&r);\n return 0;\n}".to_string(), ), ]; let blocks: Vec = raw_units .iter() .enumerate() .map(|(i, (sym, ls, le, code))| { let span = SourceSpan::Code { line_start: *ls, line_end: *le, symbol: Some((*sym).to_string()), lang: Some("c".into()), }; let bid = id_for_block(&doc_id, "code", &[], i as u32, &span); Block::Code(CodeBlock { common: CommonBlock { block_id: bid, heading_path: vec![], source_span: span, }, lang: Some("c".into()), code: code.clone(), }) }) .collect(); CanonicalDocument { doc_id, source_asset_id: aid, workspace_path: wp, title: "record.c".into(), lang: Lang("und".into()), blocks, metadata: Metadata { aliases: vec![], tags: vec![], created_at: OffsetDateTime::from_unix_timestamp(1_700_000_000).unwrap(), updated_at: OffsetDateTime::from_unix_timestamp(1_700_000_000).unwrap(), source_type: SourceType::Note, trust_level: TrustLevel::Primary, user_id_alias: None, user: Default::default(), repo: Some("kebab".into()), git_branch: Some("main".into()), git_commit: Some("0".repeat(40)), code_lang: Some("c".into()), source_id: None, }, provenance: Provenance { events: vec![] }, parser_version: pv, schema_version: 1, doc_version: 1, last_chunker_version: None, last_embedding_version: None, } } fn fixed_policy() -> ChunkPolicy { ChunkPolicy { target_tokens: 500, overlap_tokens: 80, respect_markdown_headings: false, chunker_version: ChunkerVersion("code-c-ast-v1".into()), } } #[test] fn code_c_ast_chunks_snapshot() { let doc = fixed_doc(); let policy = fixed_policy(); let chunks = CodeCAstV1Chunker.chunk(&doc, &policy).expect("chunk"); let actual = serde_json::to_value(&chunks).unwrap(); let dir = fixtures_dir(); let baseline_path = dir.join("code-sample.c.chunks.snapshot.json"); let baseline_text = match std::fs::read_to_string(&baseline_path) { Ok(s) => s, Err(_) if std::env::var("UPDATE_SNAPSHOTS").is_ok() => { std::fs::create_dir_all(&dir).unwrap(); let pretty = serde_json::to_string_pretty(&actual).unwrap(); std::fs::write(&baseline_path, format!("{pretty}\n")).unwrap(); return; } Err(e) => panic!( "missing baseline {}; run with UPDATE_SNAPSHOTS=1 to create: {e}", baseline_path.display() ), }; let expected: Value = serde_json::from_str(&baseline_text).expect("baseline parses as json"); if actual != expected { if std::env::var("UPDATE_SNAPSHOTS").is_ok() { let pretty = serde_json::to_string_pretty(&actual).unwrap(); std::fs::write(&baseline_path, format!("{pretty}\n")).unwrap(); eprintln!("updated baseline {}", baseline_path.display()); return; } let pretty = serde_json::to_string_pretty(&actual).unwrap(); panic!( "code-c-ast-v1 chunks snapshot drift\n\ --- expected ({}) ---\n{baseline_text}\n\ --- actual ---\n{pretty}\n\ If intentional, re-run with UPDATE_SNAPSHOTS=1.", baseline_path.display() ); } } /// Determinism cross-check: re-running the same pipeline yields the same /// chunk_ids byte-for-byte. #[test] fn code_c_ast_chunks_are_deterministic() { let policy = fixed_policy(); let baseline: Vec = CodeCAstV1Chunker .chunk(&fixed_doc(), &policy) .unwrap() .into_iter() .map(|c| c.chunk_id.0) .collect(); for _ in 0..5 { let again: Vec = CodeCAstV1Chunker .chunk(&fixed_doc(), &policy) .unwrap() .into_iter() .map(|c| c.chunk_id.0) .collect(); assert_eq!(again, baseline); } }