mockairlock

type Mock struct {
	Server *httptest.Server

	// LLMResponse is the NDJSON response returned by the model endpoint.
	LLMResponse []byte
	// BeforeLLMResponse, when set, runs after the request is recorded and before
	// response headers or events are written.
	BeforeLLMResponse func()
	// EnqueueJobResponse and GetJobResponse override the default job responses.
	EnqueueJobResponse        *wire.EnqueueJobResponse
	EnqueueJobError           *wire.EnqueueJobErrorResponse
	GetJobResponse            *wire.GetJobResponse
	RunCompleteStatus         int
	JobProgressStatus         int
	ConnectorCommandResponses map[string]json.RawMessage
	ConnectorJobResponses     map[string]json.RawMessage
	// contains filtered or unexported fields
}

type Request struct {
	Method string
	Path   string
	Body   []byte
	Header http.Header
}