From c18858f70504ad08058d4dab1f6060fbf8c20ef8 Mon Sep 17 00:00:00 2001
From: Containerman17 <8990432+containerman17@users.noreply.github.com>
Date: Thu, 19 Sep 2024 05:03:03 +0000
Subject: [PATCH] signing test

---
 tests/signing/signing_test.go | 107 ++++++++++++++++++++++++++++++++++
 1 file changed, 107 insertions(+)
 create mode 100644 tests/signing/signing_test.go

diff --git a/tests/signing/signing_test.go b/tests/signing/signing_test.go
new file mode 100644
index 0000000..acc1962
--- /dev/null
+++ b/tests/signing/signing_test.go
@@ -0,0 +1,107 @@
+package signing
+
+import (
+	"encoding/hex"
+	"math"
+	"testing"
+
+	"github.com/ava-labs/avalanchego/ids"
+	"github.com/ava-labs/hypersdk-starter/actions"
+	"github.com/ava-labs/hypersdk-starter/vm"
+	"github.com/ava-labs/hypersdk/auth"
+	"github.com/ava-labs/hypersdk/chain"
+	"github.com/ava-labs/hypersdk/codec"
+	"github.com/ava-labs/hypersdk/consts"
+	"github.com/ava-labs/hypersdk/crypto/ed25519"
+	"github.com/stretchr/testify/require"
+)
+
+const expectedBase = "0000018fcbcdeef0d36e467c73e2840140cc41b3d72f8a5a7446b2399c39b9c74d4cf077d250902400000002540be400"
+
+func TestSigningEmptyTx(t *testing.T) {
+	require := require.New(t)
+
+	chainID, err := ids.FromString("2c7iUW3kCDwRA9ZFd5bjZZc8iDy68uAsFSBahjqSZGttiTDSNH")
+	require.NoError(err)
+
+	base := &chain.Base{
+		ChainID:   chainID,
+		Timestamp: 1717111222000,
+		MaxFee:    uint64(10 * math.Pow(10, 9)),
+	}
+
+	tx := chain.NewTx(base, []chain.Action{})
+
+	digest, err := tx.Digest()
+	require.NoError(err)
+
+	baseBytes := marshalBase(t, base)
+	require.Equal(expectedBase, hex.EncodeToString(baseBytes))
+
+	expected := expectedBase + "00" // 00 is the length of the actions
+	require.Equal(expected, hex.EncodeToString(digest))
+}
+
+func marshalBase(t *testing.T, base *chain.Base) []byte {
+	require := require.New(t)
+
+	basePacker := codec.NewWriter(0, consts.NetworkSizeLimit)
+	base.Marshal(basePacker)
+	require.NoError(basePacker.Err())
+
+	return basePacker.Bytes()
+}
+
+func TestSigningSingleActionTx(t *testing.T) {
+	require := require.New(t)
+
+	chainID, err := ids.FromString("2c7iUW3kCDwRA9ZFd5bjZZc8iDy68uAsFSBahjqSZGttiTDSNH")
+	require.NoError(err)
+
+	addrBytes, err := hex.DecodeString("1b057de320297c29ad0c1f589ea216869cf1938d88c9fbd70d6748323dbf2fa7")
+	require.NoError(err)
+	var addr codec.Address
+	copy(addr[:], addrBytes)
+
+	action := &actions.Transfer{
+		To:    addr,
+		Value: 123,
+		Memo:  []byte("memo"),
+	}
+
+	tx := chain.NewTx(
+		&chain.Base{
+			ChainID:   chainID,
+			Timestamp: 1717111222000,
+			MaxFee:    uint64(10 * math.Pow(10, 9)),
+		},
+		[]chain.Action{action},
+	)
+
+	digest, err := tx.Digest()
+	require.NoError(err)
+
+	expected := expectedBase +
+		"01" + //how many actions
+		"00" + //action id
+		"1b057de320297c29ad0c1f589ea216869cf1938d88c9fbd70d6748323dbf2fa7" + //from
+		"00000000000000007b" + //value
+		"000000046d656d6f" //memo
+
+	require.Equal(expected, hex.EncodeToString(digest))
+
+	privateKeyHex := "323b1d8f4eed5f0da9da93071b034f2dce9d2d22692c172f3cb252a64ddfafd01b057de320297c29ad0c1f589ea216869cf1938d88c9fbd70d6748323dbf2fa7"
+	privateKeyBytes, err := hex.DecodeString(privateKeyHex)
+	require.NoError(err)
+	privateKey := ed25519.PrivateKey(privateKeyBytes[:ed25519.PrivateKeyLen])
+
+	factory := auth.NewED25519Factory(privateKey)
+	signedTx, err := tx.Sign(factory, vm.ActionParser, vm.AuthParser)
+	require.NoError(err)
+
+	signedTxBytes := signedTx.Bytes()
+	require.NoError(err)
+
+	expectedSignedTx := "0000018fcbcdeef0d36e467c73e2840140cc41b3d72f8a5a7446b2399c39b9c74d4cf077d250902400000002540be40001001b057de320297c29ad0c1f589ea216869cf1938d88c9fbd70d6748323dbf2fa700000000000000007b000000046d656d6f001b057de320297c29ad0c1f589ea216869cf1938d88c9fbd70d6748323dbf2fa72df1b5e3ea1dcf780b70e3c5f4f00ff3d28505ba26a83d7f60f5b691ec301f7b9ed128f5f5fc6289fcff736ba89b22e2fc15644d0355c778e014177b2a8f200c"
+	require.Equal(expectedSignedTx, hex.EncodeToString(signedTxBytes))
+}