github.com/annwntech/go-micro/v2@v2.9.5/config/secrets/box/box_test.go (about)

     1  package box
     2  
     3  import (
     4  	"crypto/rand"
     5  	"reflect"
     6  	"testing"
     7  
     8  	"github.com/annwntech/go-micro/v2/config/secrets"
     9  	naclbox "golang.org/x/crypto/nacl/box"
    10  )
    11  
    12  func TestBox(t *testing.T) {
    13  	alicePublicKey, alicePrivateKey, err := naclbox.GenerateKey(rand.Reader)
    14  	if err != nil {
    15  		t.Fatal(err)
    16  	}
    17  	bobPublicKey, bobPrivateKey, err := naclbox.GenerateKey(rand.Reader)
    18  	if err != nil {
    19  		t.Fatal(err)
    20  	}
    21  	alice, bob := NewSecrets(secrets.PublicKey(alicePublicKey[:]), secrets.PrivateKey(alicePrivateKey[:])), NewSecrets()
    22  	if err := alice.Init(); err != nil {
    23  		t.Error(err)
    24  	}
    25  	if err := bob.Init(secrets.PublicKey(bobPublicKey[:]), secrets.PrivateKey(bobPrivateKey[:])); err != nil {
    26  		t.Error(err)
    27  	}
    28  	if alice.String() != "nacl-box" {
    29  		t.Error("String() doesn't return nacl-box")
    30  	}
    31  	aliceSecret := []byte("Why is a raven like a writing-desk?")
    32  	if _, err := alice.Encrypt(aliceSecret); err == nil {
    33  		t.Error("alice.Encrypt succeded without a public key")
    34  	}
    35  	enc, err := alice.Encrypt(aliceSecret, secrets.RecipientPublicKey(bob.Options().PublicKey))
    36  	if err != nil {
    37  		t.Error("alice.Encrypt failed")
    38  	}
    39  	if _, err := bob.Decrypt(enc); err == nil {
    40  		t.Error("bob.Decrypt succeded without a public key")
    41  	}
    42  	if dec, err := bob.Decrypt(enc, secrets.SenderPublicKey(alice.Options().PublicKey)); err == nil {
    43  		if !reflect.DeepEqual(dec, aliceSecret) {
    44  			t.Errorf("Bob's decrypted message didn't match Alice's encrypted message: %v != %v", aliceSecret, dec)
    45  		}
    46  	} else {
    47  		t.Errorf("bob.Decrypt failed (%s)", err)
    48  	}
    49  
    50  	bobSecret := []byte("I haven't the slightest idea")
    51  	enc, err = bob.Encrypt(bobSecret, secrets.RecipientPublicKey(alice.Options().PublicKey))
    52  	if err != nil {
    53  		t.Error(err)
    54  	}
    55  	dec, err := alice.Decrypt(enc, secrets.SenderPublicKey(bob.Options().PrivateKey))
    56  	if err == nil {
    57  		t.Error(err)
    58  	}
    59  	dec, err = alice.Decrypt(enc, secrets.SenderPublicKey(bob.Options().PublicKey))
    60  	if err != nil {
    61  		t.Error(err)
    62  	}
    63  	if !reflect.DeepEqual(dec, bobSecret) {
    64  		t.Errorf("Alice's decrypted message didn't match Bob's encrypted message %v != %v", bobSecret, dec)
    65  	}
    66  }