From 3d7b51ce48f53e10ab1a3be2dad85276e0aa0de8 Mon Sep 17 00:00:00 2001 From: Moon Dongyoung Date: Wed, 10 Jul 2024 16:35:26 +0900 Subject: [PATCH] test: walk_expr with binary expressions --- src/util/visitor.rs | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/src/util/visitor.rs b/src/util/visitor.rs index 1075e0e..afe3c3f 100644 --- a/src/util/visitor.rs +++ b/src/util/visitor.rs @@ -193,4 +193,29 @@ mod tests { let ast = parser::parse(r#""1""#).unwrap(); assert!(!walk_expr(&mut visitor, &ast).unwrap()); } + + #[test] + fn test_binary_expr() { + let mut visitor = NamespaceVisitor { + namespace: "sample".to_string(), + }; + + let ast = parser::parse("pg_stat_activity_count{namespace=\"sample\"} + pg_stat_activity_count{}").unwrap(); + assert!(!walk_expr(&mut visitor, &ast).unwrap()); + + let ast = parser::parse("pg_stat_activity_count{} - pg_stat_activity_count{namespace=\"sample\"}").unwrap(); + assert!(!walk_expr(&mut visitor, &ast).unwrap()); + + let ast = parser::parse("pg_stat_activity_count{} * pg_stat_activity_count{}").unwrap(); + assert!(!walk_expr(&mut visitor, &ast).unwrap()); + + let ast = parser::parse("pg_stat_activity_count{namespace=\"sample\"} / 1").unwrap(); + assert!(!walk_expr(&mut visitor, &ast).unwrap()); + + let ast = parser::parse("1 % pg_stat_activity_count{namespace=\"sample\"}").unwrap(); + assert!(!walk_expr(&mut visitor, &ast).unwrap()); + + let ast = parser::parse("pg_stat_activity_count{namespace=\"sample\"} ^ pg_stat_activity_count{namespace=\"sample\"}").unwrap(); + assert!(walk_expr(&mut visitor, &ast).unwrap()); + } }