1use std::future::Future;
7use std::pin::Pin;
8use std::str::FromStr;
9use std::sync::Arc;
10use std::sync::atomic::{AtomicUsize, Ordering};
11use std::task::{Context, Poll};
12
13use tower::Service;
14use tracing::{debug, error, info, trace, warn};
15
16use camel_component_api::UriConfig;
17use camel_component_api::parse_uri;
18use camel_component_api::{BoxProcessor, CamelError, Exchange};
19use camel_component_api::{
20 Component, ComponentMetadata, Consumer, Endpoint, ProducerContext, RuntimeObservability,
21};
22
23#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
29pub enum LogLevel {
30 Trace,
31 Debug,
32 #[default]
33 Info,
34 Warn,
35 Error,
36}
37
38impl FromStr for LogLevel {
39 type Err = String;
40
41 fn from_str(s: &str) -> Result<Self, Self::Err> {
42 parse_log_level(s).map_err(|e| e.to_string())
43 }
44}
45
46fn parse_log_level(s: &str) -> Result<LogLevel, CamelError> {
47 match s.to_uppercase().as_str() {
48 "TRACE" => Ok(LogLevel::Trace),
49 "DEBUG" => Ok(LogLevel::Debug),
50 "INFO" => Ok(LogLevel::Info),
51 "WARN" | "WARNING" => Ok(LogLevel::Warn),
52 "ERROR" => Ok(LogLevel::Error),
53 _ => Err(CamelError::Config(format!(
54 "unknown log level: '{}'. Valid: TRACE, DEBUG, INFO, WARN, ERROR",
55 s
56 ))),
57 }
58}
59
60#[derive(Debug, Clone, UriConfig)]
68#[uri_scheme = "log"]
69#[uri_config(
70 skip_impl,
71 metadata(
72 scheme = "log",
73 description = "Logs exchanges at a configurable level",
74 producer
75 ),
76 crate = "camel_component_api"
77)]
78pub struct LogConfig {
79 pub category: String,
81 #[uri_param(
83 kind = "enum:TRACE,DEBUG,INFO,WARN,ERROR",
84 default = "INFO",
85 desc = "Log level"
86 )]
87 pub level: LogLevel,
88 #[uri_param(
90 name = "showHeaders",
91 default = "false",
92 desc = "Include headers in log output"
93 )]
94 pub show_headers: bool,
95 #[uri_param(
97 name = "showBody",
98 default = "true",
99 desc = "Include body in log output"
100 )]
101 pub show_body: bool,
102 #[uri_param(
105 name = "maxChars",
106 desc = "Truncate body to N characters. Omit for no limit"
107 )]
108 pub max_chars: Option<usize>,
109 #[uri_param(
113 name = "logMask",
114 default = "false",
115 desc = "Redact sensitive headers and body"
116 )]
117 pub log_mask: bool,
118 #[uri_param(
120 name = "showStreamInfo",
121 default = "false",
122 desc = "Show stream origin info"
123 )]
124 pub show_stream_info: bool,
125 #[uri_param(
128 name = "groupSize",
129 desc = "Emit log every N exchanges for group logging"
130 )]
131 pub group_size: Option<usize>,
132}
133
134impl UriConfig for LogConfig {
135 fn scheme() -> &'static str {
136 "log"
137 }
138
139 fn from_uri(uri: &str) -> Result<Self, CamelError> {
140 let parts = parse_uri(uri)?;
141 Self::from_components(parts)
142 }
143
144 fn from_components(parts: camel_component_api::UriComponents) -> Result<Self, CamelError> {
145 if parts.scheme != Self::scheme() {
146 return Err(CamelError::InvalidUri(format!(
147 "expected scheme '{}' but got '{}'",
148 Self::scheme(),
149 parts.scheme
150 )));
151 }
152
153 let level = match parts.params.get("level") {
154 Some(raw) => parse_log_level(raw)?,
155 None => LogLevel::Info,
156 };
157
158 let show_headers = match parts.params.get("showHeaders") {
159 Some(raw) => raw.parse::<bool>().map_err(|_| {
160 CamelError::InvalidUri(format!("invalid boolean value for showHeaders: {raw}"))
161 })?,
162 None => false,
163 };
164
165 let show_body = match parts.params.get("showBody") {
166 Some(raw) => raw.parse::<bool>().map_err(|_| {
167 CamelError::InvalidUri(format!("invalid boolean value for showBody: {raw}"))
168 })?,
169 None => true,
170 };
171
172 let max_chars = match parts.params.get("maxChars") {
173 Some(raw) => Some(raw.parse::<usize>().map_err(|_| {
174 CamelError::InvalidUri(format!("invalid integer value for maxChars: {raw}"))
175 })?),
176 None => None,
177 };
178
179 let log_mask = match parts.params.get("logMask") {
180 Some(raw) => raw.parse::<bool>().map_err(|_| {
181 CamelError::InvalidUri(format!("invalid boolean value for logMask: {raw}"))
182 })?,
183 None => false,
184 };
185
186 let show_stream_info = match parts.params.get("showStreamInfo") {
187 Some(raw) => raw.parse::<bool>().map_err(|_| {
188 CamelError::InvalidUri(format!("invalid boolean value for showStreamInfo: {raw}"))
189 })?,
190 None => false,
191 };
192
193 let group_size = match parts.params.get("groupSize") {
194 Some(raw) => Some(raw.parse::<usize>().map_err(|_| {
195 CamelError::InvalidUri(format!("invalid integer value for groupSize: {raw}"))
196 })?),
197 None => None,
198 };
199
200 Ok(Self {
201 category: parts.path,
202 level,
203 show_headers,
204 show_body,
205 max_chars,
206 log_mask,
207 show_stream_info,
208 group_size,
209 })
210 }
211}
212
213pub struct LogComponent;
219
220impl LogComponent {
221 pub fn new() -> Self {
222 Self
223 }
224}
225
226impl Default for LogComponent {
227 fn default() -> Self {
228 Self::new()
229 }
230}
231
232impl Component for LogComponent {
233 fn scheme(&self) -> &str {
234 "log"
235 }
236
237 fn metadata(&self) -> ComponentMetadata {
238 LogConfig::metadata()
239 }
240
241 fn create_endpoint(
242 &self,
243 uri: &str,
244 _ctx: &dyn camel_component_api::ComponentContext,
245 ) -> Result<Box<dyn Endpoint>, CamelError> {
246 let config = LogConfig::from_uri(uri)?;
247 Ok(Box::new(LogEndpoint {
248 uri: uri.to_string(),
249 config,
250 }))
251 }
252}
253
254struct LogEndpoint {
259 uri: String,
260 config: LogConfig,
261}
262
263impl Endpoint for LogEndpoint {
264 fn uri(&self) -> &str {
265 &self.uri
266 }
267
268 fn create_consumer(
269 &self,
270 _rt: Arc<dyn RuntimeObservability>,
271 ) -> Result<Box<dyn Consumer>, CamelError> {
272 Err(CamelError::EndpointCreationFailed(
273 "log endpoint does not support consumers".to_string(),
274 ))
275 }
276
277 fn create_producer(
278 &self,
279 _rt: Arc<dyn RuntimeObservability>,
280 _ctx: &ProducerContext,
281 ) -> Result<BoxProcessor, CamelError> {
282 Ok(BoxProcessor::new(LogProducer::new(self.config.clone())))
283 }
284}
285
286#[derive(Clone)]
291struct LogProducer {
292 config: LogConfig,
293 exchange_count: Arc<AtomicUsize>,
294}
295
296impl LogProducer {
297 fn new(config: LogConfig) -> Self {
298 Self {
299 config,
300 exchange_count: Arc::new(AtomicUsize::new(0)),
301 }
302 }
303
304 fn is_sensitive_header(key: &str) -> bool {
306 let lower = key.to_lowercase();
307 let sensitive_keywords = [
308 "password",
309 "passwd",
310 "secret",
311 "token",
312 "apikey",
313 "api-key",
314 "api_key",
315 "authorization",
316 "auth",
317 "credential",
318 "private",
319 "signature",
320 ];
321 sensitive_keywords.iter().any(|kw| {
322 lower == *kw
323 || lower.ends_with(&format!("-{kw}"))
324 || lower.ends_with(&format!("_{kw}"))
325 || lower.starts_with(&format!("{kw}-"))
326 || lower.starts_with(&format!("{kw}_"))
327 })
328 }
329
330 fn format_exchange(&self, exchange: &Exchange, count: usize) -> String {
331 let mut parts = Vec::new();
332
333 if self.config.show_body {
334 let body_str = if self.config.log_mask {
335 "[Body redacted by logMask]".to_string()
336 } else {
337 match &exchange.input.body {
338 camel_component_api::Body::Empty => "[empty]".to_string(),
339 camel_component_api::Body::Text(s) => s.clone(),
340 camel_component_api::Body::Json(v) => v.to_string(),
341 camel_component_api::Body::Xml(s) => s.clone(),
342 camel_component_api::Body::Bytes(b) => format!("[{} bytes]", b.len()),
343 camel_component_api::Body::Stream(s) => {
344 if self.config.show_stream_info {
345 format!("[Stream: origin={:?}]", s.metadata.origin)
346 } else {
347 "[Stream]".to_string()
348 }
349 }
350 _ => "[unsupported body type]".to_string(),
352 }
353 };
354
355 let body_str = if let Some(limit) = self.config.max_chars {
356 body_str.chars().take(limit).collect()
357 } else {
358 body_str
359 };
360
361 parts.push(format!("Body: {body_str}"));
362 }
363
364 if self.config.show_headers && !exchange.input.headers.is_empty() {
365 let headers: Vec<String> = exchange
366 .input
367 .headers
368 .iter()
369 .map(|(k, v)| {
370 if self.config.log_mask && Self::is_sensitive_header(k) {
371 format!("{k}=[REDACTED]")
372 } else {
373 format!("{k}={v}")
374 }
375 })
376 .collect();
377 parts.push(format!("Headers: {{{}}}", headers.join(", ")));
378 }
379
380 if parts.is_empty() {
381 format!("[{}] Exchange received", self.config.category)
382 } else if self.config.group_size.is_some() {
383 format!(
384 "[{}] Group of {count}: {}",
385 self.config.category,
386 parts.join(" | ")
387 )
388 } else {
389 format!("[{}] {}", self.config.category, parts.join(" | "))
390 }
391 }
392}
393
394impl Service<Exchange> for LogProducer {
395 type Response = Exchange;
396 type Error = CamelError;
397 type Future = Pin<Box<dyn Future<Output = Result<Exchange, CamelError>> + Send>>;
398
399 fn poll_ready(&mut self, _cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
400 Poll::Ready(Ok(()))
401 }
402
403 fn call(&mut self, exchange: Exchange) -> Self::Future {
404 let count = self.exchange_count.fetch_add(1, Ordering::Relaxed) + 1;
405
406 if !Self::should_emit(count, self.config.group_size) {
408 return Box::pin(async move { Ok(exchange) });
409 }
410
411 let msg = self.format_exchange(&exchange, count);
412 let level = self.config.level;
413
414 Box::pin(async move {
415 match level {
416 LogLevel::Trace => trace!("{msg}"),
417 LogLevel::Debug => debug!("{msg}"),
418 LogLevel::Info => info!("{msg}"),
419 LogLevel::Warn => warn!("{msg}"),
420 LogLevel::Error => error!("{msg}"),
428 }
429
430 Ok(exchange)
431 })
432 }
433}
434
435impl LogProducer {
436 fn should_emit(count: usize, group_size: Option<usize>) -> bool {
441 match group_size {
442 None => true,
443 Some(gs) => count.is_multiple_of(gs),
444 }
445 }
446}
447
448#[cfg(test)]
453mod tests {
454 use camel_component_api::test_support::PanicRuntimeObservability;
455 fn rt() -> std::sync::Arc<dyn camel_component_api::RuntimeObservability> {
456 std::sync::Arc::new(PanicRuntimeObservability)
457 }
458 use super::*;
459 use camel_component_api::Body;
460 use camel_component_api::Message;
461 use camel_component_api::NoOpComponentContext;
462 use tower::ServiceExt;
463
464 fn test_producer_ctx() -> ProducerContext {
465 ProducerContext::new()
466 }
467
468 #[test]
469 fn test_log_config_defaults() {
470 let config = LogConfig::from_uri("log:myCategory").unwrap();
471 assert_eq!(config.category, "myCategory");
472 assert_eq!(config.level, LogLevel::Info);
473 assert!(!config.show_headers);
474 assert!(config.show_body);
475 }
476
477 #[test]
478 fn test_log_config_with_params() {
479 let config =
480 LogConfig::from_uri("log:app?level=debug&showHeaders=true&showBody=false").unwrap();
481 assert_eq!(config.category, "app");
482 assert_eq!(config.level, LogLevel::Debug);
483 assert!(config.show_headers);
484 assert!(!config.show_body);
485 }
486
487 #[test]
488 fn test_log_config_wrong_scheme() {
489 let result = LogConfig::from_uri("timer:tick");
490 assert!(result.is_err());
491 }
492
493 #[test]
494 fn test_log_component_scheme() {
495 let component = LogComponent::new();
496 assert_eq!(component.scheme(), "log");
497 }
498
499 #[test]
500 fn test_log_component_default() {
501 let component = LogComponent;
502 assert_eq!(component.scheme(), "log");
503 }
504
505 #[test]
506 fn test_log_level_from_str_variants() {
507 assert_eq!("trace".parse::<LogLevel>().unwrap(), LogLevel::Trace);
508 assert_eq!("DEBUG".parse::<LogLevel>().unwrap(), LogLevel::Debug);
509 assert_eq!("Info".parse::<LogLevel>().unwrap(), LogLevel::Info);
510 assert_eq!("warning".parse::<LogLevel>().unwrap(), LogLevel::Warn);
511 assert_eq!("error".parse::<LogLevel>().unwrap(), LogLevel::Error);
512 }
513
514 #[test]
515 fn test_log_level_from_str_invalid() {
516 let err = "nope".parse::<LogLevel>().unwrap_err();
517 assert_eq!(
518 err,
519 "Configuration error: unknown log level: 'nope'. Valid: TRACE, DEBUG, INFO, WARN, ERROR"
520 );
521 }
522
523 #[test]
524 fn test_log_config_invalid_level_rejected() {
525 let err = LogConfig::from_uri("log:test?level=invalid").unwrap_err();
526 assert!(
527 err.to_string()
528 .contains("unknown log level: 'invalid'. Valid: TRACE, DEBUG, INFO, WARN, ERROR")
529 );
530 }
531
532 #[test]
533 fn test_valid_log_levels_accepted() {
534 assert!(parse_log_level("DEBUG").is_ok());
535 assert!(parse_log_level("info").is_ok());
536 assert!(parse_log_level("WARN").is_ok());
537 assert!(parse_log_level("WARNING").is_ok());
538 }
539
540 #[test]
541 fn test_invalid_log_level_rejected() {
542 assert!(parse_log_level("VERBOSE").is_err());
543 assert!(parse_log_level("").is_err());
544 assert!(parse_log_level("log").is_err());
545 }
546
547 #[test]
548 fn test_log_endpoint_uri() {
549 let component = LogComponent::new();
550 let endpoint = component
551 .create_endpoint("log:uri-check", &NoOpComponentContext)
552 .unwrap();
553 assert_eq!(endpoint.uri(), "log:uri-check");
554 }
555
556 #[test]
557 fn test_log_endpoint_no_consumer() {
558 let component = LogComponent::new();
559 let endpoint = component
560 .create_endpoint("log:info", &NoOpComponentContext)
561 .unwrap();
562 assert!(endpoint.create_consumer(rt()).is_err());
563 }
564
565 #[test]
566 fn test_log_endpoint_creates_producer() {
567 let ctx = test_producer_ctx();
568 let component = LogComponent::new();
569 let endpoint = component
570 .create_endpoint("log:info", &NoOpComponentContext)
571 .unwrap();
572 assert!(endpoint.create_producer(rt(), &ctx).is_ok());
573 }
574
575 #[tokio::test]
576 async fn test_log_producer_processes_exchange() {
577 let ctx = test_producer_ctx();
578 let component = LogComponent::new();
579 let endpoint = component
580 .create_endpoint("log:test?showHeaders=true", &NoOpComponentContext)
581 .unwrap();
582 let producer = endpoint.create_producer(rt(), &ctx).unwrap();
583
584 let mut exchange = Exchange::new(Message::new("hello world"));
585 exchange
586 .input
587 .set_header("source", serde_json::Value::String("test".into()));
588
589 let result = producer.oneshot(exchange).await.unwrap();
590 assert_eq!(result.input.body.as_text(), Some("hello world"));
592 }
593
594 #[test]
595 fn test_format_exchange_without_body_or_headers() {
596 let producer = LogProducer::new(LogConfig {
597 category: "cat".to_string(),
598 level: LogLevel::Info,
599 show_headers: false,
600 show_body: false,
601 max_chars: None,
602 log_mask: false,
603 show_stream_info: false,
604 group_size: None,
605 });
606 let exchange = Exchange::new(Message::new("ignored"));
607 let formatted = producer.format_exchange(&exchange, 1);
608 assert_eq!(formatted, "[cat] Exchange received");
609 }
610
611 #[test]
612 fn test_format_exchange_body_variants() {
613 let base = LogProducer::new(LogConfig {
614 category: "cat".to_string(),
615 level: LogLevel::Info,
616 show_headers: false,
617 show_body: true,
618 max_chars: None,
619 log_mask: false,
620 show_stream_info: true,
621 group_size: None,
622 });
623
624 let empty = Exchange::new(Message::default());
625 assert!(base.format_exchange(&empty, 1).contains("Body: [empty]"));
626
627 let mut json_msg = Message::new("");
628 json_msg.body = Body::Json(serde_json::json!({"k":"v"}));
629 let json_ex = Exchange::new(json_msg);
630 assert!(
631 base.format_exchange(&json_ex, 2)
632 .contains("Body: {\"k\":\"v\"}")
633 );
634
635 let mut xml_msg = Message::new("");
636 xml_msg.body = Body::Xml("<a/>".to_string());
637 let xml_ex = Exchange::new(xml_msg);
638 assert!(base.format_exchange(&xml_ex, 3).contains("Body: <a/>"));
639
640 let mut bytes_msg = Message::new("");
641 bytes_msg.body = Body::Bytes(b"abc".to_vec().into());
642 let bytes_ex = Exchange::new(bytes_msg);
643 assert!(
644 base.format_exchange(&bytes_ex, 4)
645 .contains("Body: [3 bytes]")
646 );
647 }
648
649 #[test]
650 fn test_log_truncates_large_body() {
651 let producer = LogProducer::new(LogConfig {
652 category: "trunc".to_string(),
653 level: LogLevel::Info,
654 show_headers: false,
655 show_body: true,
656 max_chars: Some(10),
657 log_mask: false,
658 show_stream_info: false,
659 group_size: None,
660 });
661
662 let long_body = "a".repeat(100);
663 let exchange = Exchange::new(Message::new(long_body));
664 let formatted = producer.format_exchange(&exchange, 1);
665
666 let body_part = formatted.split_once("Body: ").unwrap().1;
668 assert!(
669 body_part.chars().count() <= 10,
670 "expected body <= 10 chars, got {} chars: {body_part:?}",
671 body_part.chars().count()
672 );
673 }
674
675 #[test]
676 fn test_log_truncates_multibyte_body() {
677 let producer = LogProducer::new(LogConfig {
678 category: "mb".to_string(),
679 level: LogLevel::Info,
680 show_headers: false,
681 show_body: true,
682 max_chars: Some(3),
683 log_mask: false,
684 show_stream_info: false,
685 group_size: None,
686 });
687
688 let exchange = Exchange::new(Message::new("日本語测试"));
689 let formatted = producer.format_exchange(&exchange, 1);
690
691 let body_part = formatted.split_once("Body: ").unwrap().1;
692 assert_eq!(
693 body_part.chars().count(),
694 3,
695 "expected 3 chars, got {}: {body_part:?}",
696 body_part.chars().count()
697 );
698 }
701
702 #[test]
703 fn test_log_truncates_multibyte_no_panic() {
704 let producer = LogProducer::new(LogConfig {
705 category: "cafe".to_string(),
706 level: LogLevel::Info,
707 show_headers: false,
708 show_body: true,
709 max_chars: Some(4),
710 log_mask: false,
711 show_stream_info: false,
712 group_size: None,
713 });
714
715 let exchange = Exchange::new(Message::new("café"));
716 let formatted = producer.format_exchange(&exchange, 1);
717
718 let body_part = formatted.split_once("Body: ").unwrap().1;
719 assert_eq!(
720 body_part.chars().count(),
721 4,
722 "expected 4 chars, got {}: {body_part:?}",
723 body_part.chars().count()
724 );
725 }
726
727 #[test]
728 fn test_log_no_truncation_when_max_chars_unset() {
729 let producer = LogProducer::new(LogConfig {
730 category: "notrunc".to_string(),
731 level: LogLevel::Info,
732 show_headers: false,
733 show_body: true,
734 max_chars: None,
735 log_mask: false,
736 show_stream_info: false,
737 group_size: None,
738 });
739
740 let long_body = "b".repeat(200);
741 let exchange = Exchange::new(Message::new(long_body));
742 let formatted = producer.format_exchange(&exchange, 1);
743
744 let body_part = formatted.split_once("Body: ").unwrap().1;
745 assert_eq!(body_part.len(), 200);
746 }
747
748 #[test]
749 fn test_log_config_max_chars_param() {
750 let config = LogConfig::from_uri("log:test?maxChars=50").unwrap();
751 assert_eq!(config.max_chars, Some(50));
752 }
753
754 #[test]
755 fn test_log_config_max_chars_default_unset() {
756 let config = LogConfig::from_uri("log:test").unwrap();
757 assert_eq!(config.max_chars, None);
758 }
759
760 #[test]
763 fn test_log_config_log_mask_param() {
764 let config = LogConfig::from_uri("log:test?logMask=true").unwrap();
765 assert!(config.log_mask);
766 }
767
768 #[test]
769 fn test_log_config_log_mask_default_false() {
770 let config = LogConfig::from_uri("log:test").unwrap();
771 assert!(!config.log_mask);
772 }
773
774 #[test]
775 fn test_log_mask_redacts_sensitive_headers() {
776 let producer = LogProducer::new(LogConfig {
777 category: "cat".to_string(),
778 level: LogLevel::Info,
779 show_headers: true,
780 show_body: false,
781 max_chars: None,
782 log_mask: true,
783 show_stream_info: false,
784 group_size: None,
785 });
786
787 let mut exchange = Exchange::new(Message::new("body"));
788 exchange.input.set_header(
789 "X-Auth-Token",
790 serde_json::Value::String("secret123".into()),
791 );
792 exchange
793 .input
794 .set_header("password", serde_json::Value::String("hunter2".into()));
795 exchange
796 .input
797 .set_header("ApiKey", serde_json::Value::String("abc".into()));
798 exchange
799 .input
800 .set_header("normal-header", serde_json::Value::String("visible".into()));
801 exchange.input.set_header(
802 "user-credential",
803 serde_json::Value::String("sensitive".into()),
804 );
805 exchange
806 .input
807 .set_header("secret-value", serde_json::Value::String("hidden".into()));
808
809 let formatted = producer.format_exchange(&exchange, 1);
810 assert!(
811 formatted.contains("X-Auth-Token=[REDACTED]"),
812 "auth header must be redacted: {formatted}"
813 );
814 assert!(
815 formatted.contains("password=[REDACTED]"),
816 "password header must be redacted: {formatted}"
817 );
818 assert!(
819 formatted.contains("ApiKey=[REDACTED]"),
820 "key header must be redacted: {formatted}"
821 );
822 assert!(
823 formatted.contains("normal-header=\"visible\""),
824 "normal header must be visible: {formatted}"
825 );
826 assert!(
827 formatted.contains("user-credential=[REDACTED]"),
828 "credential header must be redacted: {formatted}"
829 );
830 assert!(
831 formatted.contains("secret-value=[REDACTED]"),
832 "secret header must be redacted: {formatted}"
833 );
834 }
835
836 #[test]
837 fn test_log_mask_redacts_body() {
838 let producer = LogProducer::new(LogConfig {
839 category: "cat".to_string(),
840 level: LogLevel::Info,
841 show_headers: false,
842 show_body: true,
843 max_chars: None,
844 log_mask: true,
845 show_stream_info: false,
846 group_size: None,
847 });
848
849 let exchange = Exchange::new(Message::new("sensitive body content"));
850 let formatted = producer.format_exchange(&exchange, 1);
851 assert!(
852 formatted.contains("[Body redacted by logMask]"),
853 "body must be redacted: {formatted}"
854 );
855 assert!(
856 !formatted.contains("sensitive body content"),
857 "body content must not appear: {formatted}"
858 );
859 }
860
861 #[test]
862 fn test_log_mask_off_shows_data() {
863 let producer = LogProducer::new(LogConfig {
864 category: "cat".to_string(),
865 level: LogLevel::Info,
866 show_headers: true,
867 show_body: true,
868 max_chars: None,
869 log_mask: false,
870 show_stream_info: false,
871 group_size: None,
872 });
873
874 let mut exchange = Exchange::new(Message::new("visible body"));
875 exchange
876 .input
877 .set_header("password", serde_json::Value::String("hunter2".into()));
878
879 let formatted = producer.format_exchange(&exchange, 1);
880 assert!(
881 formatted.contains("visible body"),
882 "body must be visible when mask off: {formatted}"
883 );
884 assert!(
885 formatted.contains("hunter2"),
886 "header value must be visible when mask off: {formatted}"
887 );
888 }
889
890 #[test]
893 fn test_log_stream_show_info() {
894 let producer_no_info = LogProducer::new(LogConfig {
896 category: "cat".to_string(),
897 level: LogLevel::Info,
898 show_headers: false,
899 show_body: true,
900 max_chars: None,
901 log_mask: false,
902 show_stream_info: false,
903 group_size: None,
904 });
905
906 let mut msg = Message::new("");
907 msg.body = Body::Stream(camel_component_api::StreamBody {
908 stream: std::sync::Arc::new(tokio::sync::Mutex::new(None)),
909 metadata: camel_component_api::StreamMetadata {
910 origin: Some("file:///data/test.txt".to_string()),
911 ..Default::default()
912 },
913 });
914 let exchange = Exchange::new(msg);
915 let formatted = producer_no_info.format_exchange(&exchange, 1);
916 assert!(
917 formatted.contains("Body: [Stream]"),
918 "must show [Stream] when show_stream_info=false: {formatted}"
919 );
920
921 let producer_with_info = LogProducer::new(LogConfig {
923 category: "cat".to_string(),
924 level: LogLevel::Info,
925 show_headers: false,
926 show_body: true,
927 max_chars: None,
928 log_mask: false,
929 show_stream_info: true,
930 group_size: None,
931 });
932
933 let formatted = producer_with_info.format_exchange(&exchange, 1);
934 assert!(
935 formatted.contains("Body: [Stream: origin=Some(\"file:///data/test.txt\")]"),
936 "must show origin when show_stream_info=true: {formatted}"
937 );
938 }
939
940 #[test]
943 fn test_log_group_size() {
944 let producer = LogProducer::new(LogConfig {
945 category: "cat".to_string(),
946 level: LogLevel::Info,
947 show_headers: false,
948 show_body: true,
949 max_chars: None,
950 log_mask: false,
951 show_stream_info: false,
952 group_size: Some(3),
953 });
954
955 let ex1 = Exchange::new(Message::new("first"));
957 let formatted1 = producer.format_exchange(&ex1, 3);
958 assert!(
959 formatted1.contains("Group of 3:"),
960 "group_size=3 must include count: {formatted1}"
961 );
962 assert!(
963 formatted1.contains("Body: first"),
964 "group log must include body: {formatted1}"
965 );
966 }
967
968 #[test]
969 fn test_group_size_gating_skips_non_multiples() {
970 assert!(!LogProducer::should_emit(1, Some(3)));
973 assert!(!LogProducer::should_emit(2, Some(3)));
974 assert!(LogProducer::should_emit(3, Some(3)));
975 assert!(!LogProducer::should_emit(4, Some(3)));
976 assert!(!LogProducer::should_emit(5, Some(3)));
977 assert!(LogProducer::should_emit(6, Some(3)));
978 }
979
980 #[test]
981 fn test_group_size_gating_none_emits_all() {
982 assert!(LogProducer::should_emit(1, None));
983 assert!(LogProducer::should_emit(100, None));
984 }
985
986 #[test]
987 fn test_log_config_group_size_param() {
988 let config = LogConfig::from_uri("log:test?groupSize=10").unwrap();
989 assert_eq!(config.group_size, Some(10));
990 }
991
992 #[test]
993 fn test_log_config_group_size_default_unset() {
994 let config = LogConfig::from_uri("log:test").unwrap();
995 assert_eq!(config.group_size, None);
996 }
997
998 #[test]
999 fn test_log_config_show_stream_info_param() {
1000 let config = LogConfig::from_uri("log:test?showStreamInfo=true").unwrap();
1001 assert!(config.show_stream_info);
1002 }
1003
1004 #[test]
1005 fn test_log_config_show_stream_info_default_false() {
1006 let config = LogConfig::from_uri("log:test").unwrap();
1007 assert!(!config.show_stream_info);
1008 }
1009}