snapshot of https://github.com/cortezaproject/corteza.git (2024.9.x, 2026-09-18; blobs > 131072B omitted)

server/tests/helpers/labels.go

@@ -0,0 +1,77 @@
+package helpers
+
+import (
+	"context"
+	"net/http"
+	"net/url"
+	"strings"
+	"testing"
+
+	"github.com/cortezaproject/corteza/server/pkg/label/types"
+	"github.com/cortezaproject/corteza/server/store"
+	"github.com/steinfletcher/apitest"
+	"github.com/stretchr/testify/require"
+)
+
+func SetLabelsViaAPI(api *apitest.APITest, t *testing.T, endpoint string, in, out interface{}) {
+	var (
+		payload = struct {
+			Response interface{}
+		}{}
+		req *apitest.Request
+	)
+
+	payload.Response = out
+
+	if strings.HasPrefix(endpoint, "PUT ") {
+		// a little workaround for our inconsistencies...
+		req = api.Put(endpoint[4:])
+	} else {
+		req = api.Post(endpoint)
+	}
+
+	req.JSON(JSON(in)).
+		Header("Accept", "application/json").
+		Expect(t).
+		Status(http.StatusOK).
+		Assert(AssertNoErrors).
+		End().
+		JSON(&payload)
+}
+
+func SearchWithLabelsViaAPI(api *apitest.APITest, t *testing.T, endpoint string, res interface{}, labels url.Values) {
+	var payload = struct{ Response struct{ Set interface{} } }{}
+
+	payload.Response.Set = res
+
+	api.
+		Get(endpoint).
+		QueryCollection(labels).
+		Expect(t).
+		Status(http.StatusOK).
+		Assert(AssertNoErrors).
+		End().
+		JSON(&payload)
+}
+
+func LoadLabelsFromStore(t *testing.T, s store.Storer, resKind string, id uint64) map[string]types.LabelValue {
+	ll, _, err := s.SearchLabels(context.Background(), types.LabelFilter{Kind: resKind})
+	require.NoError(t, err)
+	if len(ll) == 0 {
+		// small adjustment
+		return nil
+	}
+
+	return ll.FilterByResource(resKind, id)
+}
+
+func LabelsToStrings(labels map[string]types.LabelValue) map[string]string {
+	if labels == nil {
+		return nil
+	}
+	result := make(map[string]string, len(labels))
+	for k, v := range labels {
+		result[k] = v.Val
+	}
+	return result
+}