110 lines
2.5 KiB
Go
110 lines
2.5 KiB
Go
package op_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/caos/oidc/pkg/op"
|
|
"github.com/caos/oidc/pkg/op/mock"
|
|
|
|
"github.com/caos/oidc/pkg/oidc"
|
|
)
|
|
|
|
func TestValidateAuthRequest(t *testing.T) {
|
|
type args struct {
|
|
authRequest *oidc.AuthRequest
|
|
storage op.Storage
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
wantErr bool
|
|
}{
|
|
//TODO:
|
|
// {
|
|
// "oauth2 spec"
|
|
// }
|
|
{
|
|
"scope missing fails",
|
|
args{&oidc.AuthRequest{}, nil},
|
|
true,
|
|
},
|
|
{
|
|
"scope openid missing fails",
|
|
args{&oidc.AuthRequest{Scopes: []string{"profile"}}, nil},
|
|
true,
|
|
},
|
|
{
|
|
"response_type missing fails",
|
|
args{&oidc.AuthRequest{Scopes: []string{"openid"}}, nil},
|
|
true,
|
|
},
|
|
{
|
|
"client_id missing fails",
|
|
args{&oidc.AuthRequest{Scopes: []string{"openid"}, ResponseType: oidc.ResponseTypeCode}, nil},
|
|
true,
|
|
},
|
|
{
|
|
"redirect_uri missing fails",
|
|
args{&oidc.AuthRequest{Scopes: []string{"openid"}, ResponseType: oidc.ResponseTypeCode, ClientID: "client_id"}, nil},
|
|
true,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if err := op.ValidateAuthRequest(tt.args.authRequest, tt.args.storage); (err != nil) != tt.wantErr {
|
|
t.Errorf("ValidateAuthRequest() error = %v, wantErr %v", err, tt.wantErr)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestValidateRedirectURI(t *testing.T) {
|
|
type args struct {
|
|
uri string
|
|
clientID string
|
|
storage op.Storage
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
wantErr bool
|
|
}{
|
|
{
|
|
"empty fails",
|
|
args{"", "", nil},
|
|
true,
|
|
},
|
|
{
|
|
"unregistered fails",
|
|
args{"https://unregistered.com/callback", "client_id", mock.NewMockStorageExpectValidClientID(t)},
|
|
true,
|
|
},
|
|
{
|
|
"http not allowed fails",
|
|
args{"http://registered.com/callback", "client_id", mock.NewMockStorageExpectValidClientID(t)},
|
|
true,
|
|
},
|
|
{
|
|
"registered https ok",
|
|
args{"https://registered.com/callback", "client_id", mock.NewMockStorageExpectValidClientID(t)},
|
|
false,
|
|
},
|
|
{
|
|
"registered http allowed ok",
|
|
args{"http://localhost:9999/callback", "client_id", mock.NewMockStorageExpectValidClientID(t)},
|
|
false,
|
|
},
|
|
{
|
|
"registered scheme ok",
|
|
args{"custom://callback", "client_id", mock.NewMockStorageExpectValidClientID(t)},
|
|
false,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if err := op.ValidateRedirectURI(tt.args.uri, tt.args.clientID, tt.args.storage); (err != nil) != tt.wantErr {
|
|
t.Errorf("ValidateRedirectURI() error = %v, wantErr %v", err, tt.wantErr)
|
|
}
|
|
})
|
|
}
|
|
}
|