|
|
package common
import ( "encoding/hex" "math/big" "testing"
ethCommon "github.com/ethereum/go-ethereum/common" "github.com/iden3/go-iden3-crypto/babyjub" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require")
func TestNewPoolL2Tx(t *testing.T) { poolL2Tx := &PoolL2Tx{ FromIdx: 87654, ToIdx: 300, Amount: big.NewInt(4), TokenID: 5, Nonce: 144, } poolL2Tx, err := NewPoolL2Tx(poolL2Tx) assert.NoError(t, err) assert.Equal(t, "0x022669acda59b827d20ef5354a3eebd1dffb3972b0a6bf89d18bfd2efa0ab9f41e", poolL2Tx.TxID.String())}
func TestTxCompressedDataAndTxCompressedDataV2JSVectors(t *testing.T) { // test vectors values generated from javascript implementation
var skPositive babyjub.PrivateKey // 'Positive' refers to the sign
_, err := hex.Decode(skPositive[:], []byte("0001020304050607080900010203040506070809000102030405060708090001")) assert.NoError(t, err)
var skNegative babyjub.PrivateKey // 'Negative' refers to the sign
_, err = hex.Decode(skNegative[:], []byte("0001020304050607080900010203040506070809000102030405060708090002")) assert.NoError(t, err)
amount, ok := new(big.Int).SetString("343597383670000000000000000000000000000000", 10) require.True(t, ok) tx := PoolL2Tx{ FromIdx: (1 << 48) - 1, ToIdx: (1 << 48) - 1, Amount: amount, TokenID: (1 << 32) - 1, Nonce: (1 << 40) - 1, Fee: (1 << 3) - 1, ToBJJ: skPositive.Public().Compress(), } txCompressedData, err := tx.TxCompressedData(uint16((1 << 16) - 1)) require.NoError(t, err) expectedStr := "0107ffffffffffffffffffffffffffffffffffffffffffffffc60be60f" assert.Equal(t, expectedStr, hex.EncodeToString(txCompressedData.Bytes()))
txCompressedDataV2, err := tx.TxCompressedDataV2() require.NoError(t, err) expectedStr = "0107ffffffffffffffffffffffffffffffffffffffffffffffffffff" assert.Equal(t, expectedStr, hex.EncodeToString(txCompressedDataV2.Bytes()))
tx = PoolL2Tx{ FromIdx: 0, ToIdx: 0, Amount: big.NewInt(0), TokenID: 0, Nonce: 0, Fee: 0, ToBJJ: skNegative.Public().Compress(), } txCompressedData, err = tx.TxCompressedData(uint16(0)) require.NoError(t, err) expectedStr = "c60be60f" assert.Equal(t, expectedStr, hex.EncodeToString(txCompressedData.Bytes()))
txCompressedDataV2, err = tx.TxCompressedDataV2() require.NoError(t, err) assert.Equal(t, "0", txCompressedDataV2.String())
amount, ok = new(big.Int).SetString("63000000000000000", 10) require.True(t, ok) tx = PoolL2Tx{ FromIdx: 324, ToIdx: 256, Amount: amount, TokenID: 123, Nonce: 76, Fee: 214, ToBJJ: skNegative.Public().Compress(), } txCompressedData, err = tx.TxCompressedData(uint16(1)) require.NoError(t, err) expectedStr = "d6000000004c0000007b0000000001000000000001440001c60be60f" assert.Equal(t, expectedStr, hex.EncodeToString(txCompressedData.Bytes()))
txCompressedDataV2, err = tx.TxCompressedDataV2() require.NoError(t, err) expectedStr = "d6000000004c0000007b3977825f00000000000100000000000144" assert.Equal(t, expectedStr, hex.EncodeToString(txCompressedDataV2.Bytes()))
tx = PoolL2Tx{ FromIdx: 1, ToIdx: 2, TokenID: 3, Nonce: 4, Fee: 5, ToBJJ: skNegative.Public().Compress(), } txCompressedData, err = tx.TxCompressedData(uint16(0)) require.NoError(t, err) expectedStr = "050000000004000000030000000000020000000000010000c60be60f" assert.Equal(t, expectedStr, hex.EncodeToString(txCompressedData.Bytes()))
tx = PoolL2Tx{ FromIdx: 2, ToIdx: 3, TokenID: 4, Nonce: 5, Fee: 6, ToBJJ: skPositive.Public().Compress(), } txCompressedData, err = tx.TxCompressedData(uint16(0)) require.NoError(t, err) expectedStr = "01060000000005000000040000000000030000000000020000c60be60f" assert.Equal(t, expectedStr, hex.EncodeToString(txCompressedData.Bytes()))}
func TestRqTxCompressedDataV2(t *testing.T) { var sk babyjub.PrivateKey _, err := hex.Decode(sk[:], []byte("0001020304050607080900010203040506070809000102030405060708090001")) assert.NoError(t, err) tx := PoolL2Tx{ RqFromIdx: 7, RqToIdx: 8, RqAmount: big.NewInt(9), RqTokenID: 10, RqNonce: 11, RqFee: 12, RqToBJJ: sk.Public().Compress(), } txCompressedData, err := tx.RqTxCompressedDataV2() assert.NoError(t, err) // test vector value generated from javascript implementation
expectedStr := "110248805340524920412994530176819463725852160917809517418728390663" assert.Equal(t, expectedStr, txCompressedData.String()) expected, ok := new(big.Int).SetString(expectedStr, 10) assert.True(t, ok) assert.Equal(t, expected.Bytes(), txCompressedData.Bytes()) assert.Equal(t, "010c000000000b0000000a0000000009000000000008000000000007", hex.EncodeToString(txCompressedData.Bytes()))}
func TestHashToSign(t *testing.T) { chainID := uint16(0) tx := PoolL2Tx{ FromIdx: 2, ToIdx: 3, Amount: big.NewInt(4), TokenID: 5, Nonce: 6, ToEthAddr: ethCommon.HexToAddress("0xc58d29fA6e86E4FAe04DDcEd660d45BCf3Cb2370"), } toSign, err := tx.HashToSign(chainID) assert.NoError(t, err) assert.Equal(t, "2d49ce1d4136e06f64e3eb1f79a346e6ee3e93ceeac909a57806a8d87005c263", hex.EncodeToString(toSign.Bytes()))}
func TestVerifyTxSignature(t *testing.T) { chainID := uint16(0) var sk babyjub.PrivateKey _, err := hex.Decode(sk[:], []byte("0001020304050607080900010203040506070809000102030405060708090001")) assert.NoError(t, err) tx := PoolL2Tx{ FromIdx: 2, ToIdx: 3, Amount: big.NewInt(4), TokenID: 5, Nonce: 6, ToBJJ: sk.Public().Compress(), RqToEthAddr: ethCommon.HexToAddress("0xc58d29fA6e86E4FAe04DDcEd660d45BCf3Cb2370"), RqToBJJ: sk.Public().Compress(), } toSign, err := tx.HashToSign(chainID) assert.NoError(t, err) assert.Equal(t, "1571327027383224465388301747239444557034990637650927918405777653988509342917", toSign.String())
sig := sk.SignPoseidon(toSign) tx.Signature = sig.Compress() assert.True(t, tx.VerifySignature(chainID, sk.Public().Compress()))}
func TestVerifyTxSignatureEthAddrWith0(t *testing.T) { chainID := uint16(5) var sk babyjub.PrivateKey _, err := hex.Decode(sk[:], []byte("02f0b4f87065af3797aaaf934e8b5c31563c17f2272fa71bd0146535bfbb4184")) assert.NoError(t, err) tx := PoolL2Tx{ FromIdx: 10659, ToIdx: 0, ToEthAddr: ethCommon.HexToAddress("0x0004308BD15Ead4F1173624dC289DBdcC806a309"), Amount: big.NewInt(5000), TokenID: 0, Nonce: 946, Fee: 231, } toSign, err := tx.HashToSign(chainID) assert.NoError(t, err)
sig := sk.SignPoseidon(toSign) assert.Equal(t, "93ce988ecac87908513648d0d021d8ad56d3b504eb38204200f07a8f6c20551de9951397d3051143d4c71203205f8849cd7382d2bcf4bd10092c1e052bb40d01", sig.Compress().String(), ) tx.Signature = sig.Compress() assert.True(t, tx.VerifySignature(chainID, sk.Public().Compress()))}
func TestDecompressEmptyBJJComp(t *testing.T) { pkComp := EmptyBJJComp pk, err := pkComp.Decompress() require.NoError(t, err) assert.Equal(t, "2957874849018779266517920829765869116077630550401372566248359756137677864698", pk.X.String()) assert.Equal(t, "0", pk.Y.String())}
func TestPoolL2TxID(t *testing.T) { tx0 := PoolL2Tx{ FromIdx: 5, ToIdx: 5, Amount: big.NewInt(5), Fee: 126, TokenID: 5, Nonce: 5, } err := tx0.SetID() require.NoError(t, err)
// differ TokenID
tx1 := PoolL2Tx{ FromIdx: 5, ToIdx: 5, Amount: big.NewInt(5), Fee: 126, TokenID: 4, Nonce: 5, } err = tx1.SetID() require.NoError(t, err) assert.NotEqual(t, tx0.TxID, tx1.TxID) // differ Nonce
tx1 = PoolL2Tx{ FromIdx: 5, ToIdx: 5, Amount: big.NewInt(5), Fee: 126, TokenID: 5, Nonce: 4, } err = tx1.SetID() require.NoError(t, err) assert.NotEqual(t, tx0.TxID, tx1.TxID) // differ Fee
tx1 = PoolL2Tx{ FromIdx: 5, ToIdx: 5, Amount: big.NewInt(5), Fee: 124, TokenID: 5, Nonce: 5, } err = tx1.SetID() require.NoError(t, err) assert.NotEqual(t, tx0.TxID, tx1.TxID)}
|
