go-peer/pkg/client/client_test.go

// nolint: goerr113
package client

import (
	"bytes"
	"testing"

	"github.com/number571/go-peer/pkg/crypto/asymmetric"
)

func TestClient(t *testing.T) {
	t.Parallel()

	client := NewClient(
		asymmetric.NewPrivKeyChain(
			asymmetric.NewKEncPrivKey(),
			asymmetric.NewSignPrivKey(),
		),
		(8 << 10),
	)

	kemPubKey := client.GetPrivKeyChain().GetKEncPrivKey().GetPubKey()
	msg := []byte("hello, world!")

	enc, err := client.EncryptMessage(kemPubKey, msg)
	if err != nil {
		t.Error(err)
		return
	}

	// os.WriteFile("test_binary.msg", enc, 0600)
	// os.WriteFile("test_string.msg", []byte(encoding.HexEncode(enc)), 0600)

	signerPubKey := client.GetPrivKeyChain().GetSignPrivKey().GetPubKey()
	gotSignPubKey, dec, err := client.DecryptMessage(enc)
	if err != nil {
		t.Error(err)
		return
	}
	if !bytes.Equal(signerPubKey.ToBytes(), gotSignPubKey.ToBytes()) {
		t.Error("invalid decrypt signer key")
		return
	}
	if !bytes.Equal(msg, dec) {
		t.Error("invalid decrypt message")
		return
	}

	// fmt.Println(client.GetMessageLimit())
	// fmt.Println(len(client.GetPrivKeyChain().GetPubKeyChain().ToString()))
}