mirror of
				https://source.quilibrium.com/quilibrium/ceremonyclient.git
				synced 2025-11-04 03:07:28 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			97 lines
		
	
	
		
			1.5 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			97 lines
		
	
	
		
			1.5 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
package crypto
 | 
						|
 | 
						|
import (
 | 
						|
	"crypto/rand"
 | 
						|
	"testing"
 | 
						|
)
 | 
						|
 | 
						|
func TestSecp256k1BasicSignAndVerify(t *testing.T) {
 | 
						|
	priv, pub, err := GenerateSecp256k1Key(rand.Reader)
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
 | 
						|
	data := []byte("hello! and welcome to some awesome crypto primitives")
 | 
						|
 | 
						|
	sig, err := priv.Sign(data)
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
 | 
						|
	ok, err := pub.Verify(data, sig)
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
 | 
						|
	if !ok {
 | 
						|
		t.Fatal("signature didn't match")
 | 
						|
	}
 | 
						|
 | 
						|
	// change data
 | 
						|
	data[0] = ^data[0]
 | 
						|
	ok, err = pub.Verify(data, sig)
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
 | 
						|
	if ok {
 | 
						|
		t.Fatal("signature matched and shouldn't")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestSecp256k1SignZero(t *testing.T) {
 | 
						|
	priv, pub, err := GenerateSecp256k1Key(rand.Reader)
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
 | 
						|
	data := make([]byte, 0)
 | 
						|
	sig, err := priv.Sign(data)
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
 | 
						|
	ok, err := pub.Verify(data, sig)
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
	if !ok {
 | 
						|
		t.Fatal("signature didn't match")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestSecp256k1MarshalLoop(t *testing.T) {
 | 
						|
	priv, pub, err := GenerateSecp256k1Key(rand.Reader)
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
 | 
						|
	privB, err := MarshalPrivateKey(priv)
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
 | 
						|
	privNew, err := UnmarshalPrivateKey(privB)
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
 | 
						|
	if !priv.Equals(privNew) || !privNew.Equals(priv) {
 | 
						|
		t.Fatal("keys are not equal")
 | 
						|
	}
 | 
						|
 | 
						|
	pubB, err := MarshalPublicKey(pub)
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
	pubNew, err := UnmarshalPublicKey(pubB)
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
 | 
						|
	if !pub.Equals(pubNew) || !pubNew.Equals(pub) {
 | 
						|
		t.Fatal("keys are not equal")
 | 
						|
	}
 | 
						|
 | 
						|
}
 |