ory-am/hydra

// Copyright © 2022 Ory Corp
// SPDX-License-Identifier: Apache-2.0

package cli

import (
    "testing"

    "github.com/ory/x/josex"
)

func Test_toSDKFriendlyJSONWebKey(t *testing.T) {
    publicJWK := []byte(`{
        "kty": "RSA",
        "e": "AQAB",
        "use": "sig",
        "kid": "7a5ff76a-6766-11ea-bc55-0242ac130003",
        "alg": "RS256",
        "n": "l80jJJqcc1PpefIGVIjuPvA1D7NscnuF9aQqLa7I9rDUK4IaSOO3kL_EF13k-jTzcA5q4OZn5dR0kmqIMZT2gQ"
    }`)

    publicPEM := []byte(`
        -----BEGIN PUBLIC KEY-----
        MFwwDQYJKoZIhvcNAQEBBQADSwAwSAJBAPf64dykufSkwnvUiBAwd5Si0K6t4m5i
        qJD8TmLJCmFjKaOUa6nszcFt/FkAuORfdlrD9mEZLPrPx74RSluyTBMCAwEAAQ==
        -----END PUBLIC KEY-----
    `)

    type args struct {
        key []byte
        kid string
        use string
    }
    tests := []struct {
        name string
        args args
        want string
    }{
        {
            name: "JWK with algorithm",
            args: args{
                key: publicJWK,
                kid: "public:7a5ff76a-6766-11ea-bc55-0242ac130003",
                use: "sig",
            },
            want: "RS256",
        },
        {
            name: "PEM key without algorithm",
            args: args{
                key: publicPEM,
                kid: "public:7a5ff76a-6766-11ea-bc55-0242ac130003",
                use: "sig",
            },
            want: "",
        },
    }
    for _, tt := range tests {
        t.Run(tt.name, func(t *testing.T) {
            key, _ := josex.LoadPublicKey(tt.args.key)
            if got := ToSDKFriendlyJSONWebKey(key, tt.args.kid, tt.args.use); got.Algorithm != tt.want {
                t.Errorf("toSDKFriendlyJSONWebKey() = %v, want %v", got.Algorithm, tt.want)
            }
        })
    }
}