mock

package

v1.0.0-alpha.1 Latest Latest Go to latest Published: Mar 4, 2026 License: MIT Imports: 11 Imported by: 0

Details

Valid go.mod file
Redistributable license
Tagged version
Stable version
Learn more about best practices

Repository

github.com/c360studio/semstreams

Links

Open Source Insights

Documentation ¶

Overview ¶

Package mock provides mock servers for E2E testing.

Package mock provides test doubles for external services.

Constants ¶

This section is empty.

Variables ¶

This section is empty.

Functions ¶

This section is empty.

Types ¶

type AGNTCYServer ¶

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

AGNTCYServer provides mock endpoints for AGNTCY integration testing. It handles: - Directory registration (/v1/agents/register, /v1/agents/heartbeat) - OTEL HTTP traces (/v1/traces) - Health checks (/health)

func NewAGNTCYServer ¶

func NewAGNTCYServer() *AGNTCYServer

NewAGNTCYServer creates a new mock AGNTCY server.

func (*AGNTCYServer) GetRegistrations ¶

func (s *AGNTCYServer) GetRegistrations() map[string]*AgentRegistration

GetRegistrations returns all agent registrations.

func (*AGNTCYServer) MetricsReceived ¶

func (s *AGNTCYServer) MetricsReceived() int64

MetricsReceived returns the number of metric exports received.

func (*AGNTCYServer) Start ¶

func (s *AGNTCYServer) Start(addr string) error

Start starts the server on the given address.

func (*AGNTCYServer) Stats ¶

func (s *AGNTCYServer) Stats() map[string]any

Stats returns server statistics.

func (*AGNTCYServer) Stop ¶

func (s *AGNTCYServer) Stop() error

Stop stops the server.

func (*AGNTCYServer) TracesReceived ¶

func (s *AGNTCYServer) TracesReceived() int64

TracesReceived returns the number of trace exports received.

func (*AGNTCYServer) URL ¶

func (s *AGNTCYServer) URL() string

URL returns the server URL.

type AgentRegistration ¶

type AgentRegistration struct {
	AgentID       string         `json:"agent_id"`
	OASFRecord    map[string]any `json:"oasf_record"`
	RegisteredAt  time.Time      `json:"registered_at"`
	LastHeartbeat time.Time      `json:"last_heartbeat"`
	TTL           string         `json:"ttl"`
}

AgentRegistration represents a registered agent.

type ChatCompletionRequest ¶

type ChatCompletionRequest struct {
	Model       string        `json:"model"`
	Messages    []ChatMessage `json:"messages"`
	Tools       []Tool        `json:"tools,omitempty"`
	MaxTokens   int           `json:"max_tokens,omitempty"`
	Temperature float32       `json:"temperature,omitempty"`
}

ChatCompletionRequest matches OpenAI API request format.

type ChatCompletionResponse ¶

type ChatCompletionResponse struct {
	ID      string   `json:"id"`
	Object  string   `json:"object"`
	Created int64    `json:"created"`
	Model   string   `json:"model"`
	Choices []Choice `json:"choices"`
	Usage   Usage    `json:"usage"`
}

ChatCompletionResponse matches OpenAI API response format.

type ChatMessage ¶

type ChatMessage struct {
	Role       string     `json:"role"`
	Content    string     `json:"content,omitempty"`
	ToolCalls  []ToolCall `json:"tool_calls,omitempty"`
	ToolCallID string     `json:"tool_call_id,omitempty"`
}

ChatMessage matches OpenAI API message format.

type Choice ¶

type Choice struct {
	Index        int         `json:"index"`
	Message      ChatMessage `json:"message"`
	FinishReason string      `json:"finish_reason"`
}

Choice matches OpenAI API choice format.

type FunctionCall ¶

type FunctionCall struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
}

FunctionCall matches OpenAI API function call format.

type FunctionDef ¶

type FunctionDef struct {
	Name        string `json:"name"`
	Description string `json:"description"`
	Parameters  any    `json:"parameters"`
}

FunctionDef matches OpenAI API function definition.

type OpenAIServer ¶

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

OpenAIServer is a mock OpenAI-compatible server for testing.

func NewOpenAIServer ¶

func NewOpenAIServer() *OpenAIServer

NewOpenAIServer creates a new mock OpenAI server.

func (*OpenAIServer) Addr ¶

func (s *OpenAIServer) Addr() string

Addr returns the address the server is listening on.

func (*OpenAIServer) LastRequest ¶

func (s *OpenAIServer) LastRequest() *ChatCompletionRequest

LastRequest returns the last request received (for assertions).

func (*OpenAIServer) RequestCount ¶

func (s *OpenAIServer) RequestCount() int

RequestCount returns the number of requests received.

func (*OpenAIServer) ResetSequence ¶

func (s *OpenAIServer) ResetSequence()

ResetSequence resets the response sequence to the beginning.

func (*OpenAIServer) SequenceIndex ¶

func (s *OpenAIServer) SequenceIndex() int

SequenceIndex returns the current position in the response sequence.

func (*OpenAIServer) Start ¶

func (s *OpenAIServer) Start(addr string) error

Start starts the mock server on the given address. If addr is empty or ":0", a random available port is used.

func (*OpenAIServer) Stop ¶

func (s *OpenAIServer) Stop() error

Stop stops the mock server gracefully.

func (*OpenAIServer) URL ¶

func (s *OpenAIServer) URL() string

URL returns the base URL for the server.

func (*OpenAIServer) WithCompletionContent ¶

func (s *OpenAIServer) WithCompletionContent(content string) *OpenAIServer

WithCompletionContent configures the content returned on completion.

func (*OpenAIServer) WithRequestDelay ¶

func (s *OpenAIServer) WithRequestDelay(d time.Duration) *OpenAIServer

WithRequestDelay configures an artificial delay per request.

func (*OpenAIServer) WithResponseSequence ¶

func (s *OpenAIServer) WithResponseSequence(responses []string) *OpenAIServer

WithResponseSequence configures a sequence of completion contents. Each call to the chat completion endpoint will return the next response in the sequence. After the sequence is exhausted, it returns the last response.

func (*OpenAIServer) WithToolArgs ¶

func (s *OpenAIServer) WithToolArgs(toolName, argsJSON string) *OpenAIServer

WithToolArgs configures the arguments returned for a specific tool.

type Tool ¶

type Tool struct {
	Type     string      `json:"type"`
	Function FunctionDef `json:"function"`
}

Tool matches OpenAI API tool format.

type ToolCall ¶

type ToolCall struct {
	ID       string       `json:"id"`
	Type     string       `json:"type"`
	Function FunctionCall `json:"function"`
}

ToolCall matches OpenAI API tool call format.

type Usage ¶

type Usage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}

Usage matches OpenAI API usage format.

Source Files ¶

View all Source files

Directories ¶

Path	Synopsis
cmd Package main provides the standalone mock servers for E2E testing.	Package main provides the standalone mock servers for E2E testing.

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