authentication_info_test.go

// Copyright (c) Jeevanandam M. (https://github.com/jeevatkm)
// Source code and usage is governed by a MIT style
// license that can be found in the LICENSE file.

package authc

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestAuthcAuthenticationInfo(t *testing.T) {
	a1 := NewAuthenticationInfo()
	p1 := &Principal{Claim: "Username", Value: "user@sample.com", IsPrimary: true}
	p2 := &Principal{Claim: "ID", Value: "userid"}

	a1.Principals = append(a1.Principals, p1, p2)
	assert.False(t, a1.IsLocked)
	assert.False(t, a1.IsExpired)
	assert.Equal(t, "authenticationinfo([principal(realm: isprimary:true claim:Username value:user@sample.com) "+
		"principal(realm: isprimary:false claim:ID value:userid)] credential:******* islocked:false isexpired:false)", a1.String())

	p := a1.PrimaryPrincipal()
	assert.NotNil(t, p)
	assert.Equal(t, "user@sample.com", p.Value)
	assert.True(t, p.IsPrimary)
	assert.Equal(t, "principal(realm: isprimary:true claim:Username value:user@sample.com)", p.String())

	assert.NotNil(t, a1.Principals)
	assert.True(t, len(a1.Principals) == 2)

	p = a1.Principal("ID")
	assert.Equal(t, "ID", p.Claim)
	assert.Equal(t, "userid", p.Value)
	assert.Nil(t, a1.Principal("Country"))
}

func TestAuthcAuthenticationInfoMerge(t *testing.T) {
	a1 := NewAuthenticationInfo()
	a2 := NewAuthenticationInfo()
	a1.Principals = append(a1.Principals, &Principal{Value: "user@sample.com"})
	a2.IsLocked = true
	a2.IsExpired = true

	a1.Merge(a2)
	assert.True(t, a1.IsLocked)
	assert.True(t, a1.IsExpired)
	assert.Nil(t, a1.PrimaryPrincipal())
}