From 92ecfcb83cd2ec4b8fe91b0a41fdc2f24d39965c Mon Sep 17 00:00:00 2001
From: Josh <josh@joshthomas.dev>
Date: Sat, 4 Jan 2025 19:40:40 -0600
Subject: [PATCH] add test for error recovery

---
 crates/djls-template-ast/src/parser.rs        | 29 +++++++++--
 ...__tests__errors__parse_error_recovery.snap | 50 +++++++++++++++++++
 ...__tests__errors__parse_unclosed_style.snap |  7 +++
 3 files changed, 83 insertions(+), 3 deletions(-)
 create mode 100644 crates/djls-template-ast/src/snapshots/djls_template_ast__parser__tests__errors__parse_error_recovery.snap
 create mode 100644 crates/djls-template-ast/src/snapshots/djls_template_ast__parser__tests__errors__parse_unclosed_style.snap

diff --git a/crates/djls-template-ast/src/parser.rs b/crates/djls-template-ast/src/parser.rs
index 7e8334d..a31fbb8 100644
--- a/crates/djls-template-ast/src/parser.rs
+++ b/crates/djls-template-ast/src/parser.rs
@@ -736,9 +736,32 @@ mod tests {
             let source = "<style>body { color: blue; ";
             let tokens = Lexer::new(source).tokenize().unwrap();
             let mut parser = Parser::new(tokens);
-            let result = parser.parse().unwrap();
-            assert_eq!(result.errors().len(), 1);
-            assert!(matches!(&result.errors()[0], AstError::UnclosedTag(tag) if tag == "style"));
+            let ast = parser.parse().unwrap();
+            insta::assert_yaml_snapshot!(ast);
+            assert_eq!(ast.errors().len(), 1);
+            assert!(matches!(&ast.errors()[0], AstError::UnclosedTag(tag) if tag == "style"));
+        }
+
+        #[test]
+        fn test_parse_error_recovery() {
+            let source = r#"<div class="container">
+    <h1>Header</h1>
+    {% if user.is_authenticated %}
+        <p>Welcome {{ user.name }}</p>
+        <div>
+            {# This div is unclosed #}
+        {% for item in items %}
+            <span>{{ item }}</span>
+        {% endfor %}
+    {% endif %}
+    <footer>Page Footer</footer>
+</div>"#;
+            let tokens = Lexer::new(source).tokenize().unwrap();
+            let mut parser = Parser::new(tokens);
+            let ast = parser.parse().unwrap();
+            insta::assert_yaml_snapshot!(ast);
+            assert_eq!(ast.errors().len(), 1);
+            assert!(matches!(&ast.errors()[0], AstError::UnclosedTag(tag) if tag == "div"));
         }
     }
 
diff --git a/crates/djls-template-ast/src/snapshots/djls_template_ast__parser__tests__errors__parse_error_recovery.snap b/crates/djls-template-ast/src/snapshots/djls_template_ast__parser__tests__errors__parse_error_recovery.snap
new file mode 100644
index 0000000..87a4263
--- /dev/null
+++ b/crates/djls-template-ast/src/snapshots/djls_template_ast__parser__tests__errors__parse_error_recovery.snap
@@ -0,0 +1,50 @@
+---
+source: crates/djls-template-ast/src/parser.rs
+expression: ast
+---
+nodes:
+  - Html:
+      Element:
+        tag_name: div
+        attributes:
+          class:
+            Value: container
+        children:
+          - Html:
+              Element:
+                tag_name: h1
+                attributes: {}
+                children:
+                  - Text: Header
+          - Django:
+              Tag:
+                Block:
+                  name: if
+                  bits:
+                    - if
+                    - user.is_authenticated
+                  children:
+                    - Html:
+                        Element:
+                          tag_name: p
+                          attributes: {}
+                          children:
+                            - Text: "Welcome "
+                            - Django:
+                                Variable:
+                                  bits:
+                                    - user
+                                    - name
+                                  filters: []
+                    - Django:
+                        Tag:
+                          Closing:
+                            name: endif
+                            bits: []
+          - Html:
+              Element:
+                tag_name: footer
+                attributes: {}
+                children:
+                  - Text: Page Footer
+errors: []
diff --git a/crates/djls-template-ast/src/snapshots/djls_template_ast__parser__tests__errors__parse_unclosed_style.snap b/crates/djls-template-ast/src/snapshots/djls_template_ast__parser__tests__errors__parse_unclosed_style.snap
new file mode 100644
index 0000000..1dd2006
--- /dev/null
+++ b/crates/djls-template-ast/src/snapshots/djls_template_ast__parser__tests__errors__parse_unclosed_style.snap
@@ -0,0 +1,7 @@
+---
+source: crates/djls-template-ast/src/parser.rs
+expression: ast
+---
+nodes: []
+errors:
+  - UnclosedTag: style