dialect/sql/schema: disable foreign keys before opening a transaction

ent · Sep 27, 2022 · f312edd · f312edd
1 parent 8b86528
commit f312edd
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 9 deletions.
diff --git a/dialect/sql/schema/sqlite.go b/dialect/sql/schema/sqlite.go
@@ -44,7 +44,7 @@ func (d *SQLite) Tx(ctx context.Context) (dialect.Tx, error) {
 	if err != nil {
 		return nil, err
 	}
-	tx := &tx{ExecQuerier: t, Tx: t}
+	tx := &tx{t}
 	cm, err := sqlite.CommitFunc(ctx, db, tx, true)
 	return &SQLiteTx{Tx: t, commit: cm, rollback: sqlite.RollbackFunc(ctx, db, tx, true)}, nil
 }

diff --git a/dialect/sql/schema/sqlite_test.go b/dialect/sql/schema/sqlite_test.go
@@ -467,14 +467,6 @@ func (m sqliteMock) commit() {
 		WillReturnResult(sqlmock.NewResult(0, 1))
 }
 
-func (m sqliteMock) rollback() {
-	m.ExpectQuery("PRAGMA foreign_key_check").
-		WillReturnRows(sqlmock.NewRows([]string{})) // empty
-	m.ExpectRollback()
-	m.ExpectExec("PRAGMA foreign_keys = on").
-		WillReturnResult(sqlmock.NewResult(0, 1))
-}
-
 func (m sqliteMock) tableExists(table string, exists bool) {
 	count := 0
 	if exists {