constellation/coordinator/cloudprovider/role_test.go

package cloudprovider

import (
	"testing"

	"github.com/edgelesssys/constellation/coordinator/core"
	"github.com/edgelesssys/constellation/coordinator/role"
	"github.com/stretchr/testify/assert"
	"go.uber.org/goleak"
)

func TestMain(m *testing.M) {
	goleak.VerifyTestMain(m,
		// https://github.com/census-instrumentation/opencensus-go/issues/1262
		goleak.IgnoreTopFunction("go.opencensus.io/stats/view.(*worker).start"),
	)
}

func TestExtractRole(t *testing.T) {
	testCases := map[string]struct {
		metadata map[string]string
		wantRole role.Role
	}{
		"coordinator role": {
			metadata: map[string]string{
				core.RoleMetadataKey: role.Coordinator.String(),
			},
			wantRole: role.Coordinator,
		},
		"node role": {
			metadata: map[string]string{
				core.RoleMetadataKey: role.Node.String(),
			},
			wantRole: role.Node,
		},
		"unknown role": {
			metadata: map[string]string{
				core.RoleMetadataKey: "some-unknown-role",
			},
			wantRole: role.Unknown,
		},
		"no role": {
			wantRole: role.Unknown,
		},
	}

	for name, tc := range testCases {
		t.Run(name, func(t *testing.T) {
			assert := assert.New(t)

			role := ExtractRole(tc.metadata)

			assert.Equal(tc.wantRole, role)
		})
	}
}