forked from TrueCloudLab/neoneo-go
2c3e92923f
PublicKey() for PrivateKey now just can't fail and it makes no sense to return an error from it. There is a lot of associated functionality for which this also is true, so adjust it accordingly and simplify a lot of code.
36 lines
848 B
Go
Executable file
36 lines
848 B
Go
Executable file
package keys
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/crypto/hash"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestPubKeyVerify(t *testing.T) {
|
|
var data = []byte("sample")
|
|
hashedData := hash.Sha256(data)
|
|
|
|
privKey, err := NewPrivateKey()
|
|
assert.Nil(t, err)
|
|
signedData, err := privKey.Sign(data)
|
|
assert.Nil(t, err)
|
|
pubKey := privKey.PublicKey()
|
|
result := pubKey.Verify(signedData, hashedData.Bytes())
|
|
expected := true
|
|
assert.Equal(t, expected, result)
|
|
}
|
|
|
|
func TestWrongPubKey(t *testing.T) {
|
|
privKey, _ := NewPrivateKey()
|
|
sample := []byte("sample")
|
|
hashedData := hash.Sha256(sample)
|
|
signedData, _ := privKey.Sign(sample)
|
|
|
|
secondPrivKey, _ := NewPrivateKey()
|
|
wrongPubKey := secondPrivKey.PublicKey()
|
|
|
|
actual := wrongPubKey.Verify(signedData, hashedData.Bytes())
|
|
expcted := false
|
|
assert.Equal(t, expcted, actual)
|
|
}
|