feat: add support for repo parameter in mock response body
This commit is contained in:
rick
parent
53acdbf8d2
commit
0122bbc6d9
|
|
@ -84,6 +84,7 @@ items:
|
|||
body: |
|
||||
{
|
||||
"count": 1,
|
||||
"repo": "{{.Param.repo}}",
|
||||
"items": [{
|
||||
"title": "fix: there is a bug on page {{ randEnum "one", "two" }}",
|
||||
"number": 123,
|
||||
|
|
|
|||
|
|
@ -16,541 +16,550 @@ limitations under the License.
|
|||
package mock
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/base64"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net"
|
||||
"net/http"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/base64"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net"
|
||||
"net/http"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/swaggest/openapi-go/openapi3"
|
||||
"github.com/swaggest/rest/gorillamux"
|
||||
"github.com/swaggest/openapi-go/openapi3"
|
||||
"github.com/swaggest/rest/gorillamux"
|
||||
|
||||
"github.com/linuxsuren/api-testing/pkg/version"
|
||||
"github.com/linuxsuren/api-testing/pkg/version"
|
||||
|
||||
"github.com/linuxsuren/api-testing/pkg/logging"
|
||||
"github.com/linuxsuren/api-testing/pkg/render"
|
||||
"github.com/linuxsuren/api-testing/pkg/util"
|
||||
"github.com/linuxsuren/api-testing/pkg/logging"
|
||||
"github.com/linuxsuren/api-testing/pkg/render"
|
||||
"github.com/linuxsuren/api-testing/pkg/util"
|
||||
|
||||
"github.com/gorilla/mux"
|
||||
"github.com/gorilla/mux"
|
||||
)
|
||||
|
||||
var (
|
||||
memLogger = logging.DefaultLogger(logging.LogLevelInfo).WithName("memory")
|
||||
memLogger = logging.DefaultLogger(logging.LogLevelInfo).WithName("memory")
|
||||
)
|
||||
|
||||
type inMemoryServer struct {
|
||||
data map[string][]map[string]interface{}
|
||||
mux *mux.Router
|
||||
listener net.Listener
|
||||
port int
|
||||
prefix string
|
||||
wg sync.WaitGroup
|
||||
ctx context.Context
|
||||
cancelFunc context.CancelFunc
|
||||
reader Reader
|
||||
metrics RequestMetrics
|
||||
data map[string][]map[string]interface{}
|
||||
mux *mux.Router
|
||||
listener net.Listener
|
||||
port int
|
||||
prefix string
|
||||
wg sync.WaitGroup
|
||||
ctx context.Context
|
||||
cancelFunc context.CancelFunc
|
||||
reader Reader
|
||||
metrics RequestMetrics
|
||||
}
|
||||
|
||||
func NewInMemoryServer(ctx context.Context, port int) DynamicServer {
|
||||
ctx, cancel := context.WithCancel(ctx)
|
||||
return &inMemoryServer{
|
||||
port: port,
|
||||
wg: sync.WaitGroup{},
|
||||
ctx: ctx,
|
||||
cancelFunc: cancel,
|
||||
metrics: NewNoopMetrics(),
|
||||
}
|
||||
ctx, cancel := context.WithCancel(ctx)
|
||||
return &inMemoryServer{
|
||||
port: port,
|
||||
wg: sync.WaitGroup{},
|
||||
ctx: ctx,
|
||||
cancelFunc: cancel,
|
||||
metrics: NewNoopMetrics(),
|
||||
}
|
||||
}
|
||||
|
||||
func (s *inMemoryServer) SetupHandler(reader Reader, prefix string) (handler http.Handler, err error) {
|
||||
s.reader = reader
|
||||
// init the data
|
||||
s.data = make(map[string][]map[string]interface{})
|
||||
s.mux = mux.NewRouter().PathPrefix(prefix).Subrouter()
|
||||
s.prefix = prefix
|
||||
handler = s.mux
|
||||
s.metrics.AddMetricsHandler(s.mux)
|
||||
err = s.Load()
|
||||
return
|
||||
s.reader = reader
|
||||
// init the data
|
||||
s.data = make(map[string][]map[string]interface{})
|
||||
s.mux = mux.NewRouter().PathPrefix(prefix).Subrouter()
|
||||
s.prefix = prefix
|
||||
handler = s.mux
|
||||
s.metrics.AddMetricsHandler(s.mux)
|
||||
err = s.Load()
|
||||
return
|
||||
}
|
||||
|
||||
func (s *inMemoryServer) Load() (err error) {
|
||||
var server *Server
|
||||
if server, err = s.reader.Parse(); err != nil {
|
||||
return
|
||||
}
|
||||
var server *Server
|
||||
if server, err = s.reader.Parse(); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
memLogger.Info("start to run all the APIs from objects", "count", len(server.Objects))
|
||||
for _, obj := range server.Objects {
|
||||
memLogger.Info("start mock server from object", "name", obj.Name)
|
||||
s.startObject(obj)
|
||||
s.initObjectData(obj)
|
||||
}
|
||||
memLogger.Info("start to run all the APIs from objects", "count", len(server.Objects))
|
||||
for _, obj := range server.Objects {
|
||||
memLogger.Info("start mock server from object", "name", obj.Name)
|
||||
s.startObject(obj)
|
||||
s.initObjectData(obj)
|
||||
}
|
||||
|
||||
memLogger.Info("start to run all the APIs from items", "count", len(server.Items))
|
||||
for _, item := range server.Items {
|
||||
s.startItem(item)
|
||||
}
|
||||
memLogger.Info("start to run all the APIs from items", "count", len(server.Items))
|
||||
for _, item := range server.Items {
|
||||
s.startItem(item)
|
||||
}
|
||||
|
||||
memLogger.Info("start webhook servers", "count", len(server.Webhooks))
|
||||
for _, item := range server.Webhooks {
|
||||
if err = s.startWebhook(&item); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
memLogger.Info("start webhook servers", "count", len(server.Webhooks))
|
||||
for _, item := range server.Webhooks {
|
||||
if err = s.startWebhook(&item); err != nil {
|
||||
continue
|
||||
}
|
||||
}
|
||||
|
||||
s.handleOpenAPI()
|
||||
s.handleOpenAPI()
|
||||
|
||||
for _, proxy := range server.Proxies {
|
||||
memLogger.Info("start to proxy", "target", proxy.Target)
|
||||
s.mux.HandleFunc(proxy.Path, func(w http.ResponseWriter, req *http.Request) {
|
||||
api := fmt.Sprintf("%s/%s", proxy.Target, strings.TrimPrefix(req.URL.Path, s.prefix))
|
||||
api, err = render.Render("proxy api", api, s)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
memLogger.Error(err, "failed to render proxy api")
|
||||
return
|
||||
}
|
||||
memLogger.Info("redirect to", "target", api)
|
||||
for _, proxy := range server.Proxies {
|
||||
memLogger.Info("start to proxy", "target", proxy.Target)
|
||||
s.mux.HandleFunc(proxy.Path, func(w http.ResponseWriter, req *http.Request) {
|
||||
apiRaw := fmt.Sprintf("%s/%s", proxy.Target, strings.TrimPrefix(req.URL.Path, s.prefix))
|
||||
var api string
|
||||
api, err = render.Render("proxy api", apiRaw, s)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
memLogger.Error(err, "failed to render proxy api", "api", apiRaw)
|
||||
return
|
||||
}
|
||||
memLogger.Info("redirect to", "target", api)
|
||||
|
||||
targetReq, err := http.NewRequestWithContext(req.Context(), req.Method, api, req.Body)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
memLogger.Error(err, "failed to create proxy request")
|
||||
return
|
||||
}
|
||||
targetReq, err := http.NewRequestWithContext(req.Context(), req.Method, api, req.Body)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
memLogger.Error(err, "failed to create proxy request")
|
||||
return
|
||||
}
|
||||
|
||||
resp, err := http.DefaultClient.Do(targetReq)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
memLogger.Error(err, "failed to do proxy request")
|
||||
return
|
||||
}
|
||||
resp, err := http.DefaultClient.Do(targetReq)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
memLogger.Error(err, "failed to do proxy request")
|
||||
return
|
||||
}
|
||||
|
||||
data, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
memLogger.Error(err, "failed to read response body")
|
||||
return
|
||||
}
|
||||
data, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
memLogger.Error(err, "failed to read response body")
|
||||
return
|
||||
}
|
||||
|
||||
for k, v := range resp.Header {
|
||||
w.Header().Add(k, v[0])
|
||||
}
|
||||
w.Write(data)
|
||||
})
|
||||
}
|
||||
return
|
||||
for k, v := range resp.Header {
|
||||
w.Header().Add(k, v[0])
|
||||
}
|
||||
w.Write(data)
|
||||
})
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (s *inMemoryServer) Start(reader Reader, prefix string) (err error) {
|
||||
var handler http.Handler
|
||||
if handler, err = s.SetupHandler(reader, prefix); err == nil {
|
||||
if s.listener, err = net.Listen("tcp", fmt.Sprintf(":%d", s.port)); err == nil {
|
||||
go func() {
|
||||
err = http.Serve(s.listener, handler)
|
||||
}()
|
||||
}
|
||||
}
|
||||
return
|
||||
var handler http.Handler
|
||||
if handler, err = s.SetupHandler(reader, prefix); err == nil {
|
||||
if s.listener, err = net.Listen("tcp", fmt.Sprintf(":%d", s.port)); err == nil {
|
||||
go func() {
|
||||
err = http.Serve(s.listener, handler)
|
||||
}()
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (s *inMemoryServer) EnableMetrics() {
|
||||
s.metrics = NewInMemoryMetrics()
|
||||
s.metrics = NewInMemoryMetrics()
|
||||
}
|
||||
|
||||
func (s *inMemoryServer) startObject(obj Object) {
|
||||
// create a simple CRUD server
|
||||
s.mux.HandleFunc("/"+obj.Name, func(w http.ResponseWriter, req *http.Request) {
|
||||
fmt.Println("mock server received request", req.URL.Path)
|
||||
s.metrics.RecordRequest(req.URL.Path)
|
||||
method := req.Method
|
||||
w.Header().Set(util.ContentType, util.JSON)
|
||||
// create a simple CRUD server
|
||||
s.mux.HandleFunc("/"+obj.Name, func(w http.ResponseWriter, req *http.Request) {
|
||||
fmt.Println("mock server received request", req.URL.Path)
|
||||
s.metrics.RecordRequest(req.URL.Path)
|
||||
method := req.Method
|
||||
w.Header().Set(util.ContentType, util.JSON)
|
||||
|
||||
switch method {
|
||||
case http.MethodGet:
|
||||
// list all items
|
||||
allItems := s.data[obj.Name]
|
||||
filteredItems := make([]map[string]interface{}, 0)
|
||||
switch method {
|
||||
case http.MethodGet:
|
||||
// list all items
|
||||
allItems := s.data[obj.Name]
|
||||
filteredItems := make([]map[string]interface{}, 0)
|
||||
|
||||
for i, item := range allItems {
|
||||
exclude := false
|
||||
for i, item := range allItems {
|
||||
exclude := false
|
||||
|
||||
for k, v := range req.URL.Query() {
|
||||
if len(v) == 0 {
|
||||
continue
|
||||
}
|
||||
for k, v := range req.URL.Query() {
|
||||
if len(v) == 0 {
|
||||
continue
|
||||
}
|
||||
|
||||
if val, ok := item[k]; ok && val != v[0] {
|
||||
exclude = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if val, ok := item[k]; ok && val != v[0] {
|
||||
exclude = true
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if !exclude {
|
||||
filteredItems = append(filteredItems, allItems[i])
|
||||
}
|
||||
}
|
||||
if !exclude {
|
||||
filteredItems = append(filteredItems, allItems[i])
|
||||
}
|
||||
}
|
||||
|
||||
if len(filteredItems) != len(allItems) {
|
||||
allItems = filteredItems
|
||||
}
|
||||
if len(filteredItems) != len(allItems) {
|
||||
allItems = filteredItems
|
||||
}
|
||||
|
||||
data, err := json.Marshal(allItems)
|
||||
writeResponse(w, data, err)
|
||||
case http.MethodPost:
|
||||
// create an item
|
||||
if data, err := io.ReadAll(req.Body); err == nil {
|
||||
objData := map[string]interface{}{}
|
||||
data, err := json.Marshal(allItems)
|
||||
writeResponse(w, data, err)
|
||||
case http.MethodPost:
|
||||
// create an item
|
||||
if data, err := io.ReadAll(req.Body); err == nil {
|
||||
objData := map[string]interface{}{}
|
||||
|
||||
jsonErr := json.Unmarshal(data, &objData)
|
||||
if jsonErr != nil {
|
||||
memLogger.Info(jsonErr.Error())
|
||||
return
|
||||
}
|
||||
jsonErr := json.Unmarshal(data, &objData)
|
||||
if jsonErr != nil {
|
||||
memLogger.Info(jsonErr.Error())
|
||||
return
|
||||
}
|
||||
|
||||
s.data[obj.Name] = append(s.data[obj.Name], objData)
|
||||
s.data[obj.Name] = append(s.data[obj.Name], objData)
|
||||
|
||||
_, _ = w.Write(data)
|
||||
} else {
|
||||
memLogger.Info("failed to read from body", "error", err)
|
||||
}
|
||||
default:
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
}
|
||||
})
|
||||
_, _ = w.Write(data)
|
||||
} else {
|
||||
memLogger.Info("failed to read from body", "error", err)
|
||||
}
|
||||
default:
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
}
|
||||
})
|
||||
|
||||
// handle a single object
|
||||
s.mux.HandleFunc(fmt.Sprintf("/%s/{name}", obj.Name), func(w http.ResponseWriter, req *http.Request) {
|
||||
s.metrics.RecordRequest(req.URL.Path)
|
||||
w.Header().Set(util.ContentType, util.JSON)
|
||||
objects := s.data[obj.Name]
|
||||
if objects != nil {
|
||||
name := mux.Vars(req)["name"]
|
||||
var data []byte
|
||||
for _, obj := range objects {
|
||||
if obj["name"] == name {
|
||||
// handle a single object
|
||||
s.mux.HandleFunc(fmt.Sprintf("/%s/{name}", obj.Name), func(w http.ResponseWriter, req *http.Request) {
|
||||
s.metrics.RecordRequest(req.URL.Path)
|
||||
w.Header().Set(util.ContentType, util.JSON)
|
||||
objects := s.data[obj.Name]
|
||||
if objects != nil {
|
||||
name := mux.Vars(req)["name"]
|
||||
var data []byte
|
||||
for _, obj := range objects {
|
||||
if obj["name"] == name {
|
||||
|
||||
data, _ = json.Marshal(obj)
|
||||
break
|
||||
}
|
||||
}
|
||||
data, _ = json.Marshal(obj)
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if len(data) == 0 {
|
||||
w.WriteHeader(http.StatusNotFound)
|
||||
return
|
||||
}
|
||||
if len(data) == 0 {
|
||||
w.WriteHeader(http.StatusNotFound)
|
||||
return
|
||||
}
|
||||
|
||||
method := req.Method
|
||||
switch method {
|
||||
case http.MethodGet:
|
||||
writeResponse(w, data, nil)
|
||||
case http.MethodPut:
|
||||
objData := map[string]interface{}{}
|
||||
if data, err := io.ReadAll(req.Body); err == nil {
|
||||
method := req.Method
|
||||
switch method {
|
||||
case http.MethodGet:
|
||||
writeResponse(w, data, nil)
|
||||
case http.MethodPut:
|
||||
objData := map[string]interface{}{}
|
||||
if data, err := io.ReadAll(req.Body); err == nil {
|
||||
|
||||
jsonErr := json.Unmarshal(data, &objData)
|
||||
if jsonErr != nil {
|
||||
memLogger.Info(jsonErr.Error())
|
||||
return
|
||||
}
|
||||
for i, item := range s.data[obj.Name] {
|
||||
if item["name"] == name {
|
||||
s.data[obj.Name][i] = objData
|
||||
break
|
||||
}
|
||||
}
|
||||
_, _ = w.Write(data)
|
||||
}
|
||||
case http.MethodDelete:
|
||||
for i, item := range s.data[obj.Name] {
|
||||
if item["name"] == name {
|
||||
if len(s.data[obj.Name]) == i+1 {
|
||||
s.data[obj.Name] = s.data[obj.Name][:i]
|
||||
} else {
|
||||
s.data[obj.Name] = append(s.data[obj.Name][:i], s.data[obj.Name][i+1])
|
||||
}
|
||||
jsonErr := json.Unmarshal(data, &objData)
|
||||
if jsonErr != nil {
|
||||
memLogger.Info(jsonErr.Error())
|
||||
return
|
||||
}
|
||||
for i, item := range s.data[obj.Name] {
|
||||
if item["name"] == name {
|
||||
s.data[obj.Name][i] = objData
|
||||
break
|
||||
}
|
||||
}
|
||||
_, _ = w.Write(data)
|
||||
}
|
||||
case http.MethodDelete:
|
||||
for i, item := range s.data[obj.Name] {
|
||||
if item["name"] == name {
|
||||
if len(s.data[obj.Name]) == i+1 {
|
||||
s.data[obj.Name] = s.data[obj.Name][:i]
|
||||
} else {
|
||||
s.data[obj.Name] = append(s.data[obj.Name][:i], s.data[obj.Name][i+1])
|
||||
}
|
||||
|
||||
writeResponse(w, []byte(`{"msg": "deleted"}`), nil)
|
||||
}
|
||||
}
|
||||
default:
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
}
|
||||
writeResponse(w, []byte(`{"msg": "deleted"}`), nil)
|
||||
}
|
||||
}
|
||||
default:
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
}
|
||||
|
||||
}
|
||||
})
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func (s *inMemoryServer) startItem(item Item) {
|
||||
method := util.EmptyThenDefault(item.Request.Method, http.MethodGet)
|
||||
memLogger.Info("register mock service", "method", method, "path", item.Request.Path, "encoder", item.Response.Encoder)
|
||||
method := util.EmptyThenDefault(item.Request.Method, http.MethodGet)
|
||||
memLogger.Info("register mock service", "method", method, "path", item.Request.Path, "encoder", item.Response.Encoder)
|
||||
|
||||
var headerSlices []string
|
||||
for k, v := range item.Request.Header {
|
||||
headerSlices = append(headerSlices, k, v)
|
||||
}
|
||||
var headerSlices []string
|
||||
for k, v := range item.Request.Header {
|
||||
headerSlices = append(headerSlices, k, v)
|
||||
}
|
||||
|
||||
adHandler := &advanceHandler{item: &item, metrics: s.metrics}
|
||||
s.mux.HandleFunc(item.Request.Path, adHandler.handle).Methods(strings.Split(method, ",")...).Headers(headerSlices...)
|
||||
adHandler := &advanceHandler{
|
||||
item: &item,
|
||||
metrics: s.metrics,
|
||||
mu: sync.Mutex{},
|
||||
}
|
||||
s.mux.HandleFunc(item.Request.Path, adHandler.handle).Methods(strings.Split(method, ",")...).Headers(headerSlices...)
|
||||
}
|
||||
|
||||
type advanceHandler struct {
|
||||
item *Item
|
||||
metrics RequestMetrics
|
||||
item *Item
|
||||
metrics RequestMetrics
|
||||
mu sync.Mutex
|
||||
}
|
||||
|
||||
func (h *advanceHandler) handle(w http.ResponseWriter, req *http.Request) {
|
||||
h.metrics.RecordRequest(req.URL.Path)
|
||||
memLogger.Info("receiving mock request", "name", h.item.Name, "method", req.Method, "path", req.URL.Path,
|
||||
"encoder", h.item.Response.Encoder)
|
||||
h.mu.Lock()
|
||||
defer h.mu.Unlock()
|
||||
|
||||
h.item.Param = mux.Vars(req)
|
||||
if h.item.Param == nil {
|
||||
h.item.Param = make(map[string]string)
|
||||
}
|
||||
h.item.Param["Host"] = req.Host
|
||||
if h.item.Response.Header == nil {
|
||||
h.item.Response.Header = make(map[string]string)
|
||||
}
|
||||
h.item.Response.Header[headerMockServer] = fmt.Sprintf("api-testing: %s", version.GetVersion())
|
||||
for k, v := range h.item.Response.Header {
|
||||
hv, hErr := render.Render("mock-server-header", v, &h.item)
|
||||
if hErr != nil {
|
||||
hv = v
|
||||
memLogger.Error(hErr, "failed render mock-server-header", "value", v)
|
||||
}
|
||||
h.metrics.RecordRequest(req.URL.Path)
|
||||
memLogger.Info("receiving mock request", "name", h.item.Name, "method", req.Method, "path", req.URL.Path,
|
||||
"encoder", h.item.Response.Encoder)
|
||||
|
||||
w.Header().Set(k, hv)
|
||||
}
|
||||
w.WriteHeader(util.ZeroThenDefault(h.item.Response.StatusCode, http.StatusOK))
|
||||
h.item.Param = mux.Vars(req)
|
||||
if h.item.Param == nil {
|
||||
h.item.Param = make(map[string]string)
|
||||
}
|
||||
h.item.Param["Host"] = req.Host
|
||||
if h.item.Response.Header == nil {
|
||||
h.item.Response.Header = make(map[string]string)
|
||||
}
|
||||
h.item.Response.Header[headerMockServer] = fmt.Sprintf("api-testing: %s", version.GetVersion())
|
||||
for k, v := range h.item.Response.Header {
|
||||
hv, hErr := render.Render("mock-server-header", v, &h.item)
|
||||
if hErr != nil {
|
||||
hv = v
|
||||
memLogger.Error(hErr, "failed render mock-server-header", "value", v)
|
||||
}
|
||||
|
||||
var err error
|
||||
if h.item.Response.Encoder == "base64" {
|
||||
h.item.Response.BodyData, err = base64.StdEncoding.DecodeString(h.item.Response.Body)
|
||||
} else if h.item.Response.Encoder == "url" {
|
||||
var resp *http.Response
|
||||
if resp, err = http.Get(h.item.Response.Body); err == nil {
|
||||
h.item.Response.BodyData, err = io.ReadAll(resp.Body)
|
||||
}
|
||||
} else {
|
||||
if h.item.Response.BodyData, err = render.RenderAsBytes("start-item", h.item.Response.Body, h.item); err != nil {
|
||||
fmt.Printf("failed to render body: %v", err)
|
||||
}
|
||||
}
|
||||
w.Header().Set(k, hv)
|
||||
}
|
||||
|
||||
if err == nil {
|
||||
h.item.Response.Header[util.ContentLength] = fmt.Sprintf("%d", len(h.item.Response.BodyData))
|
||||
}
|
||||
var err error
|
||||
if h.item.Response.Encoder == "base64" {
|
||||
h.item.Response.BodyData, err = base64.StdEncoding.DecodeString(h.item.Response.Body)
|
||||
} else if h.item.Response.Encoder == "url" {
|
||||
var resp *http.Response
|
||||
if resp, err = http.Get(h.item.Response.Body); err == nil {
|
||||
h.item.Response.BodyData, err = io.ReadAll(resp.Body)
|
||||
}
|
||||
} else {
|
||||
if h.item.Response.BodyData, err = render.RenderAsBytes("start-item", h.item.Response.Body, h.item); err != nil {
|
||||
fmt.Printf("failed to render body: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
writeResponse(w, h.item.Response.BodyData, err)
|
||||
if err == nil {
|
||||
h.item.Response.Header[util.ContentLength] = fmt.Sprintf("%d", len(h.item.Response.BodyData))
|
||||
w.Header().Set(util.ContentLength, h.item.Response.Header[util.ContentLength])
|
||||
}
|
||||
|
||||
writeResponse(w, h.item.Response.BodyData, err)
|
||||
}
|
||||
|
||||
func writeResponse(w http.ResponseWriter, data []byte, err error) {
|
||||
if err == nil {
|
||||
w.Write(data)
|
||||
} else {
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
w.Write([]byte(err.Error()))
|
||||
}
|
||||
if err == nil {
|
||||
w.Write(data)
|
||||
} else {
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
w.Write([]byte(err.Error()))
|
||||
}
|
||||
}
|
||||
|
||||
func (s *inMemoryServer) initObjectData(obj Object) {
|
||||
if obj.Sample == "" {
|
||||
return
|
||||
}
|
||||
if obj.Sample == "" {
|
||||
return
|
||||
}
|
||||
|
||||
defaultCount := 1
|
||||
if obj.InitCount == nil {
|
||||
obj.InitCount = &defaultCount
|
||||
}
|
||||
defaultCount := 1
|
||||
if obj.InitCount == nil {
|
||||
obj.InitCount = &defaultCount
|
||||
}
|
||||
|
||||
for i := 0; i < *obj.InitCount; i++ {
|
||||
objData, jsonErr := jsonStrToInterface(obj.Sample)
|
||||
if jsonErr == nil {
|
||||
s.data[obj.Name] = append(s.data[obj.Name], objData)
|
||||
} else {
|
||||
memLogger.Info(jsonErr.Error())
|
||||
}
|
||||
}
|
||||
for i := 0; i < *obj.InitCount; i++ {
|
||||
objData, jsonErr := jsonStrToInterface(obj.Sample)
|
||||
if jsonErr == nil {
|
||||
s.data[obj.Name] = append(s.data[obj.Name], objData)
|
||||
} else {
|
||||
memLogger.Info(jsonErr.Error())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (s *inMemoryServer) startWebhook(webhook *Webhook) (err error) {
|
||||
if webhook.Timer == "" || webhook.Name == "" {
|
||||
return
|
||||
}
|
||||
if webhook.Timer == "" || webhook.Name == "" {
|
||||
return
|
||||
}
|
||||
|
||||
var duration time.Duration
|
||||
duration, err = time.ParseDuration(webhook.Timer)
|
||||
if err != nil {
|
||||
memLogger.Error(err, "Error parsing webhook timer")
|
||||
return
|
||||
}
|
||||
var duration time.Duration
|
||||
duration, err = time.ParseDuration(webhook.Timer)
|
||||
if err != nil {
|
||||
memLogger.Error(err, "Error parsing webhook timer")
|
||||
return
|
||||
}
|
||||
|
||||
s.wg.Add(1)
|
||||
go func(wh *Webhook) {
|
||||
defer s.wg.Done()
|
||||
s.wg.Add(1)
|
||||
go func(wh *Webhook) {
|
||||
defer s.wg.Done()
|
||||
|
||||
memLogger.Info("start webhook server", "name", wh.Name)
|
||||
timer := time.NewTimer(duration)
|
||||
for {
|
||||
timer.Reset(duration)
|
||||
select {
|
||||
case <-s.ctx.Done():
|
||||
memLogger.Info("stop webhook server", "name", wh.Name)
|
||||
return
|
||||
case <-timer.C:
|
||||
if err = runWebhook(s.ctx, s, wh); err != nil {
|
||||
memLogger.Error(err, "Error when run webhook")
|
||||
}
|
||||
}
|
||||
}
|
||||
}(webhook)
|
||||
return
|
||||
memLogger.Info("start webhook server", "name", wh.Name)
|
||||
timer := time.NewTimer(duration)
|
||||
for {
|
||||
timer.Reset(duration)
|
||||
select {
|
||||
case <-s.ctx.Done():
|
||||
memLogger.Info("stop webhook server", "name", wh.Name)
|
||||
return
|
||||
case <-timer.C:
|
||||
if err = runWebhook(s.ctx, s, wh); err != nil {
|
||||
memLogger.Error(err, "Error when run webhook")
|
||||
}
|
||||
}
|
||||
}
|
||||
}(webhook)
|
||||
return
|
||||
}
|
||||
|
||||
func runWebhook(ctx context.Context, objCtx interface{}, wh *Webhook) (err error) {
|
||||
client := http.DefaultClient
|
||||
client := http.DefaultClient
|
||||
|
||||
var payload io.Reader
|
||||
payload, err = render.RenderAsReader("mock webhook server payload", wh.Request.Body, wh)
|
||||
if err != nil {
|
||||
err = fmt.Errorf("error when render payload: %w", err)
|
||||
return
|
||||
}
|
||||
var payload io.Reader
|
||||
payload, err = render.RenderAsReader("mock webhook server payload", wh.Request.Body, wh)
|
||||
if err != nil {
|
||||
err = fmt.Errorf("error when render payload: %w", err)
|
||||
return
|
||||
}
|
||||
|
||||
method := util.EmptyThenDefault(wh.Request.Method, http.MethodPost)
|
||||
var api string
|
||||
api, err = render.Render("webhook request api", wh.Request.Path, objCtx)
|
||||
if err != nil {
|
||||
err = fmt.Errorf("error when render api: %w", err)
|
||||
return
|
||||
}
|
||||
method := util.EmptyThenDefault(wh.Request.Method, http.MethodPost)
|
||||
var api string
|
||||
api, err = render.Render("webhook request api", wh.Request.Path, objCtx)
|
||||
if err != nil {
|
||||
err = fmt.Errorf("error when render api: %w, template: %s", err, wh.Request.Path)
|
||||
return
|
||||
}
|
||||
|
||||
var bearerToken string
|
||||
bearerToken, err = getBearerToken(ctx, wh.Request)
|
||||
if err != nil {
|
||||
memLogger.Error(err, "Error when render bearer token")
|
||||
return
|
||||
}
|
||||
var bearerToken string
|
||||
bearerToken, err = getBearerToken(ctx, wh.Request)
|
||||
if err != nil {
|
||||
memLogger.Error(err, "Error when render bearer token")
|
||||
return
|
||||
}
|
||||
|
||||
var req *http.Request
|
||||
req, err = http.NewRequestWithContext(ctx, method, api, payload)
|
||||
if err != nil {
|
||||
memLogger.Error(err, "Error when create request")
|
||||
return
|
||||
}
|
||||
var req *http.Request
|
||||
req, err = http.NewRequestWithContext(ctx, method, api, payload)
|
||||
if err != nil {
|
||||
memLogger.Error(err, "Error when create request")
|
||||
return
|
||||
}
|
||||
|
||||
if bearerToken != "" {
|
||||
memLogger.V(7).Info("set bearer token", "token", bearerToken)
|
||||
req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", bearerToken))
|
||||
}
|
||||
if bearerToken != "" {
|
||||
memLogger.V(7).Info("set bearer token", "token", bearerToken)
|
||||
req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", bearerToken))
|
||||
}
|
||||
|
||||
for k, v := range wh.Request.Header {
|
||||
req.Header.Set(k, v)
|
||||
}
|
||||
for k, v := range wh.Request.Header {
|
||||
req.Header.Set(k, v)
|
||||
}
|
||||
|
||||
memLogger.Info("send webhook request", "api", api)
|
||||
resp, err := client.Do(req)
|
||||
if err != nil {
|
||||
err = fmt.Errorf("error when sending webhook")
|
||||
} else {
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
memLogger.Info("unexpected status", "code", resp.StatusCode)
|
||||
}
|
||||
memLogger.Info("send webhook request", "api", api)
|
||||
resp, err := client.Do(req)
|
||||
if err != nil {
|
||||
err = fmt.Errorf("error when sending webhook")
|
||||
} else {
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
memLogger.Info("unexpected status", "code", resp.StatusCode)
|
||||
}
|
||||
|
||||
data, _ := io.ReadAll(resp.Body)
|
||||
memLogger.V(7).Info("received from webhook", "code", resp.StatusCode, "response", string(data))
|
||||
}
|
||||
return
|
||||
data, _ := io.ReadAll(resp.Body)
|
||||
memLogger.V(7).Info("received from webhook", "code", resp.StatusCode, "response", string(data))
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
type bearerToken struct {
|
||||
Token string `json:"token"`
|
||||
Token string `json:"token"`
|
||||
}
|
||||
|
||||
func getBearerToken(ctx context.Context, request RequestWithAuth) (token string, err error) {
|
||||
if request.BearerAPI == "" {
|
||||
return
|
||||
}
|
||||
if request.BearerAPI == "" {
|
||||
return
|
||||
}
|
||||
|
||||
if request.BearerAPI, err = render.Render("bearer token request", request.BearerAPI, &request); err != nil {
|
||||
return
|
||||
}
|
||||
if request.BearerAPI, err = render.Render("bearer token request", request.BearerAPI, &request); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
var data []byte
|
||||
if data, err = json.Marshal(&request); err == nil {
|
||||
client := http.DefaultClient
|
||||
var req *http.Request
|
||||
if req, err = http.NewRequestWithContext(ctx, http.MethodPost, request.BearerAPI, bytes.NewBuffer(data)); err == nil {
|
||||
req.Header.Set(util.ContentType, util.JSON)
|
||||
var data []byte
|
||||
if data, err = json.Marshal(&request); err == nil {
|
||||
client := http.DefaultClient
|
||||
var req *http.Request
|
||||
if req, err = http.NewRequestWithContext(ctx, http.MethodPost, request.BearerAPI, bytes.NewBuffer(data)); err == nil {
|
||||
req.Header.Set(util.ContentType, util.JSON)
|
||||
|
||||
var resp *http.Response
|
||||
if resp, err = client.Do(req); err == nil && resp.StatusCode == http.StatusOK {
|
||||
if data, err = io.ReadAll(resp.Body); err == nil {
|
||||
var tokenObj bearerToken
|
||||
if err = json.Unmarshal(data, &tokenObj); err == nil {
|
||||
token = tokenObj.Token
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
var resp *http.Response
|
||||
if resp, err = client.Do(req); err == nil && resp.StatusCode == http.StatusOK {
|
||||
if data, err = io.ReadAll(resp.Body); err == nil {
|
||||
var tokenObj bearerToken
|
||||
if err = json.Unmarshal(data, &tokenObj); err == nil {
|
||||
token = tokenObj.Token
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return
|
||||
return
|
||||
}
|
||||
|
||||
func (s *inMemoryServer) handleOpenAPI() {
|
||||
s.mux.HandleFunc("/api.json", func(w http.ResponseWriter, req *http.Request) {
|
||||
// Setup OpenAPI schema
|
||||
reflector := openapi3.NewReflector()
|
||||
reflector.SpecSchema().SetTitle("Mock Server API")
|
||||
reflector.SpecSchema().SetVersion(version.GetVersion())
|
||||
reflector.SpecSchema().SetDescription("Powered by https://github.com/linuxsuren/api-testing")
|
||||
s.mux.HandleFunc("/api.json", func(w http.ResponseWriter, req *http.Request) {
|
||||
// Setup OpenAPI schema
|
||||
reflector := openapi3.NewReflector()
|
||||
reflector.SpecSchema().SetTitle("Mock Server API")
|
||||
reflector.SpecSchema().SetVersion(version.GetVersion())
|
||||
reflector.SpecSchema().SetDescription("Powered by https://github.com/linuxsuren/api-testing")
|
||||
|
||||
// Walk the router with OpenAPI collector
|
||||
c := gorillamux.NewOpenAPICollector(reflector)
|
||||
// Walk the router with OpenAPI collector
|
||||
c := gorillamux.NewOpenAPICollector(reflector)
|
||||
|
||||
_ = s.mux.Walk(c.Walker)
|
||||
_ = s.mux.Walk(c.Walker)
|
||||
|
||||
// Get the resulting schema
|
||||
if jsonData, err := reflector.Spec.MarshalJSON(); err == nil {
|
||||
w.Header().Set(util.ContentType, util.JSON)
|
||||
_, _ = w.Write(jsonData)
|
||||
} else {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
_, _ = w.Write([]byte(err.Error()))
|
||||
}
|
||||
})
|
||||
// Get the resulting schema
|
||||
if jsonData, err := reflector.Spec.MarshalJSON(); err == nil {
|
||||
w.Header().Set(util.ContentType, util.JSON)
|
||||
_, _ = w.Write(jsonData)
|
||||
} else {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
_, _ = w.Write([]byte(err.Error()))
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func jsonStrToInterface(jsonStr string) (objData map[string]interface{}, err error) {
|
||||
if jsonStr, err = render.Render("init object", jsonStr, nil); err == nil {
|
||||
objData = map[string]interface{}{}
|
||||
err = json.Unmarshal([]byte(jsonStr), &objData)
|
||||
}
|
||||
return
|
||||
if jsonStr, err = render.Render("init object", jsonStr, nil); err == nil {
|
||||
objData = map[string]interface{}{}
|
||||
err = json.Unmarshal([]byte(jsonStr), &objData)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (s *inMemoryServer) GetPort() string {
|
||||
return util.GetPort(s.listener)
|
||||
return util.GetPort(s.listener)
|
||||
}
|
||||
|
||||
func (s *inMemoryServer) Stop() (err error) {
|
||||
if s.listener != nil {
|
||||
err = s.listener.Close()
|
||||
} else {
|
||||
memLogger.Info("listener is nil")
|
||||
}
|
||||
if s.cancelFunc != nil {
|
||||
s.cancelFunc()
|
||||
}
|
||||
s.wg.Wait()
|
||||
return
|
||||
if s.listener != nil {
|
||||
err = s.listener.Close()
|
||||
} else {
|
||||
memLogger.Info("listener is nil")
|
||||
}
|
||||
if s.cancelFunc != nil {
|
||||
s.cancelFunc()
|
||||
}
|
||||
s.wg.Wait()
|
||||
return
|
||||
}
|
||||
|
|
|
|||
|
|
@ -18,6 +18,7 @@ package mock
|
|||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"strings"
|
||||
|
|
@ -142,7 +143,7 @@ func TestInMemoryServer(t *testing.T) {
|
|||
assert.NoError(t, err)
|
||||
|
||||
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
||||
assert.Equal(t, "176", resp.Header.Get(util.ContentLength))
|
||||
assert.Equal(t, "194", resp.Header.Get(util.ContentLength))
|
||||
assert.Equal(t, "mock", resp.Header.Get("Server"))
|
||||
assert.NotEmpty(t, resp.Header.Get(headerMockServer))
|
||||
|
||||
|
|
@ -223,4 +224,19 @@ func TestInMemoryServer(t *testing.T) {
|
|||
assert.NoError(t, err)
|
||||
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
||||
})
|
||||
|
||||
t.Run("go template support in response body", func(t *testing.T) {
|
||||
repoName := "myRepo"
|
||||
req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("%s/v1/repos/%s/prs", api, repoName), nil)
|
||||
assert.NoError(t, err)
|
||||
|
||||
var resp *http.Response
|
||||
req.Header.Set("name", "rick")
|
||||
resp, err = http.DefaultClient.Do(req)
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
||||
|
||||
data, _ := io.ReadAll(resp.Body)
|
||||
assert.Contains(t, string(data), repoName)
|
||||
})
|
||||
}
|
||||
|
|
|
|||
|
|
@ -41,6 +41,7 @@ items:
|
|||
body: |
|
||||
{
|
||||
"count": 1,
|
||||
"repo": "{{.Param.repo}}",
|
||||
"items": [{
|
||||
"title": "fix: there is a bug on page {{ randEnum "one" }}",
|
||||
"number": 123,
|
||||
|
|
|
|||
Loading…
Reference in New Issue