1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
|
package storage
import (
"crypto/x509"
"encoding/json"
"testing"
"time"
"github.com/stretchr/testify/require"
"go.step.sm/crypto/keyutil"
"go.step.sm/crypto/minica"
"go.step.sm/crypto/x509util"
)
func TestAK_MarshalUnmarshal(t *testing.T) {
ca, err := minica.New()
require.NoError(t, err)
signer, err := keyutil.GenerateSigner("RSA", "", 2048)
require.NoError(t, err)
cr, err := x509util.NewCertificateRequest(signer)
require.NoError(t, err)
cr.Subject.CommonName = "testak"
csr, err := cr.GetCertificateRequest()
require.NoError(t, err)
cert, err := ca.SignCSR(csr)
require.NoError(t, err)
ak := &AK{
Name: "ak1",
Data: []byte{1, 2, 3, 4},
Chain: []*x509.Certificate{cert, ca.Intermediate},
CreatedAt: time.Time{},
}
data, err := json.Marshal(ak)
require.NoError(t, err)
var rak = &AK{}
err = json.Unmarshal(data, rak)
require.NoError(t, err)
require.Equal(t, ak, rak)
}
func TestKey_MarshalUnmarshal(t *testing.T) {
ca, err := minica.New()
require.NoError(t, err)
signer, err := keyutil.GenerateSigner("RSA", "", 2048)
require.NoError(t, err)
cr, err := x509util.NewCertificateRequest(signer)
require.NoError(t, err)
cr.Subject.CommonName = "testkey"
csr, err := cr.GetCertificateRequest()
require.NoError(t, err)
cert, err := ca.SignCSR(csr)
require.NoError(t, err)
key := &Key{
Name: "key1",
Data: []byte{1, 2, 3, 4},
AttestedBy: "ak1",
Chain: []*x509.Certificate{cert, ca.Intermediate},
CreatedAt: time.Time{},
}
data, err := json.Marshal(key)
require.NoError(t, err)
var rkey = &Key{}
err = json.Unmarshal(data, rkey)
require.NoError(t, err)
require.Equal(t, key, rkey)
}
|