package gcp

import (
	"testing"

	"github.com/edgelesssys/constellation/bootstrapper/role"
	"github.com/stretchr/testify/assert"
)

func TestExtractRole(t *testing.T) {
	testCases := map[string]struct {
		metadata map[string]string
		wantRole role.Role
	}{
		"bootstrapper role": {
			metadata: map[string]string{
				roleMetadataKey: role.ControlPlane.String(),
			},
			wantRole: role.ControlPlane,
		},
		"node role": {
			metadata: map[string]string{
				roleMetadataKey: role.Worker.String(),
			},
			wantRole: role.Worker,
		},
		"unknown role": {
			metadata: map[string]string{
				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)
		})
	}
}