Preparing v2 API for image service.

cmd/metal-api/internal/service/v1/services/common_test.go

@@ -0,0 +1,96 @@
+package services
+
+import (
+	"encoding/json"
+	"io"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+
+	restful "github.com/emicklei/go-restful/v3"
+	"github.com/metal-stack/metal-lib/zapup"
+	"github.com/metal-stack/security"
+	"github.com/stretchr/testify/require"
+)
+
+type emptyPublisher struct {
+	doPublish func(topic string, data interface{}) error
+}
+
+func (p *emptyPublisher) Publish(topic string, data interface{}) error {
+	if p.doPublish != nil {
+		return p.doPublish(topic, data)
+	}
+	return nil
+}
+
+func (p *emptyPublisher) CreateTopic(topic string) error {
+	return nil
+}
+
+func (p *emptyPublisher) Stop() {}
+
+//nolint:deadcode,unused
+type emptyBody struct{}
+
+func webRequestPut(t *testing.T, service *restful.WebService, user *security.User, request interface{}, path string, response interface{}) int {
+	return webRequest(t, http.MethodPut, service, user, request, path, response)
+}
+
+func webRequestPost(t *testing.T, service *restful.WebService, user *security.User, request interface{}, path string, response interface{}) int {
+	return webRequest(t, http.MethodPost, service, user, request, path, response)
+}
+
+func webRequestDelete(t *testing.T, service *restful.WebService, user *security.User, request interface{}, path string, response interface{}) int {
+	return webRequest(t, http.MethodDelete, service, user, request, path, response)
+}
+
+func webRequestGet(t *testing.T, service *restful.WebService, user *security.User, request interface{}, path string, response interface{}) int {
+	return webRequest(t, http.MethodGet, service, user, request, path, response)
+}
+
+func webRequest(t *testing.T, method string, service *restful.WebService, user *security.User, request interface{}, path string, response interface{}) int {
+	container := restful.NewContainer().Add(service)
+
+	jsonBody, err := json.Marshal(request)
+	require.NoError(t, err)
+	body := io.NopCloser(strings.NewReader(string(jsonBody)))
+	createReq := httptest.NewRequest(method, path, body)
+	createReq.Header.Set("Content-Type", "application/json")
+
+	container.Filter(MockAuth(user))
+
+	w := httptest.NewRecorder()
+	container.ServeHTTP(w, createReq)
+
+	resp := w.Result()
+	defer resp.Body.Close()
+
+	err = json.NewDecoder(resp.Body).Decode(response)
+	require.NoError(t, err)
+	return resp.StatusCode
+}
+
+func MockAuth(user *security.User) restful.FilterFunction {
+	return func(req *restful.Request, resp *restful.Response, chain *restful.FilterChain) {
+		log := zapup.RequestLogger(req.Request)
+		rq := req.Request
+		ctx := security.PutUserInContext(zapup.PutLogger(rq.Context(), log), user)
+		req.Request = rq.WithContext(ctx)
+		chain.ProcessFilter(req, resp)
+	}
+}
+
+type NopPublisher struct {
+}
+
+func (p NopPublisher) Publish(topic string, data interface{}) error {
+	return nil
+}
+
+func (p NopPublisher) CreateTopic(topic string) error {
+	return nil
+}
+
+func (p NopPublisher) Stop() {}