library-go/jwt/claims_test.go

package jwt

import (
	"testing"

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

func TestDataClaimsSettersAndGetters(t *testing.T) {
	tests := []struct {
		name        string
		setterFunc  func(c DataClaims, value string)
		getterFunc  func(c DataClaims) string
		value       string
		expectedVal string
	}{
		{
			name:        "Set and Get ID",
			setterFunc:  func(c DataClaims, value string) { c.SetID(value) },
			getterFunc:  func(c DataClaims) string { return c.ID() },
			value:       "12345",
			expectedVal: "12345",
		},
		{
			name:        "Set and Get Role",
			setterFunc:  func(c DataClaims, value string) { c.SetRole(value) },
			getterFunc:  func(c DataClaims) string { return c.Role() },
			value:       "admin",
			expectedVal: "admin",
		},
		{
			name:        "Set and Get Device ID",
			setterFunc:  func(c DataClaims, value string) { c.SetDeviceID(value) },
			getterFunc:  func(c DataClaims) string { return c.DeviceID() },
			value:       "device123",
			expectedVal: "device123",
		},
		{
			name:        "Set and Get Scope",
			setterFunc:  func(c DataClaims, value string) { c.SetScope(value) },
			getterFunc:  func(c DataClaims) string { return c.Get(scopeCode) },
			value:       "read",
			expectedVal: "read",
		},
		{
			name:        "Set and Get UID",
			setterFunc:  func(c DataClaims, value string) { c.SetUID(value) },
			getterFunc:  func(c DataClaims) string { return c.UID() },
			value:       "user123",
			expectedVal: "user123",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			claims := DataClaims{}

			// Call the setter function
			tt.setterFunc(claims, tt.value)

			// Call the getter function and verify the result
			require.Equal(t, tt.expectedVal, tt.getterFunc(claims), "Expected value does not match")
		})
	}
}

func TestDataClaimsGetNonExistentKey(t *testing.T) {
	claims := DataClaims{}

	// 對於不存在的鍵，應返回空字串
	require.Equal(t, "", claims.Get("nonexistent_key"), "Should return empty string for non-existent key")
}