|
|
package common
import ( "crypto/ecdsa" "encoding/hex" "log" "math/big" "testing"
ethCommon "github.com/ethereum/go-ethereum/common" "github.com/ethereum/go-ethereum/crypto" "github.com/iden3/go-iden3-crypto/babyjub" "github.com/iden3/go-iden3-crypto/utils" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require")
func TestNewL1UserTx(t *testing.T) { toForge := int64(123456) l1Tx := &L1Tx{ ToForgeL1TxsNum: &toForge, Position: 71, UserOrigin: true, ToIdx: 301, TokenID: 5, Amount: big.NewInt(1), LoadAmount: big.NewInt(2), FromIdx: Idx(300), } l1Tx, err := NewL1Tx(l1Tx) assert.Nil(t, err) assert.Equal(t, "0x00000000000001e240004700", l1Tx.TxID.String())}
func TestNewL1CoordinatorTx(t *testing.T) { batchNum := BatchNum(51966) l1Tx := &L1Tx{ Position: 88, UserOrigin: false, ToIdx: 301, TokenID: 5, Amount: big.NewInt(1), LoadAmount: big.NewInt(2), FromIdx: Idx(300), BatchNum: &batchNum, } l1Tx, err := NewL1Tx(l1Tx) assert.Nil(t, err) assert.Equal(t, "0x01000000000000cafe005800", l1Tx.TxID.String())}
func TestL1userTxByteParsers(t *testing.T) { var pkComp babyjub.PublicKeyComp pkCompL := []byte("0x56ca90f80d7c374ae7485e9bcc47d4ac399460948da6aeeb899311097925a72c") err := pkComp.UnmarshalText(pkCompL) require.Nil(t, err)
pk, err := pkComp.Decompress() require.Nil(t, err)
l1Tx := &L1Tx{ UserOrigin: true, ToIdx: 3, TokenID: 5, Amount: big.NewInt(1), LoadAmount: big.NewInt(2), FromIdx: 2, FromBJJ: pk, FromEthAddr: ethCommon.HexToAddress("0xc58d29fA6e86E4FAe04DDcEd660d45BCf3Cb2370"), }
encodedData, err := l1Tx.BytesUser() require.Nil(t, err) decodedData, err := L1UserTxFromBytes(encodedData) require.Nil(t, err) assert.Equal(t, l1Tx, decodedData) encodedData2, err := decodedData.BytesUser() require.Nil(t, err) assert.Equal(t, encodedData, encodedData2)
// expect error if length!=68
_, err = L1UserTxFromBytes(encodedData[:66]) require.NotNil(t, err) _, err = L1UserTxFromBytes([]byte{}) require.NotNil(t, err) _, err = L1UserTxFromBytes(nil) require.NotNil(t, err)}
func TestL1TxByteParsersCompatibility(t *testing.T) { // Data from compatibility test
var pkComp babyjub.PublicKeyComp pkCompB, err := hex.DecodeString("0dd02deb2c81068e7a0f7e327df80b4ab79ee1f41a7def613e73a20c32eece5a") require.Nil(t, err) pkCompL := SwapEndianness(pkCompB) err = pkComp.UnmarshalText([]byte(hex.EncodeToString(pkCompL))) require.Nil(t, err)
pk, err := pkComp.Decompress() require.Nil(t, err)
loadAmount := new(big.Int) loadAmount.SetString("100000000000000000000", 10) l1Tx := &L1Tx{ ToIdx: 87865485, TokenID: 2098076, Amount: big.NewInt(2400000000000000000), LoadAmount: loadAmount, FromIdx: Idx(29767899), FromBJJ: pk, FromEthAddr: ethCommon.HexToAddress("0x85dab5b9e2e361d0c208d77be90efcc0439b0a53"), UserOrigin: true, }
expected, err := utils.HexDecode("85dab5b9e2e361d0c208d77be90efcc0439b0a530dd02deb2c81068e7a0f7e327df80b4ab79ee1f41a7def613e73a20c32eece5a000001c638db8be880f00020039c0000053cb88d") require.Nil(t, err)
encodedData, err := l1Tx.BytesUser() require.Nil(t, err) assert.Equal(t, expected, encodedData)}
func TestL1CoordinatorTxByteParsers(t *testing.T) { privateKey, err := crypto.HexToECDSA("fad9c8855b740a0b7ed4c221dbad0f33a83a49cad6b3fe8d5817ac83d38b6a19") require.Nil(t, err)
publicKey := privateKey.Public() publicKeyECDSA, ok := publicKey.(*ecdsa.PublicKey) if !ok { log.Fatal("error casting public key to ECDSA") } publicKeyBytes := crypto.FromECDSAPub(publicKeyECDSA) pubKey, err := crypto.UnmarshalPubkey(publicKeyBytes) require.Nil(t, err) fromEthAddr := crypto.PubkeyToAddress(*pubKey) var pkComp babyjub.PublicKeyComp pkCompL := []byte("56ca90f80d7c374ae7485e9bcc47d4ac399460948da6aeeb899311097925a72c") err = pkComp.UnmarshalText(pkCompL) require.Nil(t, err) pk, err := pkComp.Decompress() require.Nil(t, err) bytesMessage1 := []byte("\x19Ethereum Signed Message:\n98") bytesMessage2 := []byte("I authorize this babyjubjub key for hermez rollup account creation")
babyjub := pk.Compress() babyjubB := SwapEndianness(babyjub[:]) var data []byte data = append(data, bytesMessage1...) data = append(data, bytesMessage2...) data = append(data, babyjubB[:]...) hash := crypto.Keccak256Hash(data) signature, err := crypto.Sign(hash.Bytes(), privateKey) require.Nil(t, err) // Ethereum adds 27 to v
v := int(signature[64]) signature[64] = byte(v + 27)
l1Tx := &L1Tx{ TokenID: 231, FromBJJ: pk, FromEthAddr: fromEthAddr, Amount: big.NewInt(0), LoadAmount: big.NewInt(0), }
bytesCoordinatorL1, err := l1Tx.BytesCoordinatorTx(signature) require.Nil(t, err) l1txDecoded, err := L1CoordinatorTxFromBytes(bytesCoordinatorL1) require.Nil(t, err) assert.Equal(t, l1Tx, l1txDecoded) bytesCoordinatorL12, err := l1txDecoded.BytesCoordinatorTx(signature) require.Nil(t, err) assert.Equal(t, bytesCoordinatorL1, bytesCoordinatorL12)
// expect error if length!=68
_, err = L1CoordinatorTxFromBytes(bytesCoordinatorL1[:66]) require.NotNil(t, err) _, err = L1CoordinatorTxFromBytes([]byte{}) require.NotNil(t, err) _, err = L1CoordinatorTxFromBytes(nil) require.NotNil(t, err)}
func TestL1CoordinatorTxByteParsersCompatibility(t *testing.T) { // Data from compatibility test
var signature []byte r, err := hex.DecodeString("da71e5eb097e115405d84d1e7b464009b434b32c014a2df502d1f065ced8bc3b") require.Nil(t, err) s, err := hex.DecodeString("186d7122ff7f654cfed3156719774898d573900c86599a885a706dbdffe5ea8c") require.Nil(t, err) v, err := hex.DecodeString("1b") require.Nil(t, err)
signature = append(signature, r[:]...) signature = append(signature, s[:]...) signature = append(signature, v[:]...)
var pkComp babyjub.PublicKeyComp pkCompB, err := hex.DecodeString("a2c2807ee39c3b3378738cff85a46a9465bb8fcf44ea597c33da9719be7c259c") require.Nil(t, err) pkCompL := SwapEndianness(pkCompB) err = pkComp.UnmarshalText([]byte(hex.EncodeToString(pkCompL))) require.Nil(t, err) // Data from the compatibility test
require.Nil(t, err) pk, err := pkComp.Decompress() require.Nil(t, err) l1Tx := &L1Tx{ TokenID: 231, FromBJJ: pk, }
encodeData, err := l1Tx.BytesCoordinatorTx(signature) require.Nil(t, err)
expected, err := utils.HexDecode("1b186d7122ff7f654cfed3156719774898d573900c86599a885a706dbdffe5ea8cda71e5eb097e115405d84d1e7b464009b434b32c014a2df502d1f065ced8bc3ba2c2807ee39c3b3378738cff85a46a9465bb8fcf44ea597c33da9719be7c259c000000e7") require.Nil(t, err)
assert.Equal(t, expected, encodeData)}
|
