upstream-corteza/server/tests/helpers/labels.go

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
}