Source file src/edge-infra.dev/pkg/edge/iam/device/method_test.go

Documentation: edge-infra.dev/pkg/edge/iam/device

     1  package device
     2  
     3  import (
     4  	"testing"
     5  	"time"
     6  
     7  	"gotest.tools/v3/assert"
     8  
     9  	"edge-infra.dev/pkg/edge/iam/profile"
    10  )
    11  
    12  // mock time.Now() for testing.
    13  func mockNow() time.Time {
    14  	return time.Date(2023, 1, 1, 0, 0, 0, 0, time.UTC)
    15  }
    16  func TestSetAgeInProfile(t *testing.T) {
    17  	idClaims := make(map[string]interface{})
    18  	idClaims["dob"] = "2020-06-07"
    19  	var testProfile profile.Profile
    20  	setAgeInProfile(idClaims, &testProfile, mockNow, "")
    21  	assert.Equal(t, testProfile.Age, 2)
    22  
    23  	delete(idClaims, "dob")
    24  	testProfile.Age = 0
    25  	idClaims["birthdate"] = "2020-06-07"
    26  	setAgeInProfile(idClaims, &testProfile, mockNow, "")
    27  	assert.Equal(t, testProfile.Age, 2)
    28  
    29  	delete(idClaims, "birthdate")
    30  	testProfile.Age = 0
    31  	setAgeInProfile(idClaims, &testProfile, mockNow, "")
    32  	assert.Equal(t, testProfile.Age, 0)
    33  }
    34  

View as plain text