Function TestExpMcpReporter

cli/exp_mcp_test.go:680–1262

(t *testing.T)

}

func TestExpMcpReporter(t *testing.T) {
	t.Parallel()

	// Reading to / writing from the PTY is flaky on non-linux systems.
	if runtime.GOOS != "linux" {
		t.Skip("skipping on non-linux")
	}

	t.Run("Error", func(t *testing.T) {
		t.Parallel()

		ctx, cancel := context.WithCancel(testutil.Context(t, testutil.WaitShort))
		socketPath := testutil.AgentSocketPath(t)
		inv, _ := clitest.New(t,
			"exp", "mcp", "server",
			"--socket-path", socketPath,
			"--app-status-slug", "vscode",
			"--ai-agentapi-url", "not a valid url",
		)
		inv = inv.WithContext(ctx)

		pty := ptytest.New(t)
		inv.Stdin = pty.Input()
		inv.Stdout = pty.Output()
		stderr := ptytest.New(t)
		inv.Stderr = stderr.Output()

		cmdDone := make(chan struct{})
		go func() {
			defer close(cmdDone)
			err := inv.Run()
			assert.Error(t, err)
		}()

		stderr.ExpectMatch("Failed to connect to agent socket")
		cancel()
		<-cmdDone
	})

	makeStatusEvent := func(status agentapi.AgentStatus) *codersdk.ServerSentEvent {
		return &codersdk.ServerSentEvent{
			Type: ServerSentEventTypeStatusChange,
			Data: agentapi.EventStatusChange{
				Status: status,
			},
		}
	}

	makeMessageEvent := func(id int64, role agentapi.ConversationRole) *codersdk.ServerSentEvent {
		return &codersdk.ServerSentEvent{
			Type: ServerSentEventTypeMessageUpdate,
			Data: agentapi.EventMessageUpdate{
				Id:   id,
				Role: role,
			},
		}
	}