chore(cli): fix test flake in TestExpMcpServer

johnstcn · johnstcn · commit 12405ae64113 · 2025-05-12T16:04:52.000+01:00
diff --git a/cli/exp_mcp_test.go b/cli/exp_mcp_test.go
@@ -148,10 +148,19 @@ func TestExpMcpServer(t *testing.T) {
 		pty := ptytest.New(t)
 		inv.Stdin = pty.Input()
 		inv.Stdout = pty.Output()
+		inv.Stderr = pty.Output()
 		clitest.SetupConfig(t, client, root)
 
-		err := inv.Run()
-		assert.ErrorContains(t, err, "are not logged in")
+		cmdDone := make(chan struct{})
+		go func() {
+			defer close(cmdDone)
+			err := inv.Run()
+			assert.NoError(t, err)
+		}()
+
+		pty.ExpectRegexMatch(`(?is)Authentication\s+:\s+None`)
+		cancel()
+		<-cmdDone
 	})
 }
 
