test: walk_expr with binary expressions

GreptimeTeam · Jul 10, 2024 · 3d7b51c · 3d7b51c
1 parent 65ce0c4
commit 3d7b51c
Showing 1 changed file with 25 additions and 0 deletions.
diff --git 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());
+    }
 }