From 0e2b646390c42013112ee8dff8a8e31bcb8b35da Mon Sep 17 00:00:00 2001
From: Lunny Xiao <xiaolunwen@gmail.com>
Date: Sun, 11 Jul 2021 19:27:37 +0800
Subject: [PATCH] Fix test

---
 integrations/engine_test.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/integrations/engine_test.go b/integrations/engine_test.go
index 48082ce5..a594ee46 100644
--- a/integrations/engine_test.go
+++ b/integrations/engine_test.go
@@ -180,8 +180,12 @@ func TestDumpTables(t *testing.T) {
 
 	importPath := fmt.Sprintf("dump_%v-table.sql", testEngine.Dialect().URI().DBType)
 	t.Run("import_"+importPath, func(t *testing.T) {
-		_, err = testEngine.ImportFile(importPath)
+		sess := testEngine.NewSession()
+		defer sess.Close()
+		assert.NoError(t, sess.Begin())
+		_, err = sess.ImportFile(importPath)
 		assert.NoError(t, err)
+		assert.NoError(t, sess.Commit())
 	})
 }