package iam

import (
	"testing"

	"github.com/stretchr/testify/assert"
	"google.golang.org/api/cloudresourcemanager/v1"
)

var (
	testMembers = []string{"test-1-app@appspot.gserviceaccount.com", "test-2-app@appspot.gserviceaccount.com"}
)

func TestCreatePolicyBinding(t *testing.T) {
	role := "test-role"
	expectedPolicyBinding := &cloudresourcemanager.Binding{
		Role:    role,
		Members: testMembers,
	}
	actualPolicyBinding := CreatePolicyBinding(role, testMembers)
	assert.Equal(t, expectedPolicyBinding, actualPolicyBinding)
}

func TestCreatePolicy(t *testing.T) {
	role := "test-role"
	policyBinding := &cloudresourcemanager.Binding{
		Role:    role,
		Members: testMembers,
	}
	policyBindings := []*cloudresourcemanager.Binding{policyBinding}
	expectedPolicy := &cloudresourcemanager.Policy{
		Bindings: policyBindings,
	}
	actualPolicy := CreatePolicy(policyBindings)
	assert.Equal(t, expectedPolicy, actualPolicy)
}