fix: add breaking test to client validator

ory · Aug 11, 2023 · ffe450f · ffe450f
1 parent 7ca5f8b
commit ffe450f
Showing 1 changed file with 75 additions and 13 deletions.
diff --git a/client/validator_test.go b/client/validator_test.go
@@ -5,13 +5,16 @@ package client_test
 
 import (
 	"context"
+	"encoding/json"
 	"fmt"
 	"net/http"
 	"net/http/httptest"
+	"strings"
 	"testing"
 
 	"github.com/hashicorp/go-retryablehttp"
 
+	"github.com/ory/fosite"
 	"github.com/ory/hydra/v2/driver"
 	"github.com/ory/x/httpx"
 
@@ -38,11 +41,16 @@ func TestValidate(t *testing.T) {
 
 	testCtx := context.TODO()
 
+	dec := json.NewDecoder(strings.NewReader(validJWKS))
+	dec.DisallowUnknownFields()
+	var goodJWKS jose.JSONWebKeySet
+	require.NoError(t, dec.Decode(&goodJWKS))
+
 	for k, tc := range []struct {
 		in        *Client
-		check     func(t *testing.T, c *Client)
-		expectErr bool
-		v         func(t *testing.T) *Validator
+		check     func(*testing.T, *Client)
+		assertErr func(t assert.TestingT, err error, msg ...interface{}) bool
+		v         func(*testing.T) *Validator
 	}{
 		{
 			in: new(Client),
@@ -66,31 +74,57 @@ func TestValidate(t *testing.T) {
 		},
 		{
 			in:        &Client{LegacyClientID: "foo", UserinfoSignedResponseAlg: "foo"},
-			expectErr: true,
+			assertErr: assert.Error,
 		},
 		{
 			in:        &Client{LegacyClientID: "foo", TokenEndpointAuthMethod: "private_key_jwt"},
-			expectErr: true,
+			assertErr: assert.Error,
 		},
 		{
 			in:        &Client{LegacyClientID: "foo", JSONWebKeys: &x.JoseJSONWebKeySet{JSONWebKeySet: new(jose.JSONWebKeySet)}, JSONWebKeysURI: "asdf", TokenEndpointAuthMethod: "private_key_jwt"},
-			expectErr: true,
+			assertErr: assert.Error,
 		},
 		{
 			in:        &Client{LegacyClientID: "foo", JSONWebKeys: &x.JoseJSONWebKeySet{JSONWebKeySet: new(jose.JSONWebKeySet)}, TokenEndpointAuthMethod: "private_key_jwt", TokenEndpointAuthSigningAlgorithm: "HS256"},
-			expectErr: true,
+			assertErr: assert.Error,
+		},
+		{
+			in: &Client{LegacyClientID: "foo", JSONWebKeys: &x.JoseJSONWebKeySet{JSONWebKeySet: new(jose.JSONWebKeySet)}, JSONWebKeysURI: "https://example.org/jwks.json"},
+			assertErr: func(_ assert.TestingT, err error, msg ...interface{}) bool {
+				e := new(fosite.RFC6749Error)
+				assert.ErrorAs(t, err, &e)
+				assert.Contains(t, e.HintField, "jwks and jwks_uri can not both be set")
+				return true
+			},
+		},
+		{
+			in: &Client{LegacyClientID: "foo", JSONWebKeys: &x.JoseJSONWebKeySet{JSONWebKeySet: &goodJWKS}},
+			check: func(t *testing.T, c *Client) {
+				assert.Len(t, c.JSONWebKeys.Keys, 2)
+				assert.Equal(t, c.JSONWebKeys.Keys[0].KeyID, "1")
+				assert.Equal(t, c.JSONWebKeys.Keys[1].KeyID, "2011-04-29")
+			},
+		},
+		{
+			in: &Client{LegacyClientID: "foo", JSONWebKeys: &x.JoseJSONWebKeySet{JSONWebKeySet: &jose.JSONWebKeySet{Keys: []jose.JSONWebKey{{}}}}},
+			assertErr: func(_ assert.TestingT, err error, msg ...interface{}) bool {
+				e := new(fosite.RFC6749Error)
+				assert.ErrorAs(t, err, &e)
+				assert.Contains(t, e.HintField, "Invalid JSON web key in set")
+				return true
+			},
 		},
 		{
 			in:        &Client{LegacyClientID: "foo", PostLogoutRedirectURIs: []string{"https://bar/"}, RedirectURIs: []string{"https://foo/"}},
-			expectErr: true,
+			assertErr: assert.Error,
 		},
 		{
 			in:        &Client{LegacyClientID: "foo", PostLogoutRedirectURIs: []string{"http://foo/"}, RedirectURIs: []string{"https://foo/"}},
-			expectErr: true,
+			assertErr: assert.Error,
 		},
 		{
 			in:        &Client{LegacyClientID: "foo", PostLogoutRedirectURIs: []string{"https://foo:1234/"}, RedirectURIs: []string{"https://foo/"}},
-			expectErr: true,
+			assertErr: assert.Error,
 		},
 		{
 			in: &Client{LegacyClientID: "foo", PostLogoutRedirectURIs: []string{"https://foo/"}, RedirectURIs: []string{"https://foo/"}},
@@ -122,18 +156,19 @@ func TestValidate(t *testing.T) {
 		},
 		{
 			in:        &Client{LegacyClientID: "foo", SubjectType: "foo"},
-			expectErr: true,
+			assertErr: assert.Error,
 		},
 	} {
+		tc := tc
 		t.Run(fmt.Sprintf("case=%d", k), func(t *testing.T) {
 			if tc.v == nil {
 				tc.v = func(t *testing.T) *Validator {
 					return v
 				}
 			}
 			err := tc.v(t).Validate(testCtx, tc.in)
-			if tc.expectErr {
-				require.Error(t, err)
+			if tc.assertErr != nil {
+				tc.assertErr(t, err)
 			} else {
 				require.NoError(t, err)
 				tc.check(t, tc.in)
@@ -205,6 +240,33 @@ func TestValidateSectorIdentifierURL(t *testing.T) {
 	}
 }
 
+// from https://datatracker.ietf.org/doc/html/rfc7517#appendix-A.2
+const validJWKS = `
+{"keys":
+[
+  {"kty":"EC",
+   "crv":"P-256",
+   "x":"MKBCTNIcKUSDii11ySs3526iDZ8AiTo7Tu6KPAqv7D4",
+   "y":"4Etl6SRW2YiLUrN5vfvVHuhp7x8PxltmWWlbbM4IFyM",
+   "d":"870MB6gfuTJ4HtUnUvYMyJpr5eUZNP4Bk43bVdj3eAE",
+   "use":"enc",
+   "kid":"1"},
+
+  {"kty":"RSA",
+   "n":"0vx7agoebGcQSuuPiLJXZptN9nndrQmbXEps2aiAFbWhM78LhWx4cbbfAAtVT86zwu1RK7aPFFxuhDR1L6tSoc_BJECPebWKRXjBZCiFV4n3oknjhMstn64tZ_2W-5JsGY4Hc5n9yBXArwl93lqt7_RN5w6Cf0h4QyQ5v-65YGjQR0_FDW2QvzqY368QQMicAtaSqzs8KJZgnYb9c7d0zgdAZHzu6qMQvRL5hajrn1n91CbOpbISD08qNLyrdkt-bFTWhAI4vMQFh6WeZu0fM4lFd2NcRwr3XPksINHaQ-G_xBniIqbw0Ls1jF44-csFCur-kEgU8awapJzKnqDKgw",
+   "e":"AQAB",
+   "d":"X4cTteJY_gn4FYPsXB8rdXix5vwsg1FLN5E3EaG6RJoVH-HLLKD9M7dx5oo7GURknchnrRweUkC7hT5fJLM0WbFAKNLWY2vv7B6NqXSzUvxT0_YSfqijwp3RTzlBaCxWp4doFk5N2o8Gy_nHNKroADIkJ46pRUohsXywbReAdYaMwFs9tv8d_cPVY3i07a3t8MN6TNwm0dSawm9v47UiCl3Sk5ZiG7xojPLu4sbg1U2jx4IBTNBznbJSzFHK66jT8bgkuqsk0GjskDJk19Z4qwjwbsnn4j2WBii3RL-Us2lGVkY8fkFzme1z0HbIkfz0Y6mqnOYtqc0X4jfcKoAC8Q",
+   "p":"83i-7IvMGXoMXCskv73TKr8637FiO7Z27zv8oj6pbWUQyLPQBQxtPVnwD20R-60eTDmD2ujnMt5PoqMrm8RfmNhVWDtjjMmCMjOpSXicFHj7XOuVIYQyqVWlWEh6dN36GVZYk93N8Bc9vY41xy8B9RzzOGVQzXvNEvn7O0nVbfs",
+   "q":"3dfOR9cuYq-0S-mkFLzgItgMEfFzB2q3hWehMuG0oCuqnb3vobLyumqjVZQO1dIrdwgTnCdpYzBcOfW5r370AFXjiWft_NGEiovonizhKpo9VVS78TzFgxkIdrecRezsZ-1kYd_s1qDbxtkDEgfAITAG9LUnADun4vIcb6yelxk",
+   "dp":"G4sPXkc6Ya9y8oJW9_ILj4xuppu0lzi_H7VTkS8xj5SdX3coE0oimYwxIi2emTAue0UOa5dpgFGyBJ4c8tQ2VF402XRugKDTP8akYhFo5tAA77Qe_NmtuYZc3C3m3I24G2GvR5sSDxUyAN2zq8Lfn9EUms6rY3Ob8YeiKkTiBj0",
+   "dq":"s9lAH9fggBsoFR8Oac2R_E2gw282rT2kGOAhvIllETE1efrA6huUUvMfBcMpn8lqeW6vzznYY5SSQF7pMdC_agI3nG8Ibp1BUb0JUiraRNqUfLhcQb_d9GF4Dh7e74WbRsobRonujTYN1xCaP6TO61jvWrX-L18txXw494Q_cgk",
+   "qi":"GyM_p6JrXySiz1toFgKbWV-JdI3jQ4ypu9rbMWx3rQJBfmt0FoYzgUIZEVFEcOqwemRN81zoDAaa-Bk0KWNGDjJHZDdDmFhW3AN7lI-puxk_mHZGJ11rxyR8O55XLSe3SPmRfKwZI6yU24ZxvQKFYItdldUKGzO6Ia6zTKhAVRU",
+   "alg":"RS256",
+   "kid":"2011-04-29"}
+]
+}
+`
+
 func TestValidateIPRanges(t *testing.T) {
 	ctx := context.Background()
 	c := internal.NewConfigurationWithDefaults()