Function transcripts_metadata_columns

crates/ccql/src/sql/composite_storage.rs:853–919

()

    #[tokio::test]
    async fn transcripts_metadata_columns() {
        let temp = tempfile::tempdir().expect("temp");
        let root = temp.path();
        let slug = "-Users-douglance-Developer-lv-devsql";

        write(
            &root.join("projects").join(slug).join("sess-top.jsonl"),
            r#"{"type":"user","content":"hi"}"#,
        );
        write(
            &root
                .join("projects")
                .join(slug)
                .join("sess-parent")
                .join("subagents")
                .join("agent-abc123.jsonl"),
            r#"{"type":"assistant","content":"sub"}"#,
        );
        write(
            &root.join("transcripts").join("ses_legacy1.jsonl"),
            r#"{"type":"user","content":"old"}"#,
        );

        let storage = CompositeStorage::new(config_for(root)).expect("storage");
        let rows = storage.scan_transcripts().expect("scan");

        let get = |needle: &str| -> &BTreeMap<String, Value> {
            rows.iter()
                .find_map(|(_, r)| match r {
                    DataRow::Map(m) => {
                        if matches!(m.get("_source_file"), Some(Value::Str(s)) if s == needle) {
                            Some(m)
                        } else {
                            None
                        }
                    }
                    _ => None,
                })
                .unwrap_or_else(|| panic!("row {} not found", needle))
        };

        let top = get("sess-top.jsonl");
        assert_eq!(top.get("_session_id"), Some(&Value::Str("sess-top".into())));
        assert_eq!(top.get("_project"), Some(&Value::Str(slug.into())));
        assert!(top.get("_agent_id").is_none());

        let sub = get("agent-abc123.jsonl");
        assert_eq!(
            sub.get("_session_id"),
            Some(&Value::Str("sess-parent".into())),
            "subagent session id is parent dir name"
        );
        assert_eq!(sub.get("_project"), Some(&Value::Str(slug.into())));
        assert_eq!(
            sub.get("_agent_id"),
            Some(&Value::Str("agent-abc123".into()))
        );