feat: use postgres regress tests

Author: cvng

crates/parser/tests/skipped.txt

@@ -0,0 +1,213 @@
+advisory_lock
+aggregates
+alter_generic
+alter_operator
+alter_table
+amutils
+arrays
+async
+bit
+bitmapops
+boolean
+box
+brin
+brin_bloom
+brin_multi
+btree_index
+case
+char
+circle
+cluster
+collate.icu.utf8
+collate.linux.utf8
+collate
+combocid
+comments
+compression
+constraints
+conversion
+copy
+copy2
+copydml
+copyselect
+create_aggregate
+create_am
+create_cast
+create_function_c
+create_function_sql
+create_index
+create_index_spgist
+create_misc
+create_operator
+create_procedure
+create_role
+create_table
+create_table_like
+create_type
+create_view
+date
+dbsize
+delete
+dependency
+domain
+drop_if_exists
+drop_operator
+enum
+equivclass
+errors
+event_trigger
+explain
+expressions
+fast_default
+float4
+float8
+foreign_data
+foreign_key
+functional_deps
+generated
+geometry
+gin
+gist
+groupingsets
+guc
+hash_func
+hash_index
+hash_part
+horology
+identity
+incremental_sort
+index_including
+index_including_gist
+indexing
+indirect_toast
+inet
+infinite_recurse
+inherit
+init_privs
+insert
+insert_conflict
+int2
+int4
+int8
+interval
+join
+join_hash
+json
+json_encoding
+jsonb
+jsonb_jsonpath
+jsonpath
+jsonpath_encoding
+largeobject
+limit
+line
+lock
+lseg
+macaddr
+macaddr8
+matview
+memoize
+merge
+misc
+misc_functions
+misc_sanity
+money
+multirangetypes
+mvcc
+name
+namespace
+numeric
+numeric_big
+numerology
+object_address
+oid
+oidjoins
+opr_sanity
+partition_aggregate
+partition_info
+partition_join
+partition_prune
+password
+path
+pg_lsn
+plancache
+plpgsql
+point
+polygon
+polymorphism
+portals
+portals_p2
+prepare
+prepared_xacts
+privileges
+psql
+psql_crosstab
+publication
+random
+rangefuncs
+rangetypes
+regex
+regproc
+reindex_catalog
+reloptions
+replica_identity
+returning
+roleattributes
+rowsecurity
+rowtypes
+rules
+sanity_check
+security_label
+select
+select_distinct
+select_distinct_on
+select_having
+select_implicit
+select_into
+select_parallel
+select_views
+sequence
+spgist
+stats
+stats_ext
+strings
+subscription
+subselect
+sysviews
+tablesample
+tablespace
+temp
+test_setup
+text
+tid
+tidrangescan
+tidscan
+time
+timestamp
+timestamptz
+timetz
+transactions
+triggers
+truncate
+tsdicts
+tsearch
+tsrf
+tstypes
+tuplesort
+txid
+type_sanity
+typed_table
+unicode
+union
+updatable_views
+update
+uuid
+vacuum
+vacuum_parallel
+varchar
+window
+with
+write_parallel
+xid
+xml
+xmlmap

crates/parser/tests/statement_parser_test.rs

@@ -7,51 +7,56 @@ use pg_query::split_with_parser;
 use std::{fs, panic};
 
 const VALID_STATEMENTS_PATH: &str = "tests/data/statements/valid/";
+const POSTGRES_REGRESS_PATH: &str = "../../libpg_query/test/sql/postgres_regress/";
+const SKIPPED_REGRESS_TESTS: &str = include_str!("skipped.txt");
 
 #[test]
 fn valid_statements() {
     common::setup();
 
-    let mut paths: Vec<_> = fs::read_dir(VALID_STATEMENTS_PATH)
-        .unwrap()
-        .map(|r| r.unwrap())
-        .collect();
-    paths.sort_by_key(|dir| dir.path());
+    for path in [VALID_STATEMENTS_PATH, POSTGRES_REGRESS_PATH] {
+        let mut paths: Vec<_> = fs::read_dir(path).unwrap().map(|r| r.unwrap()).collect();
+        paths.sort_by_key(|dir| dir.path());
 
-    for f in paths.iter() {
-        let path = f.path();
-        let test_name = path.file_stem().unwrap().to_str().unwrap();
+        for f in paths.iter() {
+            let path = f.path();
+            let test_name = path.file_stem().unwrap().to_str().unwrap();
 
-        let contents = fs::read_to_string(&path).unwrap();
-        let cases = split_with_parser(&contents).unwrap();
+            if SKIPPED_REGRESS_TESTS.contains(&test_name) {
+                continue;
+            }
 
-        for (i, case) in cases.iter().enumerate() {
-            let case = format!("{};", case.trim());
+            let contents = fs::read_to_string(&path).unwrap();
+            let cases = split_with_parser(&contents).unwrap();
 
-            debug!("Parsing statement {}\n{}", test_name, case);
+            for (i, case) in cases.iter().enumerate() {
+                let case = format!("{};", case.trim());
 
-            let result = panic::catch_unwind(|| parse_source(&case));
+                debug!("Parsing statement {}\n{}", test_name, case);
 
-            if result.is_err() {
-                assert!(false, "Failed to parse statement {}:\n{}", test_name, case);
-            } else {
-                info!(
-                    "Successfully parsed statement {}\n'{}'\n{:#?}",
-                    test_name,
-                    case,
-                    result.as_ref().unwrap().cst
-                );
-            }
+                let result = panic::catch_unwind(|| parse_source(&case));
 
-            let mut settings = Settings::clone_current();
-            settings.set_input_file(&path);
-            settings.set_prepend_module_to_snapshot(false);
-            settings.set_description(case.to_string());
-            settings.set_omit_expression(true);
-            settings.set_snapshot_path("snapshots/statements/valid");
-            settings.set_snapshot_suffix((i + 1).to_string());
+                if result.is_err() {
+                    assert!(false, "Failed to parse statement {}:\n{}", test_name, case);
+                } else {
+                    info!(
+                        "Successfully parsed statement {}\n'{}'\n{:#?}",
+                        test_name,
+                        case,
+                        result.as_ref().unwrap().cst
+                    );
+                }
 
-            settings.bind(|| assert_debug_snapshot!(test_name, result.unwrap()));
+                let mut settings = Settings::clone_current();
+                settings.set_input_file(&path);
+                settings.set_prepend_module_to_snapshot(false);
+                settings.set_description(case.to_string());
+                settings.set_omit_expression(true);
+                settings.set_snapshot_path("snapshots/statements/valid");
+                settings.set_snapshot_suffix((i + 1).to_string());
+
+                settings.bind(|| assert_debug_snapshot!(test_name, result.unwrap()));
+            }
         }
     }
 }