forked from TrueCloudLab/neoneo-go
35 lines
779 B
Go
35 lines
779 B
Go
|
package pubkeytesthelper
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/CityOfZion/neo-go/pkg/crypto/hash"
|
||
|
|
"github.com/CityOfZion/neo-go/pkg/crypto/privatekey"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestPubKeyVerify(t *testing.T) {
|
||
|
|
actual, err := SignDataWithRandomPrivateKey([]byte("sample"))
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
expected := true
|
||
|
|
|
||
|
|
assert.Equal(t, expected, actual)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestWrongPubKey(t *testing.T) {
|
||
|
|
privKey, _ := privatekey.NewPrivateKey()
|
||
|
|
sample := []byte("sample")
|
||
|
|
hashedData, _ := hash.Sha256(sample)
|
||
|
|
signedData, _ := privKey.Sign(sample)
|
||
|
|
|
||
|
|
secondPrivKey, _ := privatekey.NewPrivateKey()
|
||
|
|
wrongPubKey, _ := secondPrivKey.PublicKey()
|
||
|
|
|
||
|
|
actual := wrongPubKey.Verify(signedData, hashedData.Bytes())
|
||
|
|
expcted := false
|
||
|
|
assert.Equal(t, expcted, actual)
|
||
|
|
}
|