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

doc: explain how to use policies in migrations (#4141)

Browse Source
This commit is contained in:
Ariel Mashraki
2024-07-14 14:53:22 +03:00
committed by GitHub
parent 1073ce511e
commit 9f61938bcc
37 changed files with 5768 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

87
dialect/sql/driver_test.go Normal file
View File

@@ -0,0 +1,87 @@
// Copyright 2019-present Facebook Inc. All rights reserved.
// This source code is licensed under the Apache 2.0 license found
// in the LICENSE file in the root directory of this source tree.
package sql
import (
"context"
"testing"
"github.com/DATA-DOG/go-sqlmock"
"github.com/stretchr/testify/require"
)
func TestWithVars(t *testing.T) {
db, mock, err := sqlmock.New()
require.NoError(t, err)
db.SetMaxOpenConns(1)
drv := OpenDB("sqlite3", db)
mock.ExpectExec("SET foo = 'bar'").WillReturnResult(sqlmock.NewResult(0, 0))
mock.ExpectQuery("SELECT 1").WillReturnRows(sqlmock.NewRows([]string{"1"}).AddRow(1))
rows := &Rows{}
err = drv.Query(
WithVar(context.Background(), "foo", "bar"),
"SELECT 1",
[]any{},
rows,
)
require.NoError(t, err)
require.NoError(t, mock.ExpectationsWereMet())
require.NoError(t, rows.Close(), "rows should be closed to release the connection")
mock.ExpectExec("SET foo = 'bar'").WillReturnResult(sqlmock.NewResult(0, 0))
mock.ExpectExec("SET foo = 'baz'").WillReturnResult(sqlmock.NewResult(0, 0))
mock.ExpectQuery("SELECT 1").WillReturnRows(sqlmock.NewRows([]string{"1"}).AddRow(1))
err = drv.Query(
WithVar(WithVar(context.Background(), "foo", "bar"), "foo", "baz"),
"SELECT 1",
[]any{},
rows,
)
require.NoError(t, err)
require.NoError(t, mock.ExpectationsWereMet())
require.NoError(t, rows.Close(), "rows should be closed to release the connection")
mock.ExpectBegin()
mock.ExpectExec("SET foo = 'bar'").WillReturnResult(sqlmock.NewResult(0, 0))
mock.ExpectQuery("SELECT 1").WillReturnRows(sqlmock.NewRows([]string{"1"}).AddRow(1))
mock.ExpectCommit()
tx, err := drv.Tx(context.Background())
require.NoError(t, err)
err = tx.Query(
WithVar(context.Background(), "foo", "bar"),
"SELECT 1",
[]any{},
rows,
)
require.NoError(t, err)
require.NoError(t, tx.Commit())
require.NoError(t, mock.ExpectationsWereMet())
// Rows should not be closed to release the session,
// as a transaction is always scoped to a single connection.
mock.ExpectExec("SET foo = 'qux'").WillReturnResult(sqlmock.NewResult(0, 0))
mock.ExpectExec("INSERT INTO users DEFAULT VALUES").WillReturnResult(sqlmock.NewResult(0, 0))
err = drv.Exec(
WithVar(context.Background(), "foo", "qux"),
"INSERT INTO users DEFAULT VALUES",
[]any{},
nil,
)
require.NoError(t, err)
require.NoError(t, mock.ExpectationsWereMet())
// No rows are returned, so no need to close them.
mock.ExpectExec("SET foo = 'foo'").WillReturnResult(sqlmock.NewResult(0, 0))
mock.ExpectExec("INSERT INTO users DEFAULT VALUES").WillReturnResult(sqlmock.NewResult(0, 0))
err = drv.Exec(
WithVar(context.Background(), "foo", "foo"),
"INSERT INTO users DEFAULT VALUES",
[]any{},
nil,
)
require.NoError(t, err)
require.NoError(t, mock.ExpectationsWereMet())
// No rows are returned, so no need to close them.
}