57 lines
1.4 KiB
Go
57 lines
1.4 KiB
Go
|
// Copyright (C) 2018 Storj Labs, Inc.
|
||
|
// See LICENSE for copying information.
|
||
|
|
||
|
package certdb_test
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"crypto/ecdsa"
|
||
|
"crypto/x509"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/stretchr/testify/assert"
|
||
|
"github.com/stretchr/testify/require"
|
||
|
|
||
|
"storj.io/storj/internal/testcontext"
|
||
|
"storj.io/storj/internal/testidentity"
|
||
|
"storj.io/storj/pkg/certdb"
|
||
|
"storj.io/storj/satellite"
|
||
|
"storj.io/storj/satellite/satellitedb/satellitedbtest"
|
||
|
)
|
||
|
|
||
|
func TestCertDB(t *testing.T) {
|
||
|
satellitedbtest.Run(t, func(t *testing.T, db satellite.DB) {
|
||
|
ctx := testcontext.New(t)
|
||
|
defer ctx.Cleanup()
|
||
|
|
||
|
testDatabase(ctx, t, db.CertDB())
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func testDatabase(ctx context.Context, t *testing.T, upldb certdb.DB) {
|
||
|
//testing variables
|
||
|
upID, err := testidentity.NewTestIdentity(ctx)
|
||
|
require.NoError(t, err)
|
||
|
publicKeyEcdsa, _ := upID.Leaf.PublicKey.(*ecdsa.PublicKey)
|
||
|
upIDpubbytes, err := x509.MarshalPKIXPublicKey(publicKeyEcdsa)
|
||
|
require.NoError(t, err)
|
||
|
|
||
|
{ // New entry
|
||
|
err := upldb.SavePublicKey(ctx, upID.ID, upID.Leaf.PublicKey.(*ecdsa.PublicKey))
|
||
|
assert.NoError(t, err)
|
||
|
}
|
||
|
|
||
|
{ // New entry
|
||
|
err := upldb.SavePublicKey(ctx, upID.ID, upID.Leaf.PublicKey)
|
||
|
assert.NoError(t, err)
|
||
|
}
|
||
|
|
||
|
{ // Get the corresponding Public key for the serialnum
|
||
|
pubkey, err := upldb.GetPublicKey(ctx, upID.ID)
|
||
|
assert.NoError(t, err)
|
||
|
pubbytes, err := x509.MarshalPKIXPublicKey(pubkey)
|
||
|
assert.NoError(t, err)
|
||
|
assert.EqualValues(t, upIDpubbytes, pubbytes)
|
||
|
}
|
||
|
}
|