fix(rollup): fix l1DataFee tests

rollup/fees/rollup_fee_test.go

@@ -1,10 +1,17 @@
 package fees
 
 import (
+	"bytes"
 	"math/big"
 	"testing"
 
+	"encoding/json"
+
 	"github.com/stretchr/testify/assert"
+
+	"github.com/scroll-tech/go-ethereum/common/hexutil"
+	"github.com/scroll-tech/go-ethereum/core/types"
+	"github.com/scroll-tech/go-ethereum/crypto"
 )
 
 func TestCalculateEncodedL1DataFee(t *testing.T) {
@@ -14,7 +21,119 @@ func TestCalculateEncodedL1DataFee(t *testing.T) {
 	overhead := new(big.Int).SetUint64(100)
 	scalar := new(big.Int).SetUint64(10)
 
-	expected := new(big.Int).SetUint64(184) // 184.2
+	expected := new(big.Int).SetUint64(30) // 30.8
 	actual := calculateEncodedL1DataFee(data, overhead, l1BaseFee, scalar)
 	assert.Equal(t, expected, actual)
 }
+
+const example_tx1 = `
+	{
+		"type": 0,
+		"nonce": 4,
+		"txHash": "0x8da3fedb103b6da8ccc2514094336d1a76df166238f4d8e8558fbe54cce2516a",
+		"gas": 53854,
+		"gasPrice": "0x3b9aca00",
+		"from": "0x1c5a77d9fa7ef466951b2f01f724bca3a5820b63",
+		"to": "0x03f8133dd5ed58838b20af1296f62f44e69baa48",
+		"chainId": "0xcf55",
+		"value": "0x0",
+		"data": "0xa9059cbb000000000000000000000000c0c4c8baea3f6acb49b6e1fb9e2adeceeacb0ca200000000000000000000000000000000000000000000000000000000000003e8",
+		"isCreate": false
+	}
+`
+const example_tx2 = `
+	{
+		"type": 0,
+		"nonce": 20,
+		"txHash": "0xfef778b40acae6c4f00205f3dafae2af1dff90d402c19b090c4b12cad08e7461",
+		"gas": 23730,
+		"gasPrice": "0x3b9aca00",
+		"from": "0x1c5a77d9fa7ef466951b2f01f724bca3a5820b63",
+		"to": "0x58a2239aa5412f78d8f675c4d8ad5102a3fa5837",
+		"chainId": "0xcf55",
+		"value": "0x0",
+		"data": "0xb0f2b72a000000000000000000000000000000000000000000000000000000000000000a",
+		"isCreate": false
+	}
+`
+
+func transactionDataToMessage(txdata *types.TransactionData) types.Message {
+	databytes, err := hexutil.Decode(txdata.Data)
+	if err != nil {
+		panic(err)
+	}
+	return types.NewMessage(txdata.From, txdata.To, txdata.Nonce, (*big.Int)(txdata.Value), txdata.Gas,
+		(*big.Int)(txdata.GasPrice), (*big.Int)(txdata.GasPrice), (*big.Int)(txdata.GasPrice), databytes, nil, false)
+}
+
+type l1DataFeeTestCase struct {
+	TxDataSample      string
+	EIP1559BaseFee    *big.Int
+	L1basefee         *big.Int
+	L1feeOverHead     *big.Int
+	L1feeScalar       *big.Int
+	EncodedExpected   int
+	L1DataFeeExpected *big.Int
+}
+
+func testEstimateL1DataFeeForTransactionData(t *testing.T, t_case *l1DataFeeTestCase) {
+	txdata := new(types.TransactionData)
+	assert.NoError(t, json.Unmarshal([]byte(t_case.TxDataSample), txdata), "parse json fail")
+
+	// we have decomposed EstimateL1DataFeeForMessage here so
+	// to catch more detail inside the process
+	var msg types.Message
+
+	if t_case.EIP1559BaseFee != nil {
+		//TODO: EIP1559 test
+		panic("not implemented")
+	} else {
+		msg = transactionDataToMessage(txdata)
+	}
+	chainID := (*big.Int)(txdata.ChainId)
+	signer := types.NewLondonSigner(chainID)
+	unsigned := asUnsignedTx(msg, t_case.EIP1559BaseFee, chainID)
+
+	tx, err := unsigned.WithSignature(signer, append(bytes.Repeat([]byte{0xff}, crypto.SignatureLength-1), 0x01))
+	assert.NoError(t, err, "build dummy tx fail")
+	raw, err := rlpEncode(tx)
+	assert.NoError(t, err, "rlp fail")
+
+	if t_case.EncodedExpected != 0 {
+		assert.Equal(t, t_case.EncodedExpected, len(raw))
+	} else {
+		t.Log("caluldated encoded rlp len:", len(raw))
+	}
+
+	l1DataFee := calculateEncodedL1DataFee(raw, t_case.L1feeOverHead, t_case.L1basefee, t_case.L1feeScalar)
+	if t_case.L1DataFeeExpected != nil {
+		assert.Equal(t, t_case.L1DataFeeExpected, l1DataFee)
+	} else {
+		t.Log("calculated l1data fee:", l1DataFee)
+	}
+}
+
+func TestEstimateL1DataFeeForTransactionData(t *testing.T) {
+
+	for _, tcase := range []*l1DataFeeTestCase{
+		{
+			TxDataSample:      example_tx1,
+			L1basefee:         big.NewInt(0x64),
+			L1feeOverHead:     big.NewInt(0x17d4),
+			L1feeScalar:       big.NewInt(0x4a42fc80),
+			EncodedExpected:   173,
+			L1DataFeeExpected: big.NewInt(0xfffe8),
+		},
+		{
+			TxDataSample:      example_tx2,
+			L1basefee:         big.NewInt(0x64),
+			L1feeOverHead:     big.NewInt(0x17d4),
+			L1feeScalar:       big.NewInt(0x4a42fc80),
+			EncodedExpected:   140,
+			L1DataFeeExpected: big.NewInt(0xf3f2f),
+		},
+	} {
+		testEstimateL1DataFeeForTransactionData(t, tcase)
+	}
+
+}