1package common
  2
  3import (
  4	"encoding/hex"
  5	"math/big"
  6	"testing"
  7
  8	ethCommon "github.com/ethereum/go-ethereum/common"
  9	"github.com/iden3/go-iden3-crypto/babyjub"
 10	"github.com/stretchr/testify/assert"
 11)
 12
 13func TestNonceParser(t *testing.T) {
 14	n := Nonce(1)
 15	nBytes, err := n.Bytes()
 16	assert.Nil(t, err)
 17	assert.Equal(t, 5, len(nBytes))
 18	assert.Equal(t, "0100000000", hex.EncodeToString(nBytes[:]))
 19	n2 := NonceFromBytes(nBytes)
 20	assert.Equal(t, n, n2)
 21
 22	// value before overflow
 23	n = Nonce(1099511627775)
 24	nBytes, err = n.Bytes()
 25	assert.Nil(t, err)
 26	assert.Equal(t, 5, len(nBytes))
 27	assert.Equal(t, "ffffffffff", hex.EncodeToString(nBytes[:]))
 28	n2 = NonceFromBytes(nBytes)
 29	assert.Equal(t, n, n2)
 30
 31	// expect value overflow
 32	n = Nonce(1099511627776)
 33	nBytes, err = n.Bytes()
 34	assert.NotNil(t, err)
 35	assert.Equal(t, ErrNonceOverflow, err)
 36}
 37
 38func TestTxCompressedData(t *testing.T) {
 39	var sk babyjub.PrivateKey
 40	_, err := hex.Decode(sk[:], []byte("0001020304050607080900010203040506070809000102030405060708090001"))
 41	assert.Nil(t, err)
 42
 43	tx := PoolL2Tx{
 44		FromIdx: 2,
 45		ToIdx:   3,
 46		Amount:  big.NewInt(4),
 47		TokenID: 5,
 48		Nonce:   6,
 49		ToBJJ:   sk.Public(),
 50	}
 51	txCompressedData, err := tx.TxCompressedData()
 52	assert.Nil(t, err)
 53	// test vector value generated from javascript implementation
 54	assert.Equal(t, "1766847064778421992193717128424891165872736891548909569553540449389241871", txCompressedData.String())
 55	assert.Equal(t, "10000000000060000000500040000000000030000000000020001c60be60f", hex.EncodeToString(txCompressedData.Bytes())[1:])
 56
 57	tx = PoolL2Tx{
 58		FromIdx: 7,
 59		ToIdx:   8,
 60		Amount:  big.NewInt(9),
 61		TokenID: 10,
 62		Nonce:   11,
 63		Fee:     12,
 64		ToBJJ:   sk.Public(),
 65	}
 66	txCompressedData, err = tx.TxCompressedDataV2()
 67	assert.Nil(t, err)
 68	// test vector value generated from javascript implementation
 69	assert.Equal(t, "6571340879233176732837827812956721483162819083004853354503", txCompressedData.String())
 70	assert.Equal(t, "10c000000000b0000000a0009000000000008000000000007", hex.EncodeToString(txCompressedData.Bytes())[1:])
 71
 72}
 73
 74func TestHashToSign(t *testing.T) {
 75	var sk babyjub.PrivateKey
 76	_, err := hex.Decode(sk[:], []byte("0001020304050607080900010203040506070809000102030405060708090001"))
 77	assert.Nil(t, err)
 78	ethAddr := ethCommon.HexToAddress("0xc58d29fA6e86E4FAe04DDcEd660d45BCf3Cb2370")
 79
 80	tx := PoolL2Tx{
 81		FromIdx:     2,
 82		ToIdx:       3,
 83		Amount:      big.NewInt(4),
 84		TokenID:     5,
 85		Nonce:       6,
 86		ToBJJ:       sk.Public(),
 87		RqToEthAddr: ethAddr,
 88		RqToBJJ:     sk.Public(),
 89	}
 90	toSign, err := tx.HashToSign()
 91	assert.Nil(t, err)
 92	assert.Equal(t, "14526446928649310956370997581245770629723313742905751117262272426489782809503", toSign.String())
 93}
 94
 95func TestVerifyTxSignature(t *testing.T) {
 96	var sk babyjub.PrivateKey
 97	_, err := hex.Decode(sk[:], []byte("0001020304050607080900010203040506070809000102030405060708090001"))
 98	assert.Nil(t, err)
 99	ethAddr := ethCommon.HexToAddress("0xc58d29fA6e86E4FAe04DDcEd660d45BCf3Cb2370")
100
101	tx := PoolL2Tx{
102		FromIdx:     2,
103		ToIdx:       3,
104		Amount:      big.NewInt(4),
105		TokenID:     5,
106		Nonce:       6,
107		ToBJJ:       sk.Public(),
108		RqToEthAddr: ethAddr,
109		RqToBJJ:     sk.Public(),
110	}
111	toSign, err := tx.HashToSign()
112	assert.Nil(t, err)
113	assert.Equal(t, "14526446928649310956370997581245770629723313742905751117262272426489782809503", toSign.String())
114
115	sig := sk.SignPoseidon(toSign)
116	tx.Signature = sig
117	assert.True(t, tx.VerifySignature(sk.Public()))
118}