@@ -3,8 +3,8 @@ package tables
33import (
44 "testing"
55
6- pg_query "github.com/pganalyze/pg_query_go/v6"
76 "github.com/stretchr/testify/require"
7+ pgquery "github.com/wasilibs/go-pgquery"
88)
99
1010func TestPatternMatcher (t * testing.T ) {
@@ -66,7 +66,7 @@ func TestPatternMatcher(t *testing.T) {
6666 t .Run (tc .name , func (t * testing.T ) {
6767 t .Parallel ()
6868
69- parsed , err := pg_query .Parse (tc .query )
69+ parsed , err := pgquery .Parse (tc .query )
7070 require .NoError (t , err )
7171
7272 selectStmt := parsed .Stmts [0 ].Stmt .GetSelectStmt ()
@@ -134,7 +134,7 @@ func TestEqualityPattern(t *testing.T) {
134134 t .Run (tc .name , func (t * testing.T ) {
135135 t .Parallel ()
136136
137- parsed , err := pg_query .Parse (tc .query )
137+ parsed , err := pgquery .Parse (tc .query )
138138 require .NoError (t , err )
139139
140140 selectStmt := parsed .Stmts [0 ].Stmt .GetSelectStmt ()
@@ -187,7 +187,7 @@ func TestSubQueryPlaceholderPattern(t *testing.T) {
187187 t .Run (tc .name , func (t * testing.T ) {
188188 t .Parallel ()
189189
190- parsed , err := pg_query .Parse (tc .query )
190+ parsed , err := pgquery .Parse (tc .query )
191191 require .NoError (t , err )
192192
193193 selectStmt := parsed .Stmts [0 ].Stmt .GetSelectStmt ()
@@ -240,7 +240,7 @@ func TestColumnRefPattern(t *testing.T) {
240240 t .Run (tc .name , func (t * testing.T ) {
241241 t .Parallel ()
242242
243- parsed , err := pg_query .Parse (tc .query )
243+ parsed , err := pgquery .Parse (tc .query )
244244 require .NoError (t , err )
245245
246246 selectStmt := parsed .Stmts [0 ].Stmt .GetSelectStmt ()
@@ -266,7 +266,7 @@ func TestColumnRefPattern(t *testing.T) {
266266// BenchmarkPatternMatcher compares the performance of pattern matching vs old approach
267267func BenchmarkPatternMatcher (b * testing.B ) {
268268 query := "SELECT * FROM relationships WHERE resource_type = 'document' AND resource_id = 'doc1' AND relation = 'viewer'"
269- parsed , err := pg_query .Parse (query )
269+ parsed , err := pgquery .Parse (query )
270270 require .NoError (b , err )
271271
272272 selectStmt := parsed .Stmts [0 ].Stmt .GetSelectStmt ()
0 commit comments