Function test_sse_stream_ignores_events_without_usage

src/translate.rs:1081–1102

()

    #[test]
    fn test_sse_stream_ignores_events_without_usage() {
        let stats = Arc::new(Stats::new(None));
        let scanner = Arc::new(DlpScanner::new(&[], false).unwrap());
        let sse = "data: {\"choices\":[{\"delta\":{\"content\":\"hello\"}}]}\n\
                   data: {\"choices\":[{\"delta\":{\"content\":\" world\"}}]}\n\
                   data: [DONE]\n";
        let body = Body::from(sse);
        let wrapped = wrap_body_with_dlp_sse_stream(body, scanner, stats.clone());

        let rt = tokio::runtime::Builder::new_current_thread()
            .build()
            .unwrap();
        rt.block_on(async {
            use http_body_util::BodyExt;
            wrapped.collect().await.unwrap();
        });

        let snap = stats.snapshot();
        assert_eq!(snap.prompt_tokens_total, 0);
        assert_eq!(snap.completion_tokens_total, 0);
        assert_eq!(snap.total_tokens_total, 0);
    }
}