Function newWorkspaceToolTestServer

coderd/x/chatd/chatd_test.go:229–261

(
	t *testing.T,
	db database.Store,
	ps dbpubsub.Pubsub,
	agentID uuid.UUID,
	planContent string,
)

}

func newWorkspaceToolTestServer(
	t *testing.T,
	db database.Store,
	ps dbpubsub.Pubsub,
	agentID uuid.UUID,
	planContent string,
) *chatd.Server {
	t.Helper()

	ctrl := gomock.NewController(t)
	mockConn := agentconnmock.NewMockAgentConn(ctrl)
	mockConn.EXPECT().SetExtraHeaders(gomock.Any()).AnyTimes()
	mockConn.EXPECT().ContextConfig(gomock.Any()).
		Return(workspacesdk.ContextConfigResponse{}, xerrors.New("not supported")).AnyTimes()
	mockConn.EXPECT().ListMCPTools(gomock.Any()).
		Return(workspacesdk.ListMCPToolsResponse{}, nil).AnyTimes()
	mockConn.EXPECT().LS(gomock.Any(), gomock.Any(), gomock.Any()).
		Return(workspacesdk.LSResponse{}, nil).AnyTimes()
	mockConn.EXPECT().ReadFile(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).
		DoAndReturn(func(_ context.Context, path string, _, _ int64) (io.ReadCloser, string, error) {
			if path == "/home/coder/PLAN.md" {
				return io.NopCloser(strings.NewReader(planContent)), "", nil
			}
			return io.NopCloser(strings.NewReader("")), "", nil
		}).AnyTimes()

	return newActiveTestServer(t, db, ps, func(cfg *chatd.Config) {
		cfg.AgentConn = func(_ context.Context, gotAgentID uuid.UUID) (workspacesdk.AgentConn, func(), error) {
			require.Equal(t, agentID, gotAgentID)
			return mockConn, func() {}, nil
		}
	})
}

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