Add support for automatic query logging.

tests/postgres/main_test.go

@@ -1,10 +1,13 @@
 package postgres
 
 import (
+	"context"
 	"database/sql"
+	"github.com/go-jet/jet/postgres"
 	"github.com/go-jet/jet/tests/dbconfig"
 	_ "github.com/lib/pq"
 	"github.com/pkg/profile"
+	"github.com/stretchr/testify/require"
 	"math/rand"
 	"os"
 	"os/exec"
@@ -43,3 +46,21 @@ func setTestRoot() {
 
 	testRoot = strings.TrimSpace(string(byteArr)) + "/tests/"
 }
+
+var loggedSQL string
+var loggedSQLArgs []interface{}
+var loggedDebugSQL string
+
+func init() {
+	postgres.SetLogger(func(ctx context.Context, statement postgres.LoggableStatement) {
+		loggedSQL, loggedSQLArgs = statement.Sql()
+		loggedDebugSQL = statement.DebugSql()
+	})
+}
+
+func requireLogged(t *testing.T, statement postgres.Statement) {
+	query, args := statement.Sql()
+	require.Equal(t, loggedSQL, query)
+	require.Equal(t, loggedSQLArgs, args)
+	require.Equal(t, loggedDebugSQL, statement.DebugSql())
+}