/*
 * 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.tests.acceptance.plugins;

import org.hyperledger.besu.datatypes.Hash;
import org.hyperledger.besu.tests.acceptance.dsl.AcceptanceTestBase;
import org.hyperledger.besu.tests.acceptance.dsl.account.Account;
import org.hyperledger.besu.tests.acceptance.dsl.blockchain.Amount;
import org.hyperledger.besu.tests.acceptance.dsl.node.BesuNode;
import org.hyperledger.besu.tests.acceptance.dsl.node.configuration.BesuNodeConfigurationBuilder;
import org.hyperledger.besu.tests.acceptance.dsl.transaction.account.TransferTransaction;

import java.util.List;

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;

public class PermissioningPluginTest extends AcceptanceTestBase {
  private BesuNode minerNode;

  private BesuNode aliceNode;
  private BesuNode bobNode;
  private BesuNode charlieNode;

  @BeforeEach
  public void setUp() throws Exception {
    final BesuNodeConfigurationBuilder builder =
        new BesuNodeConfigurationBuilder()
            .miningEnabled(false)
            .plugins(List.of("testPlugins"))
            .extraCLIOptions(List.of("--plugin-permissioning-test-enabled=true"))
            .jsonRpcEnabled()
            .jsonRpcTxPool()
            .jsonRpcAdmin();

    minerNode = besu.create(builder.name("miner").build());

    aliceNode = besu.create(builder.name("alice").keyFilePath("key").build());

    bobNode = besu.create(builder.name("bob").keyFilePath("key1").build());

    charlieNode = besu.create(builder.name("charlie").keyFilePath("key2").build());

    cluster.start(minerNode, charlieNode);

    cluster.startNode(aliceNode);
    aliceNode.awaitPeerDiscovery(net.awaitPeerCount(2));

    cluster.startNode(bobNode);
    bobNode.awaitPeerDiscovery(net.awaitPeerCount(2));
  }

  @Test
  public void blockedConnectionNodeCanOnlyConnectToTransactionNode() {
    minerNode.verify(admin.hasPeer(aliceNode));
    minerNode.verify(admin.hasPeer(bobNode));
    minerNode.verify(admin.hasPeer(charlieNode));

    aliceNode.verify(admin.doesNotHavePeer(bobNode));
    aliceNode.verify(admin.hasPeer(minerNode));
    aliceNode.verify(admin.hasPeer(charlieNode));

    bobNode.verify(admin.hasPeer(minerNode));
    bobNode.verify(admin.doesNotHavePeer(aliceNode));
    bobNode.verify(admin.hasPeer(charlieNode));

    charlieNode.verify(admin.hasPeer(minerNode));
    charlieNode.verify(admin.hasPeer(aliceNode));
    charlieNode.verify(admin.hasPeer(bobNode));
  }

  @Test
  public void transactionsAreNotSendToBlockPendingTransactionsNode() {
    final Account account = accounts.createAccount("account-one");
    final Amount balance = Amount.ether(20);

    final TransferTransaction tx = accountTransactions.createTransfer(account, balance);

    final Hash txHash = aliceNode.execute(tx);

    aliceNode.verify(txPoolConditions.inTransactionPool(txHash));
    bobNode.verify(txPoolConditions.inTransactionPool(txHash));
    charlieNode.verify(txPoolConditions.notInTransactionPool(txHash));
    minerNode.verify(txPoolConditions.inTransactionPool(txHash));
  }
}
