QueryRewriter.RewriteQuery now returns an error

https://github.com/jackc/pgx/issues/1186#issuecomment-1288207250
2022-10-29 09:32:35 -05:00
parent 6515e183ff
commit 7d3b9c1e44
4 changed files with 29 additions and 11 deletions
@@ -6,6 +6,7 @@ import (

 	"github.com/jackc/pgx/v5"
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 )

 func TestNamedArgsRewriteQuery(t *testing.T) {
@@ -95,7 +96,8 @@ func TestNamedArgsRewriteQuery(t *testing.T) {

 		// test comments and quotes
 	} {
-		sql, args := tt.namedArgs.RewriteQuery(context.Background(), nil, tt.sql, tt.args)
+		sql, args, err := tt.namedArgs.RewriteQuery(context.Background(), nil, tt.sql, tt.args)
+		require.NoError(t, err)
 		assert.Equalf(t, tt.expectedSQL, sql, "%d", i)
 		assert.Equalf(t, tt.expectedArgs, args, "%d", i)
 	}