81 lines
1.7 KiB
Go
81 lines
1.7 KiB
Go
package keys
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/internal/keytestcases"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNEP2Encrypt(t *testing.T) {
|
|
for _, testCase := range keytestcases.Arr {
|
|
|
|
privKey, err := NewPrivateKeyFromHex(testCase.PrivateKey)
|
|
if testCase.Invalid {
|
|
assert.Error(t, err)
|
|
continue
|
|
}
|
|
|
|
assert.Nil(t, err)
|
|
|
|
encryptedWif, err := NEP2Encrypt(privKey, testCase.Passphrase)
|
|
assert.Nil(t, err)
|
|
|
|
assert.Equal(t, testCase.EncryptedWif, encryptedWif)
|
|
}
|
|
}
|
|
|
|
func TestNEP2Decrypt(t *testing.T) {
|
|
for _, testCase := range keytestcases.Arr {
|
|
privKey, err := NEP2Decrypt(testCase.EncryptedWif, testCase.Passphrase)
|
|
if testCase.Invalid {
|
|
assert.Error(t, err)
|
|
continue
|
|
}
|
|
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, testCase.PrivateKey, privKey.String())
|
|
|
|
wif := privKey.WIF()
|
|
assert.Equal(t, testCase.Wif, wif)
|
|
|
|
address := privKey.Address()
|
|
assert.Equal(t, testCase.Address, address)
|
|
}
|
|
}
|
|
|
|
func TestNEP2DecryptErrors(t *testing.T) {
|
|
p := "qwerty"
|
|
|
|
// Not a base58-encoded value
|
|
s := "qazwsx"
|
|
_, err := NEP2Decrypt(s, p)
|
|
assert.Error(t, err)
|
|
|
|
// Valid base58, but not a NEP-2 format.
|
|
s = "KxhEDBQyyEFymvfJD96q8stMbJMbZUb6D1PmXqBWZDU2WvbvVs9o"
|
|
_, err = NEP2Decrypt(s, p)
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestValidateNEP2Format(t *testing.T) {
|
|
// Wrong length.
|
|
s := []byte("gobbledygook")
|
|
assert.Error(t, validateNEP2Format(s))
|
|
|
|
// Wrong header 1.
|
|
s = []byte("gobbledygookgobbledygookgobbledygookgob")
|
|
assert.Error(t, validateNEP2Format(s))
|
|
|
|
// Wrong header 2.
|
|
s[0] = 0x01
|
|
assert.Error(t, validateNEP2Format(s))
|
|
|
|
// Wrong header 3.
|
|
s[1] = 0x42
|
|
assert.Error(t, validateNEP2Format(s))
|
|
|
|
// OK
|
|
s[2] = 0xe0
|
|
assert.NoError(t, validateNEP2Format(s))
|
|
}
|