Fix TestAuthPlugin2 testcase

pingcap · Nov 26, 2021 · 9cc513d · 9cc513d
1 parent 91a7804
commit 9cc513d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/server/conn_test.go b/server/conn_test.go
@@ -1155,7 +1155,10 @@ func TestAuthPlugin2(t *testing.T) {
 	}
 
 	cc.isUnixSocket = true
-	_, err = cc.checkAuthPlugin(ctx, &resp)
+	require.NoError(t, failpoint.Enable("github.com/pingcap/tidb/server/FakeAuthSwitch", "return(1)"))
+	respAuthSwitch, err := cc.checkAuthPlugin(ctx, &resp)
+	require.NoError(t, failpoint.Disable("github.com/pingcap/tidb/server/FakeAuthSwitch"))
+	require.Equal(t, respAuthSwitch, []byte(mysql.AuthNativePassword))
 	require.NoError(t, err)
 
 }