package session

import (
	"testing"

	"github.com/stretchr/testify/assert"

	"edge-infra.dev/pkg/edge/auth-proxy/faker"
	"edge-infra.dev/pkg/edge/auth-proxy/utils"
)

func TestToUserInstance_AuthUser(t *testing.T) {
	mockUser := faker.MockAuthUser()
	res := ToUserInstance(mockUser)
	assert.NotEmpty(t, res)
	assert.Equal(t, mockUser.Username, res.username)
	assert.Equal(t, mockUser.Email, res.email)
	assert.Equal(t, mockUser.Organization, res.organization)
	assert.Equal(t, utils.FilterEdgeRoles(mockUser.Roles), res.roles)
	assert.Equal(t, bannersToBannerEdgeIDs(mockUser.Banners), res.bannerEdgeIDs)
	assert.Equal(t, mockUser.Token, res.token)
}

func TestToUserInstance_OktaAuth(t *testing.T) {
	mockUser := faker.MockOktaAuth()
	res := ToUserInstance(mockUser)
	assert.NotEmpty(t, res)
	assert.Equal(t, mockUser.Username, res.username)
	assert.Equal(t, mockUser.Email, res.email)
	assert.Equal(t, mockUser.Organization, res.organization)
	assert.Equal(t, utils.FilterEdgeRoles(mockUser.Roles), res.roles)
	assert.Equal(t, bannersToBannerEdgeIDs(mockUser.Banners), res.bannerEdgeIDs)
	assert.Equal(t, mockUser.Token, res.token)
}

func TestToUserInstance_Unknown(t *testing.T) {
	res := ToUserInstance(nil)
	assert.Empty(t, res)
}