Modify tests

tool/tsh/tsh_test.go

@@ -1942,20 +1942,8 @@ func TestKubeCredentialsLock(t *testing.T) {
 			"--teleport-cluster", "teleport",
 			"--kube-cluster", kubeClusterName,
 		}, setHomePath(tmpHomePath))
-		require.Error(t, firstErr) // First run fails because fake proxy doesn't exist
+		require.Error(t, firstErr) // Fails because fake proxy doesn't exist
 		require.NoFileExists(t, keypaths.KubeCredLockfilePath(tmpHomePath, "fake-proxy"))
-
-		// Second run should fail for the same reason, kube cred lockfile shouldn't be created
-		// because we failed at a stage of creating the client.
-		secondErr := Run(ctx, []string{
-			"kube",
-			"credentials",
-			"--proxy", "fake-proxy",
-			"--teleport-cluster", "teleport",
-			"--kube-cluster", kubeClusterName,
-		}, setHomePath(tmpHomePath))
-		require.Error(t, secondErr)
-		require.Equal(t, firstErr.Error(), secondErr.Error())
 	})
 
 	t.Run("kube credentials called multiple times, SSO login called only once", func(t *testing.T) {
@@ -2146,6 +2134,7 @@ func TestSSHHeadless(t *testing.T) {
 	sshHostname := "test-ssh-server"
 	node := makeTestSSHNode(t, authAddr, withHostname(sshHostname), withSSHLabel("access", "true"))
 	sshHostID := node.Config.HostUUID
+
 	hasNodes := func(hostIDs ...string) func() bool {
 		return func() bool {
 			nodes, err := rootAuth.GetAuthServer().GetNodes(ctx, apidefaults.Namespace)