// Copyright 2015 The Vanadium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package security

import (
	"crypto/ecdsa"
	"crypto/elliptic"
	"crypto/rand"
	"reflect"
	"testing"
)

func TestCryptoUtilPublicKeyCoder(t *testing.T) {
	priv, err := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
	if err != nil {
		t.Fatalf("couldn't generate private key: %v", err)
	}
	pub := &priv.PublicKey
	data, err := marshalPKIXPublicKey(pub)
	if err != nil {
		t.Fatalf("couldn't marshal public key: %v", err)
	}
	pubNew, err := parsePKIXPublicKey(data)
	if err != nil {
		t.Fatalf("couldn't parse public key: %v", err)
	}
	if !reflect.DeepEqual(pub, pubNew) {
		t.Fatalf("public keys don't match")
	}
}

func TestCryptoUtilPrivateKeyCoder(t *testing.T) {
	priv, err := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
	if err != nil {
		t.Fatalf("couldn't generate private key: %v", err)
	}
	data, err := marshalPKCS8PrivateKey(priv)
	if err != nil {
		t.Fatalf("couldn't marshal private key: %v", err)
	}
	privNew, err := parsePKCS8PrivateKey(data)
	if err != nil {
		t.Fatalf("couldn't parse private key: %v", err)
	}
	if !reflect.DeepEqual(priv, privNew) {
		t.Fatalf("private keys don't match")
	}
}