/*
 * Copyright ConsenSys AG.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 *
 * SPDX-License-Identifier: Apache-2.0
 */
package org.hyperledger.besu.ethereum.core.encoding;

import static org.assertj.core.api.Assertions.assertThat;

import org.hyperledger.besu.ethereum.core.Block;
import org.hyperledger.besu.ethereum.core.BlockDataGenerator;
import org.hyperledger.besu.ethereum.core.Transaction;
import org.hyperledger.besu.ethereum.mainnet.MainnetBlockHeaderFunctions;
import org.hyperledger.besu.ethereum.rlp.BytesValueRLPOutput;
import org.hyperledger.besu.ethereum.rlp.RLP;
import org.hyperledger.besu.ethereum.rlp.RLPInput;

import org.apache.tuweni.bytes.Bytes;
import org.junit.jupiter.api.Test;

class TransactionRLPEncoderTest {
  private static final String FRONTIER_TX_RLP =
      "0xf901fc8032830138808080b901ae60056013565b6101918061001d6000396000f35b3360008190555056006001600060e060020a6000350480630a874df61461003a57806341c0e1b514610058578063a02b161e14610066578063dbbdf0831461007757005b610045600435610149565b80600160a060020a031660005260206000f35b610060610161565b60006000f35b6100716004356100d4565b60006000f35b61008560043560243561008b565b60006000f35b600054600160a060020a031632600160a060020a031614156100ac576100b1565b6100d0565b8060018360005260205260406000208190555081600060005260206000a15b5050565b600054600160a060020a031633600160a060020a031614158015610118575033600160a060020a0316600182600052602052604060002054600160a060020a031614155b61012157610126565b610146565b600060018260005260205260406000208190555080600060005260206000a15b50565b60006001826000526020526040600020549050919050565b600054600160a060020a031633600160a060020a0316146101815761018f565b600054600160a060020a0316ff5b561ca0c5689ed1ad124753d54576dfb4b571465a41900a1dff4058d8adf16f752013d0a01221cbd70ec28c94a3b55ec771bcbc70778d6ee0b51ca7ea9514594c861b1884";
  private static final String EIP1559_TX_RLP =
      "0xb8a902f8a686796f6c6f7632800285012a05f20082753094000000000000000000000000000000000000aaaa8080f838f794000000000000000000000000000000000000aaaae1a0000000000000000000000000000000000000000000000000000000000000000001a00c1d69648e348fe26155b45de45004f0e4195f6352d8f0935bc93e98a3e2a862a060064e5b9765c0ac74223b0cf49635c59ae0faf82044fd17bcc68a549ade6f95";
  private static final String NONCE_64_BIT_MAX_MINUS_2_TX_RLP =
      "0xf86788fffffffffffffffe0182520894095e7baea6a6c7c4c2dfeb977efac326af552d8780801ba048b55bfa915ac795c431978d8a6a992b628d557da5ff759b307d495a36649353a01fffd310ac743f371de3b9f7f9cb56c0b28ad43601b4ab949f53faa07bd2c804";

  @Test
  void encodeFrontierTxNominalCase() {
    final Transaction transaction = decodeRLP(RLP.input(Bytes.fromHexString(FRONTIER_TX_RLP)));
    final BytesValueRLPOutput output = new BytesValueRLPOutput();
    encodeRLP(transaction, output);
    assertThat(output.encoded().toHexString()).isEqualTo(FRONTIER_TX_RLP);
  }

  @Test
  void encodeEIP1559TxNominalCase() {
    final Transaction transaction = decodeRLP(RLP.input(Bytes.fromHexString(EIP1559_TX_RLP)));
    final BytesValueRLPOutput output = new BytesValueRLPOutput();
    encodeRLP(transaction, output);
    assertThat(output.encoded().toHexString()).isEqualTo(EIP1559_TX_RLP);
  }

  @Test
  void blockWithLegacyAndEIP2930TransactionsRoundTrips() {
    final BlockDataGenerator gen = new BlockDataGenerator();

    final Block block =
        gen.block(
            BlockDataGenerator.BlockOptions.create()
                .addTransaction(gen.transactionsWithAllTypes().toArray(new Transaction[] {})));

    assertThat(
            Block.readFrom(
                RLP.input(RLP.encode(block::writeTo)), new MainnetBlockHeaderFunctions()))
        .isEqualTo(block);
  }

  @Test
  void shouldEncodeWithHighNonce() {
    final Transaction transaction =
        decodeRLP(RLP.input(Bytes.fromHexString(NONCE_64_BIT_MAX_MINUS_2_TX_RLP)));
    final BytesValueRLPOutput output = new BytesValueRLPOutput();
    encodeRLP(transaction, output);
    assertThat(output.encoded().toHexString()).isEqualTo(NONCE_64_BIT_MAX_MINUS_2_TX_RLP);
  }

  private Transaction decodeRLP(final RLPInput input) {
    return TransactionDecoder.decodeRLP(input, EncodingContext.BLOCK_BODY);
  }

  private void encodeRLP(final Transaction transaction, final BytesValueRLPOutput output) {
    TransactionEncoder.encodeRLP(transaction, output, EncodingContext.BLOCK_BODY);
  }
}
