Add test

surrealdb · May 1, 2024 · 958bd76 · 958bd76
1 parent dec1359
commit 958bd76
Showing 1 changed file with 42 additions and 19 deletions.
diff --git a/core/src/syn/parser/test/limit.rs b/core/src/syn/parser/test/limit.rs
@@ -45,6 +45,29 @@ fn object_depth() {
  .expect_err("recursion limit of 5 didn't trigger on 6 deep object");
 }
 
+#[test]
+fn array_depth() {
+ let mut stack = Stack::new();
+
+ let source = r#"
+ RETURN [ [ [ [ [ ] ] ] ] ]
+ "#;
+ let mut parser = Parser::new(source.as_bytes()).with_object_recursion_limit(5);
+ stack
+ .enter(|stk| parser.parse_query(stk))
+ .finish()
+ .expect("recursion limit of 5 couldn't parse 5 deep object");
+
+ let source = r#"
+ RETURN [ [ [ [ [ [ ] ] ] ] ] ]
+ "#;
+ let mut parser = Parser::new(source.as_bytes()).with_object_recursion_limit(5);
+ stack
+ .enter(|stk| parser.parse_query(stk))
+ .finish()
+ .expect_err("recursion limit of 5 didn't trigger on 6 deep object");
+}
+
 #[test]
 fn object_depth_succeed_then_fail() {
  let mut stack = Stack::new();
@@ -59,16 +82,16 @@ fn object_depth_succeed_then_fail() {
  }
  }
  };
- RETURN {
- a: {
- b: {
- c: {
- d: {
- }
+ RETURN {
+ a: {
+ b: {
+ c: {
+ d: {
  }
  }
  }
- };
+ }
+ };
  "#;
 
  let mut parser = Parser::new(source.as_bytes()).with_object_recursion_limit(5);
@@ -89,18 +112,18 @@ fn object_depth_succeed_then_fail() {
  }
  }
  };
- RETURN {
- a: {
- b: {
- c: {
- d: {
- e: {
- }
+ RETURN {
+ a: {
+ b: {
+ c: {
+ d: {
+ e: {
  }
  }
  }
  }
- };
+ }
+ };
  "#;
 
  let mut parser = Parser::new(source.as_bytes()).with_object_recursion_limit(5);
@@ -116,7 +139,7 @@ fn query_depth_subquery() {
 
  let source = r#"
  RETURN select (select (select ( select foo from bar ) from bar ) from bar) from bar
- "#;
+  "#;
  let mut parser = Parser::new(source.as_bytes()).with_query_recursion_limit(5);
  stack
  .enter(|stk| parser.parse_query(stk))
@@ -125,7 +148,7 @@ fn query_depth_subquery() {
 
  let source = r#"
  RETURN select (select (select ( select (select foo from bar) from bar ) from bar ) from bar) from bar
- "#;
+  "#;
  let mut parser = Parser::new(source.as_bytes()).with_query_recursion_limit(5);
  stack
  .enter(|stk| parser.parse_query(stk))
@@ -179,7 +202,7 @@ fn query_depth_if() {
  let mut stack = Stack::new();
 
  let source = r#"
- IF IF IF IF IF true THEN false END { false } { false } { false } { false }
+ IF IF IF IF IF true THEN false END { false } { false } { false } { false }
  "#;
  let mut parser = Parser::new(source.as_bytes()).with_query_recursion_limit(5);
  stack
@@ -188,7 +211,7 @@ fn query_depth_if() {
  .expect("recursion limit of 5 couldn't parse 5 deep query");
 
  let source = r#"
- IF IF IF IF IF IF true THEN false END { false } { false } { false } { false } { false }
+ IF IF IF IF IF IF true THEN false END { false } { false } { false } { false } { false }
  "#;
  let mut parser = Parser::new(source.as_bytes()).with_query_recursion_limit(5);
  stack