statistics: better code

pingcap · Sep 5, 2023 · be8b49e · be8b49e
1 parent 35af128
commit be8b49e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 11 deletions.
diff --git a/statistics/handle/lockstats/lock_stats_test.go b/statistics/handle/lockstats/lock_stats_test.go
@@ -96,7 +96,7 @@ func TestInsertIntoStatsTableLocked(t *testing.T) {
 	// Executed SQL should be:
 	exec.EXPECT().ExecRestrictedSQL(
 		gomock.Eq(ctx),
-		gomock.Eq(useCurrentSession),
+		useCurrentSession,
 		gomock.Eq(insertSQL),
 		gomock.Eq([]interface{}{int64(1), int64(1)}),
 	)
@@ -123,7 +123,7 @@ func TestAddLockedTables(t *testing.T) {
 	// Executed SQL should be:
 	exec.EXPECT().ExecRestrictedSQL(
 		gomock.All(&ctxMatcher{}),
-		gomock.Eq(useCurrentSession),
+		useCurrentSession,
 		gomock.Eq("BEGIN PESSIMISTIC"),
 	)
 	// Return table 1 is locked.
@@ -132,33 +132,33 @@ func TestAddLockedTables(t *testing.T) {
 	rows := []chunk.Row{c.GetRow(0)}
 	exec.EXPECT().ExecRestrictedSQL(
 		gomock.All(&ctxMatcher{}),
-		gomock.Eq(useCurrentSession),
+		useCurrentSession,
 		selectSQL,
 	).Return(rows, nil, nil)
 
 	exec.EXPECT().ExecRestrictedSQL(
 		gomock.All(&ctxMatcher{}),
-		gomock.Eq(useCurrentSession),
+		useCurrentSession,
 		insertSQL,
 		gomock.Eq([]interface{}{int64(2), int64(2)}),
 	)
 	exec.EXPECT().ExecRestrictedSQL(
 		gomock.All(&ctxMatcher{}),
-		gomock.Eq(useCurrentSession),
+		useCurrentSession,
 		insertSQL,
 		gomock.Eq([]interface{}{int64(3), int64(3)}),
 	)
 
 	exec.EXPECT().ExecRestrictedSQL(
 		gomock.All(&ctxMatcher{}),
-		gomock.Eq(useCurrentSession),
+		useCurrentSession,
 		insertSQL,
 		gomock.Eq([]interface{}{int64(4), int64(4)}),
 	)
 
 	exec.EXPECT().ExecRestrictedSQL(
 		gomock.All(&ctxMatcher{}),
-		gomock.Eq(useCurrentSession),
+		useCurrentSession,
 		"COMMIT",
 	)
 

diff --git a/statistics/handle/lockstats/unlock_stats_test.go b/statistics/handle/lockstats/unlock_stats_test.go
@@ -180,7 +180,7 @@ func TestRemoveLockedTables(t *testing.T) {
 	// Executed SQL should be:
 	exec.EXPECT().ExecRestrictedSQL(
 		gomock.All(&ctxMatcher{}),
-		gomock.Eq(useCurrentSession),
+		useCurrentSession,
 		gomock.Eq("BEGIN PESSIMISTIC"),
 	)
 
@@ -190,7 +190,7 @@ func TestRemoveLockedTables(t *testing.T) {
 	rows := []chunk.Row{c.GetRow(0)}
 	exec.EXPECT().ExecRestrictedSQL(
 		gomock.All(&ctxMatcher{}),
-		gomock.Eq(useCurrentSession),
+		useCurrentSession,
 		selectSQL,
 	).Return(rows, nil, nil)
 
@@ -210,14 +210,14 @@ func TestRemoveLockedTables(t *testing.T) {
 
 	exec.EXPECT().ExecRestrictedSQL(
 		gomock.All(&ctxMatcher{}),
-		gomock.Eq(useCurrentSession),
+		useCurrentSession,
 		deleteLockSQL,
 		gomock.Eq([]interface{}{int64(1)}),
 	).Return(nil, nil, nil)
 
 	exec.EXPECT().ExecRestrictedSQL(
 		gomock.All(&ctxMatcher{}),
-		gomock.Eq(useCurrentSession),
+		useCurrentSession,
 		"COMMIT",
 	)