/*
 * 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.api.jsonrpc.internal.methods;

import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatThrownBy;
import static org.mockito.ArgumentMatchers.anyLong;
import static org.mockito.Mockito.verifyNoMoreInteractions;
import static org.mockito.Mockito.when;

import org.hyperledger.besu.datatypes.Hash;
import org.hyperledger.besu.datatypes.Wei;
import org.hyperledger.besu.ethereum.api.jsonrpc.internal.JsonRpcRequest;
import org.hyperledger.besu.ethereum.api.jsonrpc.internal.JsonRpcRequestContext;
import org.hyperledger.besu.ethereum.api.jsonrpc.internal.exception.InvalidJsonRpcParameters;
import org.hyperledger.besu.ethereum.api.jsonrpc.internal.response.JsonRpcResponse;
import org.hyperledger.besu.ethereum.api.jsonrpc.internal.response.JsonRpcSuccessResponse;
import org.hyperledger.besu.ethereum.api.query.BlockWithMetadata;
import org.hyperledger.besu.ethereum.api.query.BlockchainQueries;
import org.hyperledger.besu.ethereum.api.query.TransactionWithMetadata;
import org.hyperledger.besu.ethereum.chain.Blockchain;
import org.hyperledger.besu.ethereum.core.BlockHeader;
import org.hyperledger.besu.ethereum.core.BlockHeaderTestFixture;
import org.hyperledger.besu.ethereum.core.Difficulty;
import org.hyperledger.besu.ethereum.mainnet.ProtocolSchedule;
import org.hyperledger.besu.ethereum.mainnet.ProtocolSpec;

import java.util.Collections;
import java.util.Optional;

import org.assertj.core.util.Arrays;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnitRunner;

@RunWith(MockitoJUnitRunner.class)
public class EthGetMinerDataByBlockNumberTest {
  @Mock private BlockchainQueries blockchainQueries;
  @Mock private ProtocolSchedule protocolSchedule;
  @Mock private ProtocolSpec protocolSpec;
  @Mock private Blockchain blockChain;
  private EthGetMinerDataByBlockNumber method;
  private final String ETH_METHOD = "eth_getMinerDataByBlockNumber";
  private final BlockHeaderTestFixture blockHeaderTestFixture = new BlockHeaderTestFixture();

  @Before
  public void before() {
    this.method = new EthGetMinerDataByBlockNumber(blockchainQueries, protocolSchedule);
  }

  @Test
  public void shouldReturnExpectedMethodNameTest() {
    assertThat(method.getName()).isEqualTo(ETH_METHOD);
  }

  @Test
  public void successTest() {
    final BlockHeader header = blockHeaderTestFixture.buildHeader();
    final BlockWithMetadata<TransactionWithMetadata, Hash> blockWithMetadata =
        new BlockWithMetadata<>(
            header, Collections.emptyList(), Collections.emptyList(), Difficulty.of(100L), 5);

    when(blockchainQueries.blockByNumber(anyLong())).thenReturn(Optional.of(blockWithMetadata));
    when(protocolSchedule.getByBlockHeader(header)).thenReturn(protocolSpec);
    when(protocolSpec.getBlockReward()).thenReturn(Wei.fromEth(2));
    when(blockchainQueries.getBlockchain()).thenReturn(blockChain);

    JsonRpcRequest request = new JsonRpcRequest("2.0", ETH_METHOD, Arrays.array("5094833"));
    JsonRpcRequestContext requestContext = new JsonRpcRequestContext(request);
    JsonRpcResponse response = method.response(requestContext);

    assertThat(response).isNotNull().isInstanceOf(JsonRpcSuccessResponse.class);
    assertThat(((JsonRpcSuccessResponse) response).getResult()).isNotNull();
    assertThat(((JsonRpcSuccessResponse) response).getResult())
        .hasFieldOrProperty("netBlockReward")
        .hasFieldOrProperty("staticBlockReward")
        .hasFieldOrProperty("transactionFee")
        .hasFieldOrProperty("uncleInclusionReward")
        .hasFieldOrProperty("uncleRewards")
        .hasFieldOrProperty("coinbase")
        .hasFieldOrProperty("extraData")
        .hasFieldOrProperty("difficulty")
        .hasFieldOrProperty("totalDifficulty");
  }

  @Test
  public void exceptionWhenNoNumberSuppliedTest() {
    JsonRpcRequest request = new JsonRpcRequest("2.0", ETH_METHOD, Arrays.array());
    JsonRpcRequestContext requestContext = new JsonRpcRequestContext(request);
    assertThatThrownBy(() -> method.response(requestContext))
        .isInstanceOf(InvalidJsonRpcParameters.class)
        .hasMessage("Missing required json rpc parameter at index 0");

    verifyNoMoreInteractions(blockchainQueries);
  }

  @Test
  public void exceptionWhenNumberParamInvalidTest() {
    JsonRpcRequest request = new JsonRpcRequest("2.0", ETH_METHOD, Arrays.array("number"));
    JsonRpcRequestContext requestContext = new JsonRpcRequestContext(request);
    assertThatThrownBy(() -> method.response(requestContext))
        .isInstanceOf(InvalidJsonRpcParameters.class)
        .hasMessageContaining("Invalid json rpc parameter at index 0");

    verifyNoMoreInteractions(blockchainQueries);
  }
}
