fix ut

pingcap · Dec 20, 2021 · 7f237df · 7f237df
1 parent 794f393
commit 7f237df
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/v4/export/sql_test.go b/v4/export/sql_test.go
@@ -1673,9 +1673,10 @@ func TestPickupPossibleField(t *testing.T) {
 			mock.ExpectQuery(query).WillReturnRows(rows)
 		}
 
-		field, err := pickupPossibleField(meta, conn)
+		var field string
+		field, err = pickupPossibleField(meta, conn)
 		if expectedErr != nil {
-			require.ErrorIs(t, err, expectedErr)
+			require.ErrorIs(t, errors.Cause(err), expectedErr)
 		} else {
 			require.NoError(t, err)
 			require.Equal(t, testCase.expectedField, field)