mirror of
https://gitlab.com/pulsechaincom/go-pulse.git
synced 2024-12-25 12:57:17 +00:00
47d3b3dd58
* Remove flags field from key struct * Change JSON struct fields from string to []byte * Change GenerateNewKey API to take io.Reader for random source * Remove mixing entropy source function * Use testing Fatal in tests
86 lines
1.5 KiB
Go
86 lines
1.5 KiB
Go
package crypto
|
|
|
|
import (
|
|
crand "crypto/rand"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestKeyStorePlain(t *testing.T) {
|
|
ks := NewKeyStorePlain(DefaultDataDir())
|
|
pass := "" // not used but required by API
|
|
k1, err := ks.GenerateNewKey(crand.Reader, pass)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
k2 := new(Key)
|
|
k2, err = ks.GetKey(k1.Id, pass)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(k1.Id, k2.Id) {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(k1.PrivateKey, k2.PrivateKey) {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = ks.DeleteKey(k2.Id, pass)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestKeyStorePassphrase(t *testing.T) {
|
|
ks := NewKeyStorePassphrase(DefaultDataDir())
|
|
pass := "foo"
|
|
k1, err := ks.GenerateNewKey(crand.Reader, pass)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
k2 := new(Key)
|
|
k2, err = ks.GetKey(k1.Id, pass)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !reflect.DeepEqual(k1.Id, k2.Id) {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(k1.PrivateKey, k2.PrivateKey) {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = ks.DeleteKey(k2.Id, pass) // also to clean up created files
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestKeyStorePassphraseDecryptionFail(t *testing.T) {
|
|
ks := NewKeyStorePassphrase(DefaultDataDir())
|
|
pass := "foo"
|
|
k1, err := ks.GenerateNewKey(crand.Reader, pass)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
_, err = ks.GetKey(k1.Id, "bar") // wrong passphrase
|
|
if err == nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = ks.DeleteKey(k1.Id, "bar") // wrong passphrase
|
|
if err == nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = ks.DeleteKey(k1.Id, pass) // to clean up
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|