UT for marshal/unmarshal of log

0xPolygon · May 2, 2023 · c76923c · c76923c
1 parent 3933af1
commit c76923c
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/consensus/polybft/stake_manager_test.go b/consensus/polybft/stake_manager_test.go
@@ -6,6 +6,7 @@ import (
 
 	"github.com/0xPolygon/polygon-edge/consensus/polybft/contractsapi"
 	"github.com/0xPolygon/polygon-edge/consensus/polybft/wallet"
+	"github.com/0xPolygon/polygon-edge/contracts"
 	"github.com/0xPolygon/polygon-edge/helper/hex"
 	"github.com/0xPolygon/polygon-edge/txrelayer"
 	"github.com/0xPolygon/polygon-edge/types"
@@ -248,6 +249,29 @@ func TestStakeCounter_ShouldBeDeterministic(t *testing.T) {
 	}
 }
 
+func TestStakeManager_RLPEncoding_Log(t *testing.T) {
+	t.Parallel()
+
+	testLog := createTestLogForTransferEvent(t, contracts.ValidatorSetContract,
+		types.StringToAddress("0x00001111111"), types.StringToAddress("0x00002222222"), 10000000000000)
+
+	receipt := &types.Receipt{
+		Root:              types.ZeroHash,
+		CumulativeGasUsed: 1111111,
+		Logs:              []*types.Log{testLog},
+		GasUsed:           1111111,
+		TxHash:            types.ZeroHash,
+		ContractAddress:   &contracts.ValidatorSetContract,
+	}
+
+	marshalled := receipt.MarshalRLP()
+
+	unmarshalledReceipt := &types.Receipt{}
+	require.NoError(t, unmarshalledReceipt.UnmarshalRLP(marshalled))
+	require.Len(t, receipt.Logs, 1)
+	require.Equal(t, receipt.Logs[0].Data, testLog.Data)
+}
+
 func createTestLogForTransferEvent(t *testing.T, validatorSet, from, to types.Address, stake uint64) *types.Log {
 	t.Helper()