file snapshots

Author: juleswritescode

crates/pgls_treesitter_grammar/tests/grammar_tests.rs

@@ -1,3 +1,5 @@
+use std::fmt::Write;
+
 use insta::assert_snapshot;
 
 fn printed_tree(sql: &str) -> String {
@@ -10,28 +12,25 @@ fn printed_tree(sql: &str) -> String {
     result
 }
 
+fn file_snapshot(name: &str, sql: &str) {
+    let mut writer = String::new();
+
+    write!(&mut writer, "{}", sql).unwrap();
+
+    writeln!(&mut writer).unwrap();
+    write!(&mut writer, "-----------------------").unwrap();
+    writeln!(&mut writer).unwrap();
+
+    write!(&mut writer, "{}", printed_tree(sql)).unwrap();
+
+    assert_snapshot!(name, writer);
+}
+
 #[test]
 fn test_1() {
     let sql = "select * from auth.users;";
 
-    assert_snapshot!(printed_tree(sql), @r"
-    program [0..25] 'select * from auth.users;'
-      statement [0..24] 'select * from auth.users'
-        select [0..8] 'select *'
-          keyword_select [0..6] 'select'
-          select_expression [7..8] '*'
-            term [7..8] '*'
-              all_fields [7..8] '*'
-                * [7..8] '*'
-        from [9..24] 'from auth.users'
-          keyword_from [9..13] 'from'
-          relation [14..24] 'auth.users'
-            table_reference [14..24] 'auth.users'
-              schema_identifier [14..18] 'auth'
-              . [18..19] '.'
-              table_identifier [19..24] 'users'
-      ; [24..25] ';'
-    ");
+    file_snapshot("test_1", sql);
 }
 
 #[test]
@@ -40,68 +39,9 @@ fn test_2() {
     let sql2 = "update auth.users set users.email = '[email protected]';";
     let sql3 = "update auth.users set auth.users.email = '[email protected]';";
 
-    assert_snapshot!(printed_tree(sql1), @r"
-    program [0..44] 'update auth.users set email = '[email protected]';'
-      statement [0..43] 'update auth.users set email = '[email protected]''
-        update [0..43] 'update auth.users set email = '[email protected]''
-          keyword_update [0..6] 'update'
-          relation [7..17] 'auth.users'
-            table_reference [7..17] 'auth.users'
-              schema_identifier [7..11] 'auth'
-              . [11..12] '.'
-              table_identifier [12..17] 'users'
-          keyword_set [18..21] 'set'
-          assignment [22..43] 'email = '[email protected]''
-            column_reference [22..27] 'email'
-              any_identifier [22..27] 'email'
-            = [28..29] '='
-            literal [30..43] ''[email protected]''
-      ; [43..44] ';'
-    ");
-
-    assert_snapshot!(printed_tree(sql2), @r"
-    program [0..50] 'update auth.users set users.email = '[email protected]';'
-      statement [0..49] 'update auth.users set users.email = '[email protected]''
-        update [0..49] 'update auth.users set users.email = '[email protected]''
-          keyword_update [0..6] 'update'
-          relation [7..17] 'auth.users'
-            table_reference [7..17] 'auth.users'
-              schema_identifier [7..11] 'auth'
-              . [11..12] '.'
-              table_identifier [12..17] 'users'
-          keyword_set [18..21] 'set'
-          assignment [22..49] 'users.email = '[email protected]''
-            column_reference [22..33] 'users.email'
-              any_identifier [22..27] 'users'
-              . [27..28] '.'
-              any_identifier [28..33] 'email'
-            = [34..35] '='
-            literal [36..49] ''[email protected]''
-      ; [49..50] ';'
-    ");
-
-    assert_snapshot!(printed_tree(sql3), @r"
-    program [0..55] 'update auth.users set auth.users.email = '[email protected]';'
-      statement [0..54] 'update auth.users set auth.users.email = '[email protected]''
-        update [0..54] 'update auth.users set auth.users.email = '[email protected]''
-          keyword_update [0..6] 'update'
-          relation [7..17] 'auth.users'
-            table_reference [7..17] 'auth.users'
-              schema_identifier [7..11] 'auth'
-              . [11..12] '.'
-              table_identifier [12..17] 'users'
-          keyword_set [18..21] 'set'
-          assignment [22..54] 'auth.users.email = '[email protected]''
-            column_reference [22..38] 'auth.users.email'
-              schema_identifier [22..26] 'auth'
-              . [26..27] '.'
-              table_identifier [27..32] 'users'
-              . [32..33] '.'
-              column_identifier [33..38] 'email'
-            = [39..40] '='
-            literal [41..54] ''[email protected]''
-      ; [54..55] ';'
-    ");
+    file_snapshot("test_2_sql1", sql1);
+    file_snapshot("test_2_sql2", sql2);
+    file_snapshot("test_2_sql3", sql3);
 }
 
 #[test]
@@ -119,84 +59,14 @@ from
 
 "#;
 
-    assert_snapshot!(printed_tree(sql), @r"
-    program [0..25] 'select u.id, u.email, cs.user_settings, cs.client_id from auth.users u join public.client_settings cs on u.id = cs.user_id;'
-      statement [0..24] 'select u.id, u.email, cs.user_settings, cs.client_id from auth.users u join public.client_settings cs on u.id = cs.user_id'
-        select [0..16] 'select u.id, u.email, cs.user_settings, cs.client_id'
-          keyword_select [0..6] 'select'
-          select_expression [4..16] 'u.id, u.email, cs.user_settings, cs.client_id'
-            term [4..8] 'u.id'
-              object_reference [4..8] 'u.id'
-                any_identifier [4..5] 'u'
-                . [5..6] '.'
-                any_identifier [6..8] 'id'
-            , [8..9] ','
-            term [4..11] 'u.email'
-              object_reference [4..11] 'u.email'
-                any_identifier [4..5] 'u'
-                . [5..6] '.'
-                any_identifier [6..11] 'email'
-            , [11..12] ','
-            term [4..20] 'cs.user_settings'
-              object_reference [4..20] 'cs.user_settings'
-                any_identifier [4..6] 'cs'
-                . [6..7] '.'
-                any_identifier [7..20] 'user_settings'
-            , [20..21] ','
-            term [4..16] 'cs.client_id'
-              object_reference [4..16] 'cs.client_id'
-                any_identifier [4..6] 'cs'
-                . [6..7] '.'
-                any_identifier [7..16] 'client_id'
-        from [0..24] 'from auth.users u join public.client_settings cs on u.id = cs.user_id'
-          keyword_from [0..4] 'from'
-          relation [4..16] 'auth.users u'
-            table_reference [4..14] 'auth.users'
-              schema_identifier [4..8] 'auth'
-              . [8..9] '.'
-              table_identifier [9..14] 'users'
-            alias [15..16] 'u'
-              any_identifier [15..16] 'u'
-          join [4..24] 'join public.client_settings cs on u.id = cs.user_id'
-            keyword_join [4..8] 'join'
-            relation [9..34] 'public.client_settings cs'
-              table_reference [9..31] 'public.client_settings'
-                schema_identifier [9..15] 'public'
-                . [15..16] '.'
-                table_identifier [16..31] 'client_settings'
-              alias [32..34] 'cs'
-                any_identifier [32..34] 'cs'
-            keyword_on [4..6] 'on'
-            binary_expression [7..24] 'u.id = cs.user_id'
-              object_reference [7..11] 'u.id'
-                any_identifier [7..8] 'u'
-                . [8..9] '.'
-                any_identifier [9..11] 'id'
-              = [12..13] '='
-              object_reference [14..24] 'cs.user_id'
-                any_identifier [14..16] 'cs'
-                . [16..17] '.'
-                any_identifier [17..24] 'user_id'
-      ; [24..25] ';'
-    ");
+    file_snapshot("test_3", sql);
 }
 
 #[test]
 fn test_4() {
     let sql = r#"select "auth".REPLACED_TOKEN"#;
 
-    assert_snapshot!(printed_tree(sql), @r#"
-    program [0..28] 'select "auth".REPLACED_TOKEN'
-      statement [0..28] 'select "auth".REPLACED_TOKEN'
-        select [0..28] 'select "auth".REPLACED_TOKEN'
-          keyword_select [0..6] 'select'
-          select_expression [7..28] '"auth".REPLACED_TOKEN'
-            term [7..28] '"auth".REPLACED_TOKEN'
-              object_reference [7..28] '"auth".REPLACED_TOKEN'
-                any_identifier [7..13] '"auth"'
-                . [13..14] '.'
-                any_identifier [14..28] 'REPLACED_TOKEN'
-    "#);
+    file_snapshot("test_4", sql);
 }
 
 #[test]
@@ -206,55 +76,7 @@ fn test_5() {
   where u.active = true and (u.role = 'admin' or u.role = 'moderator');
   "#;
 
-    assert_snapshot!(printed_tree(sql), @r"
-    program [0..71] 'select * from users u where u.active = true and (u.role = 'admin' or u.role = 'moderator');'
-      statement [0..70] 'select * from users u where u.active = true and (u.role = 'admin' or u.role = 'moderator')'
-        select [0..8] 'select *'
-          keyword_select [0..6] 'select'
-          select_expression [7..8] '*'
-            term [7..8] '*'
-              all_fields [7..8] '*'
-                * [7..8] '*'
-        from [9..70] 'from users u where u.active = true and (u.role = 'admin' or u.role = 'moderator')'
-          keyword_from [9..13] 'from'
-          relation [14..21] 'users u'
-            table_reference [14..19] 'users'
-              any_identifier [14..19] 'users'
-            alias [20..21] 'u'
-              any_identifier [20..21] 'u'
-          where [2..70] 'where u.active = true and (u.role = 'admin' or u.role = 'moderator')'
-            keyword_where [2..7] 'where'
-            binary_expression [8..70] 'u.active = true and (u.role = 'admin' or u.role = 'moderator')'
-              binary_expression [8..23] 'u.active = true'
-                object_reference [8..16] 'u.active'
-                  any_identifier [8..9] 'u'
-                  . [9..10] '.'
-                  any_identifier [10..16] 'active'
-                = [17..18] '='
-                literal [19..23] 'true'
-                  keyword_true [19..23] 'true'
-              keyword_and [24..27] 'and'
-              parenthesized_expression [28..70] '(u.role = 'admin' or u.role = 'moderator')'
-                ( [28..29] '('
-                binary_expression [29..69] 'u.role = 'admin' or u.role = 'moderator''
-                  binary_expression [29..45] 'u.role = 'admin''
-                    object_reference [29..35] 'u.role'
-                      any_identifier [29..30] 'u'
-                      . [30..31] '.'
-                      any_identifier [31..35] 'role'
-                    = [36..37] '='
-                    literal [38..45] ''admin''
-                  keyword_or [46..48] 'or'
-                  binary_expression [49..69] 'u.role = 'moderator''
-                    object_reference [49..55] 'u.role'
-                      any_identifier [49..50] 'u'
-                      . [50..51] '.'
-                      any_identifier [51..55] 'role'
-                    = [56..57] '='
-                    literal [58..69] ''moderator''
-                ) [69..70] ')'
-      ; [70..71] ';'
-    ");
+    file_snapshot("test_5", sql);
 }
 
 #[test]
@@ -263,52 +85,5 @@ fn test_6() {
     select (create_composite_type(a, b)).email, (schema.actual_type).id, client from client_settings;
   "#;
 
-    assert_snapshot!(printed_tree(sql), @r"
-    program [0..97] 'select (create_composite_type(a, b)).email, (schema.actual_type).id, client from client_settings;'
-      statement [0..96] 'select (create_composite_type(a, b)).email, (schema.actual_type).id, client from client_settings'
-        select [0..75] 'select (create_composite_type(a, b)).email, (schema.actual_type).id, client'
-          keyword_select [0..6] 'select'
-          select_expression [7..75] '(create_composite_type(a, b)).email, (schema.actual_type).id, client'
-            term [7..42] '(create_composite_type(a, b)).email'
-              field_selection [7..42] '(create_composite_type(a, b)).email'
-                parenthesized_expression [7..36] '(create_composite_type(a, b))'
-                  ( [7..8] '('
-                  invocation [8..35] 'create_composite_type(a, b)'
-                    function_reference [8..29] 'create_composite_type'
-                      any_identifier [8..29] 'create_composite_type'
-                    ( [29..30] '('
-                    term [30..31] 'a'
-                      object_reference [30..31] 'a'
-                        any_identifier [30..31] 'a'
-                    , [31..32] ','
-                    term [33..34] 'b'
-                      object_reference [33..34] 'b'
-                        any_identifier [33..34] 'b'
-                    ) [34..35] ')'
-                  ) [35..36] ')'
-                . [36..37] '.'
-                any_identifier [37..42] 'email'
-            , [42..43] ','
-            term [44..67] '(schema.actual_type).id'
-              field_selection [44..67] '(schema.actual_type).id'
-                composite_reference [44..64] '(schema.actual_type)'
-                  ( [44..45] '('
-                  object_reference [45..63] 'schema.actual_type'
-                    any_identifier [45..51] 'schema'
-                    . [51..52] '.'
-                    any_identifier [52..63] 'actual_type'
-                  ) [63..64] ')'
-                . [64..65] '.'
-                any_identifier [65..67] 'id'
-            , [67..68] ','
-            term [69..75] 'client'
-              object_reference [69..75] 'client'
-                any_identifier [69..75] 'client'
-        from [76..96] 'from client_settings'
-          keyword_from [76..80] 'from'
-          relation [81..96] 'client_settings'
-            table_reference [81..96] 'client_settings'
-              any_identifier [81..96] 'client_settings'
-      ; [96..97] ';'
-    ");
+    file_snapshot("test_6", sql);
 }

crates/pgls_treesitter_grammar/tests/snapshots/grammar_tests__test_1.snap

@@ -0,0 +1,22 @@
+---
+source: crates/pgls_treesitter_grammar/tests/grammar_tests.rs
+expression: writer
+---
+select * from auth.users;
+-----------------------
+program [0..25] 'select * from auth.users;'
+  statement [0..24] 'select * from auth.users'
+    select [0..8] 'select *'
+      keyword_select [0..6] 'select'
+      select_expression [7..8] '*'
+        term [7..8] '*'
+          all_fields [7..8] '*'
+            * [7..8] '*'
+    from [9..24] 'from auth.users'
+      keyword_from [9..13] 'from'
+      relation [14..24] 'auth.users'
+        table_reference [14..24] 'auth.users'
+          schema_identifier [14..18] 'auth'
+          . [18..19] '.'
+          table_identifier [19..24] 'users'
+  ; [24..25] ';'

crates/pgls_treesitter_grammar/tests/snapshots/grammar_tests__test_2_sql1.snap

@@ -0,0 +1,22 @@
+---
+source: crates/pgls_treesitter_grammar/tests/grammar_tests.rs
+expression: writer
+---
+update auth.users set email = '[email protected]';
+-----------------------
+program [0..44] 'update auth.users set email = 'my@mail.com';'
+  statement [0..43] 'update auth.users set email = 'my@mail.com''
+    update [0..43] 'update auth.users set email = 'my@mail.com''
+      keyword_update [0..6] 'update'
+      relation [7..17] 'auth.users'
+        table_reference [7..17] 'auth.users'
+          schema_identifier [7..11] 'auth'
+          . [11..12] '.'
+          table_identifier [12..17] 'users'
+      keyword_set [18..21] 'set'
+      assignment [22..43] 'email = 'my@mail.com''
+        column_reference [22..27] 'email'
+          any_identifier [22..27] 'email'
+        = [28..29] '='
+        literal [30..43] ''my@mail.com''
+  ; [43..44] ';'