types_test.go

package jwt_test

import (
"encoding/json"
"math"
"testing"
"time"

"github.com/golang-jwt/jwt/v5"
)

funcTestNumericDate(t*testing.T) {
varsstruct {
Iat jwt.NumericDate`json:"iat"`
Exp jwt.NumericDate`json:"exp"`
 }

oldPrecision:=jwt.TimePrecision

jwt.TimePrecision=time.Microsecond

raw:=`{"iat":1516239022.000000,"exp":1516239022.123450}`

iferr:=json.Unmarshal([]byte(raw), &s); err!=nil {
t.Fatalf("Unexpected error: %s", err)
 }

b, _:=json.Marshal(s)

ifraw!=string(b) {
t.Errorf("Serialized format of numeric date mismatch. Expecting: %s Got: %s", raw, string(b))
 }

jwt.TimePrecision=oldPrecision
}

funcTestSingleArrayMarshal(t*testing.T) {
jwt.MarshalSingleStringAsArray=false

s:= jwt.ClaimStrings{"test"}
expected:=`"test"`

b, err:=json.Marshal(s)
iferr!=nil {
t.Errorf("Unexpected error: %s", err)
 }

ifexpected!=string(b) {
t.Errorf("Serialized format of string array mismatch. Expecting: %s Got: %s", expected, string(b))
 }

jwt.MarshalSingleStringAsArray=true

expected=`["test"]`

b, err=json.Marshal(s)

iferr!=nil {
t.Errorf("Unexpected error: %s", err)
 }

ifexpected!=string(b) {
t.Errorf("Serialized format of string array mismatch. Expecting: %s Got: %s", expected, string(b))
 }
}

funcTestNumericDate_MarshalJSON(t*testing.T) {
// Do not run this test in parallel because it's changing
// global state.
oldPrecision:=jwt.TimePrecision
t.Cleanup(func() {
jwt.TimePrecision=oldPrecision
 })

tt:= []struct {
in time.Time
wantstring
precision time.Duration
 }{
 {time.Unix(5243700879, 0), "5243700879", time.Second},
 {time.Unix(5243700879, 0), "5243700879.000", time.Millisecond},
 {time.Unix(5243700879, 0), "5243700879.000000", time.Microsecond},
 {time.Unix(5243700879, 0), "5243700879.000000000", time.Nanosecond},
//
 {time.Unix(4239425898, 0), "4239425898", time.Second},
 {time.Unix(4239425898, 0), "4239425898.000", time.Millisecond},
 {time.Unix(4239425898, 0), "4239425898.000000", time.Microsecond},
 {time.Unix(4239425898, 0), "4239425898.000000000", time.Nanosecond},
//
 {time.Unix(253402271999, 0), "253402271999", time.Second},
 {time.Unix(253402271999, 0), "253402271999.000", time.Millisecond},
 {time.Unix(253402271999, 0), "253402271999.000000", time.Microsecond},
 {time.Unix(253402271999, 0), "253402271999.000000000", time.Nanosecond},
//
 {time.Unix(0, 1644285000210402000), "1644285000", time.Second},
 {time.Unix(0, 1644285000210402000), "1644285000.210", time.Millisecond},
 {time.Unix(0, 1644285000210402000), "1644285000.210402", time.Microsecond},
 {time.Unix(0, 1644285000210402000), "1644285000.210402000", time.Nanosecond},
//
 {time.Unix(0, 1644285315063096000), "1644285315", time.Second},
 {time.Unix(0, 1644285315063096000), "1644285315.063", time.Millisecond},
 {time.Unix(0, 1644285315063096000), "1644285315.063096", time.Microsecond},
 {time.Unix(0, 1644285315063096000), "1644285315.063096000", time.Nanosecond},
// Maximum time that a go time.Time can represent
 {time.Unix(math.MaxInt64, 999999999), "9223372036854775807", time.Second},
 {time.Unix(math.MaxInt64, 999999999), "9223372036854775807.999", time.Millisecond},
 {time.Unix(math.MaxInt64, 999999999), "9223372036854775807.999999", time.Microsecond},
 {time.Unix(math.MaxInt64, 999999999), "9223372036854775807.999999999", time.Nanosecond},
// Strange precisions
 {time.Unix(math.MaxInt64, 999999999), "9223372036854775807", time.Second},
 {time.Unix(math.MaxInt64, 999999999), "9223372036854775756", time.Minute},
 {time.Unix(math.MaxInt64, 999999999), "9223372036854774016", time.Hour},
 {time.Unix(math.MaxInt64, 999999999), "9223372036854745216", 24*time.Hour},
 }

fori, tc:=rangett {
jwt.TimePrecision=tc.precision
by, err:=jwt.NewNumericDate(tc.in).MarshalJSON()
iferr!=nil {
t.Fatal(err)
 }
ifgot:=string(by); got!=tc.want {
t.Errorf("[%d]: failed encoding: got %q want %q", i, got, tc.want)
 }
 }
}