1 files changed, 99 insertions, 0 deletions
diff --git a/crypto/key_store_test.go b/crypto/key_store_test.go
new file mode 100644
index 000000000..11531d460
--- /dev/null
+++ b/crypto/key_store_test.go
@@ -0,0 +1,99 @@
+package crypto
+
+import (
+	"github.com/ethereum/go-ethereum/crypto/randentropy"
+	"github.com/ethereum/go-ethereum/ethutil"
+	"reflect"
+	"testing"
+)
+
+func TestKeyStorePlain(t *testing.T) {
+	ks := NewKeyStorePlain(ethutil.DefaultDataDir())
+	pass := "" // not used but required by API
+	k1, err := ks.GenerateNewKey(randentropy.Reader, pass)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	k2 := new(Key)
+	k2, err = ks.GetKey(k1.Address, pass)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if !reflect.DeepEqual(k1.Address, k2.Address) {
+		t.Fatal(err)
+	}
+
+	if !reflect.DeepEqual(k1.PrivateKey, k2.PrivateKey) {
+		t.Fatal(err)
+	}
+
+	err = ks.DeleteKey(k2.Address, pass)
+	if err != nil {
+		t.Fatal(err)
+	}
+}
+
+func TestKeyStorePassphrase(t *testing.T) {
+	ks := NewKeyStorePassphrase(ethutil.DefaultDataDir())
+	pass := "foo"
+	k1, err := ks.GenerateNewKey(randentropy.Reader, pass)
+	if err != nil {
+		t.Fatal(err)
+	}
+	k2 := new(Key)
+	k2, err = ks.GetKey(k1.Address, pass)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !reflect.DeepEqual(k1.Address, k2.Address) {
+		t.Fatal(err)
+	}
+
+	if !reflect.DeepEqual(k1.PrivateKey, k2.PrivateKey) {
+		t.Fatal(err)
+	}
+
+	err = ks.DeleteKey(k2.Address, pass) // also to clean up created files
+	if err != nil {
+		t.Fatal(err)
+	}
+}
+
+func TestKeyStorePassphraseDecryptionFail(t *testing.T) {
+	ks := NewKeyStorePassphrase(ethutil.DefaultDataDir())
+	pass := "foo"
+	k1, err := ks.GenerateNewKey(randentropy.Reader, pass)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	_, err = ks.GetKey(k1.Address, "bar") // wrong passphrase
+	if err == nil {
+		t.Fatal(err)
+	}
+
+	err = ks.DeleteKey(k1.Address, "bar") // wrong passphrase
+	if err == nil {
+		t.Fatal(err)
+	}
+
+	err = ks.DeleteKey(k1.Address, pass) // to clean up
+	if err != nil {
+		t.Fatal(err)
+	}
+}
+
+func TestImportPreSaleKey(t *testing.T) {
+	// file content of a presale key file generated with:
+	// python pyethsaletool.py genwallet
+	// with password "foo"
+	fileContent := "{\"encseed\": \"26d87f5f2bf9835f9a47eefae571bc09f9107bb13d54ff12a4ec095d01f83897494cf34f7bed2ed34126ecba9db7b62de56c9d7cd136520a0427bfb11b8954ba7ac39b90d4650d3448e31185affcd74226a68f1e94b1108e6e0a4a91cdd83eba\", \"ethaddr\": \"d4584b5f6229b7be90727b0fc8c6b91bb427821f\", \"email\": \"gustav.simonsson@gmail.com\", \"btcaddr\": \"1EVknXyFC68kKNLkh6YnKzW41svSRoaAcx\"}"
+	ks := NewKeyStorePassphrase(ethutil.DefaultDataDir())
+	pass := "foo"
+	_, err := ImportPreSaleKey(ks, []byte(fileContent), pass)
+	if err != nil {
+		t.Fatal(err)
+	}
+}