MCPX MCP Gateway

package testutils import ( "encoding/json" "fmt" "net/url" "strings" public_types "lunar/engine/streams/public-types" streamtypes "lunar/engine/streams/types" json_path "lunar/toolkit-core/json-path" "github.com/rs/zerolog/log" ) type mockAPIStream struct { streamType public_types.StreamType method string Request public_types.TransactionI `json:"request,omitempty"` Response public_types.TransactionI `json:"response,omitempty"` } func NewMockAPIStream( rawURL string, reqHeaders, respHeaders map[string]string, reqBody, respBody string, ) public_types.APIStreamI { return NewMockAPIStreamFull( public_types.StreamTypeRequest, "GET", rawURL, reqHeaders, respHeaders, reqBody, respBody, 200, ) } func NewMockAPIStreamFull( streamType public_types.StreamType, method, rawURL string, reqHeaders, respHeaders map[string]string, reqBody, respBody string, statusCode int, ) public_types.APIStreamI { return &mockAPIStream{ streamType: streamType, method: method, Request: newMockTransaction(method, rawURL, reqHeaders, reqBody, public_types.StreamTypeRequest, 0), Response: newMockTransaction( method, rawURL, respHeaders, respBody, public_types.StreamTypeResponse, statusCode, ), } } func NewMockAPIResponseStream( rawURL string, respHeaders map[string]string, respBody string, statusCode int, ) public_types.APIStreamI { return &mockAPIStream{ streamType: public_types.StreamTypeResponse, Response: newMockTransaction( "GET", rawURL, respHeaders, respBody, public_types.StreamTypeResponse, statusCode, ), } } func (m *mockAPIStream) GetRequest() public_types.TransactionI { return m.Request } func (m *mockAPIStream) GetResponse() public_types.TransactionI { return m.Response } func (m *mockAPIStream) DiscardRequest() {} func (m *mockAPIStream) StoreRequest() {} func (m *mockAPIStream) GetID() string { return "stream-id" } func (m *mockAPIStream) GetSequenceID() string { return "seq-id" } func (m *mockAPIStream) GetType() public_types.StreamType { return m.streamType } func (m *mockAPIStream) GetActionsType() public_types.StreamType { return 0 } func (m *mockAPIStream) GetName() string { return "mock-stream" } func (m *mockAPIStream) GetHeaders() map[string]string { if m.GetType() == public_types.StreamTypeRequest { return m.Request.GetHeaders() } return m.Response.GetHeaders() } func (m *mockAPIStream) GetURL() string { if m.GetType() == public_types.StreamTypeRequest { return m.Request.GetURL() } return m.Response.GetURL() } func (m *mockAPIStream) GetHost() string { if m.GetType() == public_types.StreamTypeRequest { return m.Request.GetHost() } return m.Response.GetHost() } func (m *mockAPIStream) GetBody() string { if m.GetType() == public_types.StreamTypeRequest { return m.Request.GetBody() } return m.Response.GetBody() } func (m *mockAPIStream) GetStrStatus() string { return fmt.Sprintf("%v", m.Response.GetStatus()) } //nolint:lll func (m *mockAPIStream) GetMethod() string { return m.method } func (m *mockAPIStream) GetSize() int { return 1234 } func (m *mockAPIStream) GetHeader(key string) (string, bool) { hdrs := m.GetHeaders() if hdrs == nil { return "", false } val, found := m.GetHeaders()[key] return val, found } func (m *mockAPIStream) DoesHeaderValueMatch(headerName, headerValue string) bool { if existingHeaderValue, found := m.GetHeader(headerName); found { return strings.EqualFold(existingHeaderValue, headerValue) } return false } func (m *mockAPIStream) GetContext() public_types.LunarContextI { return nil } func (m *mockAPIStream) SetRequest(public_types.TransactionI) {} func (m *mockAPIStream) SetResponse(public_types.TransactionI) {} func (m *mockAPIStream) SetContext(public_types.LunarContextI) { } func (m *mockAPIStream) SetType(actionType public_types.StreamType) { m.streamType = actionType } func (m *mockAPIStream) JSONPathQuery(path string) ([]any, error) { wrapper, err := json_path.NewJSONWrapper(m) if err != nil { return nil, err } return wrapper.QueryJSON(path) } func (m *mockAPIStream) JSONPathWrite(string, interface{}) error { return nil } func (m *mockAPIStream) SetActionsType(public_types.StreamType) {} func (m *mockAPIStream) WithLunarContext(public_types.LunarContextI) public_types.APIStreamI { return m } func newMockTransaction( method, rawURL string, headers map[string]string, body string, reqType public_types.StreamType, statusCode int, ) public_types.TransactionI { parsedURL, err := url.Parse(rawURL) if err != nil { return nil } bodyMap := make(map[string]any) if body != "" { if err := json.Unmarshal([]byte(body), &bodyMap); err != nil { log.Error().Err(err).Msg("failed to unmarshal body") } } if reqType == public_types.StreamTypeRequest { return &streamtypes.OnRequest{ Method: method, Scheme: parsedURL.Scheme, ParsedURL: parsedURL, ParsedQuery: parsedURL.Query(), Size: 1234, URL: rawURL, Headers: headers, Body: body, BodyMap: bodyMap, Path: parsedURL.Path, Query: parsedURL.RawQuery, } } return &streamtypes.OnResponse{ URL: rawURL, Headers: headers, Body: body, BodyMap: bodyMap, Status: statusCode, Method: method, Size: 1234, } }