feat(test): add more tests for the presets, and compat with iso-ucan
This commit is contained in:
Michael Muré
191
common_test.go
191
common_test.go
@@ -1,68 +1,149 @@
|
||||
package varsig_test
|
||||
|
||||
import (
|
||||
"encoding/hex"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"github.com/ucan-wg/go-varsig"
|
||||
)
|
||||
|
||||
func TestEd25519(t *testing.T) {
|
||||
t.Parallel()
|
||||
func TestRoundTrip(t *testing.T) {
|
||||
for _, tc := range []struct {
|
||||
name string
|
||||
varsig varsig.Varsig
|
||||
dataHex string
|
||||
dataBytes []byte
|
||||
}{
|
||||
// Arbitrary use of presets
|
||||
{
|
||||
name: "Ed25519",
|
||||
varsig: must(varsig.Ed25519(varsig.PayloadEncodingDAGCBOR)),
|
||||
dataHex: "3401ed01ed011371",
|
||||
},
|
||||
{
|
||||
name: "Ed448",
|
||||
varsig: must(varsig.Ed448(varsig.PayloadEncodingDAGCBOR)),
|
||||
dataHex: "3401ed0183241971",
|
||||
},
|
||||
{
|
||||
name: "RS256",
|
||||
varsig: must(varsig.RS256(0x100, varsig.PayloadEncodingDAGCBOR)),
|
||||
dataHex: "3401852412800271",
|
||||
},
|
||||
{
|
||||
name: "RS384",
|
||||
varsig: must(varsig.RS384(0x100, varsig.PayloadEncodingDAGCBOR)),
|
||||
dataHex: "3401852420800271",
|
||||
},
|
||||
{
|
||||
name: "RS512",
|
||||
varsig: must(varsig.RS512(0x100, varsig.PayloadEncodingDAGCBOR)),
|
||||
dataHex: "3401852413800271",
|
||||
},
|
||||
{
|
||||
name: "ES256",
|
||||
varsig: must(varsig.ES256(varsig.PayloadEncodingDAGCBOR)),
|
||||
dataHex: "3401ec0180241271",
|
||||
},
|
||||
{
|
||||
name: "ES256K",
|
||||
varsig: must(varsig.ES256K(varsig.PayloadEncodingDAGCBOR)),
|
||||
dataHex: "3401ec01e7011271",
|
||||
},
|
||||
{
|
||||
name: "ES384",
|
||||
varsig: must(varsig.ES384(varsig.PayloadEncodingDAGCBOR)),
|
||||
dataHex: "3401ec0181242071",
|
||||
},
|
||||
{
|
||||
name: "ES512",
|
||||
varsig: must(varsig.ES512(varsig.PayloadEncodingDAGCBOR)),
|
||||
dataHex: "3401ec0182241371",
|
||||
},
|
||||
{
|
||||
name: "EIP191",
|
||||
varsig: must(varsig.EIP191()),
|
||||
dataHex: "3401ec01e7011b91a303",
|
||||
},
|
||||
|
||||
in, err := varsig.Ed25519(varsig.PayloadEncodingDAGCBOR)
|
||||
mustVarsig(t, in, err)
|
||||
out := roundTrip(t, in, "3401ed01ed011371")
|
||||
assertEdDSAEqual(t, in, out)
|
||||
// from https://github.com/hugomrdias/iso-repo/blob/main/packages/iso-ucan/test/varsig.test.js
|
||||
{
|
||||
name: "RS256+RAW",
|
||||
varsig: must(varsig.RS256(256, varsig.PayloadEncodingVerbatim)),
|
||||
dataBytes: []byte{52, 1, 133, 36, 18, 128, 2, 95},
|
||||
},
|
||||
{
|
||||
name: "ES256+RAW",
|
||||
varsig: must(varsig.ES256(varsig.PayloadEncodingVerbatim)),
|
||||
dataBytes: []byte{52, 1, 236, 1, 128, 36, 18, 95},
|
||||
},
|
||||
{
|
||||
name: "ES512+RAW",
|
||||
varsig: must(varsig.ES512(varsig.PayloadEncodingVerbatim)),
|
||||
dataBytes: []byte{52, 1, 236, 1, 130, 36, 19, 95},
|
||||
},
|
||||
{
|
||||
name: "ES256K+RAW",
|
||||
varsig: must(varsig.ES256K(varsig.PayloadEncodingVerbatim)),
|
||||
dataBytes: []byte{52, 1, 236, 1, 231, 1, 18, 95},
|
||||
},
|
||||
// the two cases below in iso-ucan are actually EIP191 preset where the encoding is overridden
|
||||
// therefore, we build them manually.
|
||||
{
|
||||
name: "EIP191+RAW",
|
||||
varsig: must(varsig.NewECDSAVarsig(varsig.CurveSecp256k1, varsig.HashKeccak256, varsig.PayloadEncodingVerbatim)),
|
||||
dataBytes: []byte{52, 1, 236, 1, 231, 1, 27, 145, 195, 3, 95},
|
||||
},
|
||||
{
|
||||
name: "EIP191+DAG-CBOR",
|
||||
varsig: must(varsig.NewECDSAVarsig(varsig.CurveSecp256k1, varsig.HashKeccak256, varsig.PayloadEncodingDAGCBOR)),
|
||||
dataBytes: []byte{52, 1, 236, 1, 231, 1, 27, 145, 195, 3, 113},
|
||||
},
|
||||
} {
|
||||
t.Run(tc.name, func(t *testing.T) {
|
||||
// round-trip encode and back
|
||||
data := tc.varsig.Encode()
|
||||
|
||||
if tc.dataBytes != nil {
|
||||
require.Equal(t, tc.dataBytes, data)
|
||||
}
|
||||
if tc.dataHex != "" {
|
||||
require.Equal(t, tc.dataHex, hex.EncodeToString(data))
|
||||
}
|
||||
|
||||
rt, err := varsig.Decode(data)
|
||||
require.NoError(t, err)
|
||||
|
||||
require.Equal(t, tc.varsig.Version(), rt.Version())
|
||||
require.Equal(t, tc.varsig.Discriminator(), rt.Discriminator())
|
||||
require.Equal(t, tc.varsig.PayloadEncoding(), rt.PayloadEncoding())
|
||||
require.Equal(t, tc.varsig.Signature(), rt.Signature())
|
||||
|
||||
switch vs := tc.varsig.(type) {
|
||||
case varsig.EdDSAVarsig:
|
||||
rt := rt.(varsig.EdDSAVarsig)
|
||||
require.Equal(t, vs.Curve(), rt.Curve())
|
||||
require.Equal(t, vs.Hash(), rt.Hash())
|
||||
case varsig.ECDSAVarsig:
|
||||
rt := rt.(varsig.ECDSAVarsig)
|
||||
require.Equal(t, vs.Curve(), rt.Curve())
|
||||
require.Equal(t, vs.Hash(), rt.Hash())
|
||||
case varsig.RSAVarsig:
|
||||
rt := rt.(varsig.RSAVarsig)
|
||||
require.Equal(t, vs.Hash(), rt.Hash())
|
||||
require.Equal(t, vs.KeyLength(), rt.KeyLength())
|
||||
default:
|
||||
t.Fatalf("unexpected varsig type: %T", vs)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestEd448(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
in, err := varsig.Ed448(varsig.PayloadEncodingDAGCBOR)
|
||||
mustVarsig(t, in, err)
|
||||
out := roundTrip(t, in, "3401ed0183241971")
|
||||
assertEdDSAEqual(t, in, out)
|
||||
}
|
||||
|
||||
func TestRS256(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
in, err := varsig.RS256(0x100, varsig.PayloadEncodingDAGCBOR)
|
||||
mustVarsig(t, in, err)
|
||||
out := roundTrip(t, in, "3401852412800271")
|
||||
assertRSAEqual(t, in, out)
|
||||
}
|
||||
|
||||
func TestRS384(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
in, err := varsig.RS384(0x100, varsig.PayloadEncodingDAGCBOR)
|
||||
mustVarsig(t, in, err)
|
||||
out := roundTrip(t, in, "3401852420800271")
|
||||
assertRSAEqual(t, in, out)
|
||||
}
|
||||
|
||||
func TestRS512(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
in, err := varsig.RS512(0x100, varsig.PayloadEncodingDAGCBOR)
|
||||
mustVarsig(t, in, err)
|
||||
out := roundTrip(t, in, "3401852413800271")
|
||||
assertRSAEqual(t, in, out)
|
||||
}
|
||||
|
||||
func assertEdDSAEqual(t *testing.T, in, out varsig.EdDSAVarsig) {
|
||||
t.Helper()
|
||||
|
||||
assert.Equal(t, in.Curve(), out.Curve())
|
||||
assert.Equal(t, in.Hash(), out.Hash())
|
||||
}
|
||||
|
||||
func assertRSAEqual(t *testing.T, in, out varsig.RSAVarsig) {
|
||||
t.Helper()
|
||||
|
||||
assert.Equal(t, in.Hash(), out.Hash())
|
||||
assert.Equal(t, in.KeyLength(), out.KeyLength())
|
||||
func must[T any](v T, err error) T {
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return v
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user