mirror of
https://github.com/pomerium/pomerium.git
synced 2025-05-02 11:56:02 +02:00
70 lines
2.2 KiB
Go
70 lines
2.2 KiB
Go
package manager
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"google.golang.org/protobuf/types/known/structpb"
|
|
"google.golang.org/protobuf/types/known/timestamppb"
|
|
|
|
"github.com/pomerium/pomerium/pkg/grpc/session"
|
|
"github.com/pomerium/pomerium/pkg/grpc/user"
|
|
"github.com/pomerium/pomerium/pkg/protoutil"
|
|
)
|
|
|
|
func TestUser_UnmarshalJSON(t *testing.T) {
|
|
u := new(user.User)
|
|
err := json.Unmarshal([]byte(`{
|
|
"name": "joe",
|
|
"email": "joe@test.com",
|
|
"some-other-claim": "xyz"
|
|
}`), newUserUnmarshaler(u))
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "joe", u.Name)
|
|
assert.Equal(t, "joe@test.com", u.Email)
|
|
assert.Equal(t, map[string]*structpb.ListValue{
|
|
"some-other-claim": {Values: []*structpb.Value{protoutil.ToStruct("xyz")}},
|
|
}, u.Claims)
|
|
}
|
|
|
|
func TestSession_NextRefresh(t *testing.T) {
|
|
tm1 := time.Date(2020, 6, 5, 12, 0, 0, 0, time.UTC)
|
|
s := &session.Session{}
|
|
gracePeriod := time.Second * 10
|
|
coolOffDuration := time.Minute
|
|
assert.Equal(t, tm1.Add(time.Minute), nextSessionRefresh(s, tm1, gracePeriod, coolOffDuration))
|
|
|
|
tm2 := time.Date(2020, 6, 5, 13, 0, 0, 0, time.UTC)
|
|
s.OauthToken = &session.OAuthToken{
|
|
ExpiresAt: timestamppb.New(tm2),
|
|
}
|
|
assert.Equal(t, tm2.Add(-time.Second*10), nextSessionRefresh(s, tm1, gracePeriod, coolOffDuration))
|
|
|
|
tm3 := time.Date(2020, 6, 5, 12, 15, 0, 0, time.UTC)
|
|
s.ExpiresAt = timestamppb.New(tm3)
|
|
assert.Equal(t, tm3, nextSessionRefresh(s, tm1, gracePeriod, coolOffDuration))
|
|
}
|
|
|
|
func TestSession_UnmarshalJSON(t *testing.T) {
|
|
tm := time.Date(2020, 6, 5, 12, 0, 0, 0, time.UTC)
|
|
s := new(session.Session)
|
|
err := json.Unmarshal([]byte(`{
|
|
"iss": "https://some.issuer.com",
|
|
"sub": "subject",
|
|
"exp": `+fmt.Sprint(tm.Unix())+`,
|
|
"iat": `+fmt.Sprint(tm.Unix())+`,
|
|
"some-other-claim": "xyz"
|
|
}`), newSessionUnmarshaler(s))
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, s.IdToken)
|
|
assert.Equal(t, "https://some.issuer.com", s.IdToken.Issuer)
|
|
assert.Equal(t, "subject", s.IdToken.Subject)
|
|
assert.Equal(t, timestamppb.New(tm), s.IdToken.ExpiresAt)
|
|
assert.Equal(t, timestamppb.New(tm), s.IdToken.IssuedAt)
|
|
assert.Equal(t, map[string]*structpb.ListValue{
|
|
"some-other-claim": {Values: []*structpb.Value{protoutil.ToStruct("xyz")}},
|
|
}, s.Claims)
|
|
}
|