diff --git a/dialect/sql/schema/writer.go b/dialect/sql/schema/writer.go index cecd3b776..ce2917d62 100644 --- a/dialect/sql/schema/writer.go +++ b/dialect/sql/schema/writer.go @@ -110,6 +110,7 @@ func (w *WriteDriver) Query(ctx context.Context, query string, args, res any) er if rr, ok := res.(*sql.Rows); ok { *rr = sql.Rows{ColumnScanner: noRows{}} } + return nil } switch w.Driver.(type) { case nil, nopDriver: diff --git a/dialect/sql/schema/writer_test.go b/dialect/sql/schema/writer_test.go index fd95d4dc4..deaf28179 100644 --- a/dialect/sql/schema/writer_test.go +++ b/dialect/sql/schema/writer_test.go @@ -64,6 +64,12 @@ func TestWriteDriver(t *testing.T) { err = w.Exec(ctx, `INSERT INTO "users" (name) VALUES("a8m") RETURNING id`, nil, nil) require.NoError(t, err) require.Equal(t, `INSERT INTO "users" (name) VALUES("a8m") RETURNING id;`+"\n", b.String()) + + // batchCreator uses tx.Query when doing an insert + b.Reset() + err = w.Query(ctx, `INSERT INTO "users" (name) VALUES("a8m") RETURNING id`, nil, nil) + require.NoError(t, err) + require.Equal(t, `INSERT INTO "users" (name) VALUES("a8m") RETURNING id;`+"\n", b.String()) } func TestDirWriter(t *testing.T) {