package device

import (
	"testing"
	"time"

	"gotest.tools/v3/assert"

	"edge-infra.dev/pkg/edge/iam/profile"
)

// mock time.Now() for testing.
func mockNow() time.Time {
	return time.Date(2023, 1, 1, 0, 0, 0, 0, time.UTC)
}
func TestSetAgeInProfile(t *testing.T) {
	idClaims := make(map[string]interface{})
	idClaims["dob"] = "2020-06-07"
	var testProfile profile.Profile
	setAgeInProfile(idClaims, &testProfile, mockNow, "")
	assert.Equal(t, testProfile.Age, 2)

	delete(idClaims, "dob")
	testProfile.Age = 0
	idClaims["birthdate"] = "2020-06-07"
	setAgeInProfile(idClaims, &testProfile, mockNow, "")
	assert.Equal(t, testProfile.Age, 2)

	delete(idClaims, "birthdate")
	testProfile.Age = 0
	setAgeInProfile(idClaims, &testProfile, mockNow, "")
	assert.Equal(t, testProfile.Age, 0)
}