fix build

pingcap · Jan 3, 2024 · cce19c4 · cce19c4
1 parent dd443a7
commit cce19c4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/session/test/session_test.go b/pkg/session/test/session_test.go
@@ -1015,13 +1015,13 @@ insert into test.t values ("abc"); -- invalid statement
 	session.DisableRunBootstrapSQLFileInTest()
 
 	// Bootstrap with the second sql file, which would not been executed.
-	store, err = mockstore.NewMockStore()
+	store, err = mockstore.NewMockStore(mockstore.WithStoreType(mockstore.EmbedUnistore))
 	require.NoError(t, err)
 	defer func() {
 		require.NoError(t, store.Close())
 	}()
 	config.GetGlobalConfig().InitializeSQLFile = sqlFiles[1].Name()
-	dom, err = session.BootstrapSession(store, mockstore.WithStoreType(mockstore.EmbedUnistore))
+	dom, err = session.BootstrapSession(store)
 	require.NoError(t, err)
 	se := session.CreateSessionAndSetID(t, store)
 	ctx := context.Background()