test

package
v0.2.5 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Jun 28, 2025 License: Apache-2.0 Imports: 10 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type MCPError added in v0.2.4

type MCPError struct {
	Code    int         `json:"code"`
	Message string      `json:"message"`
	Data    interface{} `json:"data,omitempty"`
}

MCPError represents an MCP error

type MCPRequest added in v0.2.4

type MCPRequest struct {
	JSONRPC string      `json:"jsonrpc"`
	ID      string      `json:"id"`
	Method  string      `json:"method"`
	Params  interface{} `json:"params,omitempty"`
}

MCPRequest represents an MCP request

type MCPResponse added in v0.2.4

type MCPResponse struct {
	JSONRPC string      `json:"jsonrpc"`
	ID      string      `json:"id"`
	Result  interface{} `json:"result,omitempty"`
	Error   *MCPError   `json:"error,omitempty"`
}

MCPResponse represents an MCP response

type MockClient

type MockClient struct {
	ChatOutput string
	Err        error
}

func NewMockClient

func NewMockClient() *MockClient

func (*MockClient) Chat

func (m *MockClient) Chat(messages []chat.Message, options *chat.ChatOptions) (*chat.Message, chat.ResponseInfo, error)

func (*MockClient) ChatWithFunctions added in v0.0.6

func (c *MockClient) ChatWithFunctions(messages []chat.Message, functions []tools.FunctionConfig, options *chat.ChatOptions) (*chat.Message, chat.ResponseInfo, error)

type MockMCPServer added in v0.2.4

type MockMCPServer struct {
	Server    *httptest.Server
	Responses map[string]interface{} // Predefined responses indexed by method name
	Requests  []MCPRequest           // Record of received requests
	// contains filtered or unexported fields
}

MockMCPServer is a mock MCP server for testing

func NewMockMCPServer added in v0.2.4

func NewMockMCPServer() *MockMCPServer

NewMockMCPServer creates a new mock MCP server

func (*MockMCPServer) ClearRequests added in v0.2.4

func (m *MockMCPServer) ClearRequests()

ClearRequests clears the request history

func (*MockMCPServer) Close added in v0.2.4

func (m *MockMCPServer) Close()

Close closes the server

func (*MockMCPServer) GetLastRequest added in v0.2.4

func (m *MockMCPServer) GetLastRequest() *MCPRequest

GetLastRequest returns the last received request

func (*MockMCPServer) GetRequests added in v0.2.4

func (m *MockMCPServer) GetRequests() []MCPRequest

GetRequests returns all received requests

func (*MockMCPServer) SetDefaultResponses added in v0.2.4

func (m *MockMCPServer) SetDefaultResponses()

SetDefaultResponses sets default MCP responses

func (*MockMCPServer) SetErrorResponse added in v0.2.4

func (m *MockMCPServer) SetErrorResponse(method string, code int, message string)

SetErrorResponse sets an error response for a specific method

func (*MockMCPServer) SetResponse added in v0.2.4

func (m *MockMCPServer) SetResponse(method string, response interface{})

SetResponse sets a specific response for a method

func (*MockMCPServer) URL added in v0.2.4

func (m *MockMCPServer) URL() string

URL returns the server URL

type MockSSEServer added in v0.2.5

type MockSSEServer struct {
	Server    *httptest.Server
	EventChan chan []byte
	Requests  []MCPRequest
	Responses map[string]interface{}
	// contains filtered or unexported fields
}

MockSSEServer is a specialized mock server for SSE testing

func NewMockSSEServer added in v0.2.5

func NewMockSSEServer() *MockSSEServer

NewMockSSEServer creates a new mock SSE server

func (*MockSSEServer) ClearRequests added in v0.2.5

func (m *MockSSEServer) ClearRequests()

ClearRequests clears the request history for SSE server

func (*MockSSEServer) Close added in v0.2.5

func (m *MockSSEServer) Close()

Close closes the SSE server

func (*MockSSEServer) GetRequests added in v0.2.5

func (m *MockSSEServer) GetRequests() []MCPRequest

GetRequests returns all received requests for SSE server

func (*MockSSEServer) SetErrorResponse added in v0.2.5

func (m *MockSSEServer) SetErrorResponse(method string, code int, message string)

SetErrorResponse sets an error response for SSE server

func (*MockSSEServer) SetResponse added in v0.2.5

func (m *MockSSEServer) SetResponse(method string, response interface{})

SetResponse sets a specific response for SSE server

func (*MockSSEServer) URL added in v0.2.5

func (m *MockSSEServer) URL() string

URL returns the SSE server URL

type MockServer

type MockServer struct {
	*httptest.Server

	RequestHandler func(w http.ResponseWriter, r *http.Request)
}

func NewTestServer

func NewTestServer() *MockServer

func (*MockServer) Close

func (m *MockServer) Close()

func (*MockServer) Init

func (m *MockServer) Init()

func (*MockServer) SetRequestHandler

func (m *MockServer) SetRequestHandler(handler func(w http.ResponseWriter, r *http.Request))

func (*MockServer) Start

func (m *MockServer) Start()

func (*MockServer) URL

func (m *MockServer) URL() string

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL