golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-05-24 09:31:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

dialect/sql/schema: fix issue with WriteDriver when using Postgres with Query (#3162)

Browse Source
This commit is contained in:
Pedro Henrique
2022-12-09 16:08:55 -03:00
committed by GitHub
parent d7f2f3a098
commit 3f4916ff8b
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
dialect/sql/schema/writer.go
View File

@@ -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:

6
dialect/sql/schema/writer_test.go
View File

@@ -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) {