type MockDRPCClient struct {// contains filtered or unexported fields}

funcNewMockDRPCClient¶

func NewMockDRPCClient(ctrl *gomock.Controller) *MockDRPCClient

NewMockDRPCClient creates a new mock instance.

func (*MockDRPCClient)DRPCConn¶

func (m *MockDRPCClient) DRPCConn()drpc.Conn

DRPCConn mocks base method.

func (*MockDRPCClient)EXPECT¶

func (m *MockDRPCClient) EXPECT() *MockDRPCClientMockRecorder

EXPECT returns an object that allows the caller to indicate expected use.

func (*MockDRPCClient)GetMCPServerAccessTokensBatch¶

func (m *MockDRPCClient) GetMCPServerAccessTokensBatch(ctxcontext.Context, in *proto.GetMCPServerAccessTokensBatchRequest) (*proto.GetMCPServerAccessTokensBatchResponse,error)

GetMCPServerAccessTokensBatch mocks base method.

func (*MockDRPCClient)GetMCPServerConfigs¶

func (m *MockDRPCClient) GetMCPServerConfigs(ctxcontext.Context, in *proto.GetMCPServerConfigsRequest) (*proto.GetMCPServerConfigsResponse,error)

GetMCPServerConfigs mocks base method.

func (*MockDRPCClient)IsAuthorized¶

func (m *MockDRPCClient) IsAuthorized(ctxcontext.Context, in *proto.IsAuthorizedRequest) (*proto.IsAuthorizedResponse,error)

IsAuthorized mocks base method.

func (*MockDRPCClient)RecordInterception¶

func (m *MockDRPCClient) RecordInterception(ctxcontext.Context, in *proto.RecordInterceptionRequest) (*proto.RecordInterceptionResponse,error)

RecordInterception mocks base method.

func (*MockDRPCClient)RecordInterceptionEnded¶

func (m *MockDRPCClient) RecordInterceptionEnded(ctxcontext.Context, in *proto.RecordInterceptionEndedRequest) (*proto.RecordInterceptionEndedResponse,error)

RecordInterceptionEnded mocks base method.

func (*MockDRPCClient)RecordPromptUsage¶

func (m *MockDRPCClient) RecordPromptUsage(ctxcontext.Context, in *proto.RecordPromptUsageRequest) (*proto.RecordPromptUsageResponse,error)

RecordPromptUsage mocks base method.

func (*MockDRPCClient)RecordTokenUsage¶

func (m *MockDRPCClient) RecordTokenUsage(ctxcontext.Context, in *proto.RecordTokenUsageRequest) (*proto.RecordTokenUsageResponse,error)

RecordTokenUsage mocks base method.

func (*MockDRPCClient)RecordToolUsage¶

func (m *MockDRPCClient) RecordToolUsage(ctxcontext.Context, in *proto.RecordToolUsageRequest) (*proto.RecordToolUsageResponse,error)

RecordToolUsage mocks base method.

type MockDRPCClientMockRecorder struct {// contains filtered or unexported fields}

func (*MockDRPCClientMockRecorder)DRPCConn¶

func (mr *MockDRPCClientMockRecorder) DRPCConn() *gomock.Call

DRPCConn indicates an expected call of DRPCConn.

func (*MockDRPCClientMockRecorder)GetMCPServerAccessTokensBatch¶

func (mr *MockDRPCClientMockRecorder) GetMCPServerAccessTokensBatch(ctx, inany) *gomock.Call

GetMCPServerAccessTokensBatch indicates an expected call of GetMCPServerAccessTokensBatch.

func (*MockDRPCClientMockRecorder)GetMCPServerConfigs¶

func (mr *MockDRPCClientMockRecorder) GetMCPServerConfigs(ctx, inany) *gomock.Call

GetMCPServerConfigs indicates an expected call of GetMCPServerConfigs.

func (*MockDRPCClientMockRecorder)IsAuthorized¶

func (mr *MockDRPCClientMockRecorder) IsAuthorized(ctx, inany) *gomock.Call

IsAuthorized indicates an expected call of IsAuthorized.

func (*MockDRPCClientMockRecorder)RecordInterception¶

func (mr *MockDRPCClientMockRecorder) RecordInterception(ctx, inany) *gomock.Call

RecordInterception indicates an expected call of RecordInterception.

func (*MockDRPCClientMockRecorder)RecordInterceptionEnded¶

func (mr *MockDRPCClientMockRecorder) RecordInterceptionEnded(ctx, inany) *gomock.Call

RecordInterceptionEnded indicates an expected call of RecordInterceptionEnded.

func (*MockDRPCClientMockRecorder)RecordPromptUsage¶

func (mr *MockDRPCClientMockRecorder) RecordPromptUsage(ctx, inany) *gomock.Call

RecordPromptUsage indicates an expected call of RecordPromptUsage.

func (*MockDRPCClientMockRecorder)RecordTokenUsage¶

func (mr *MockDRPCClientMockRecorder) RecordTokenUsage(ctx, inany) *gomock.Call

RecordTokenUsage indicates an expected call of RecordTokenUsage.

func (*MockDRPCClientMockRecorder)RecordToolUsage¶

func (mr *MockDRPCClientMockRecorder) RecordToolUsage(ctx, inany) *gomock.Call

RecordToolUsage indicates an expected call of RecordToolUsage.

type MockPooler struct {// contains filtered or unexported fields}

funcNewMockPooler¶

func NewMockPooler(ctrl *gomock.Controller) *MockPooler

NewMockPooler creates a new mock instance.

func (*MockPooler)Acquire¶

func (m *MockPooler) Acquire(ctxcontext.Context, reqaibridged.Request, clientFnaibridged.ClientFunc, mcpBootstrapperaibridged.MCPProxyBuilder) (http.Handler,error)

Acquire mocks base method.

func (*MockPooler)EXPECT¶

func (m *MockPooler) EXPECT() *MockPoolerMockRecorder

EXPECT returns an object that allows the caller to indicate expected use.

func (*MockPooler)Shutdown¶

func (m *MockPooler) Shutdown(ctxcontext.Context)error

Shutdown mocks base method.

type MockPoolerMockRecorder struct {// contains filtered or unexported fields}

func (*MockPoolerMockRecorder)Acquire¶

func (mr *MockPoolerMockRecorder) Acquire(ctx, req, clientFn, mcpBootstrapperany) *gomock.Call

Acquire indicates an expected call of Acquire.

func (*MockPoolerMockRecorder)Shutdown¶

func (mr *MockPoolerMockRecorder) Shutdown(ctxany) *gomock.Call

Shutdown indicates an expected call of Shutdown.