"""
This file is part of nucypher.

nucypher is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

nucypher is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Affero General Public License for more details.

You should have received a copy of the GNU Affero General Public License
along with nucypher.  If not, see <https://www.gnu.org/licenses/>.
"""


import pytest
from nucypher.utilities.sandbox.constants import NUMBER_OF_ETH_TEST_ACCOUNTS, NUMBER_OF_URSULAS_IN_BLOCKCHAIN_TESTS, \
    DEVELOPMENT_ETH_AIRDROP_AMOUNT, TEST_PROVIDER_URI

from nucypher.blockchain.eth.interfaces import BlockchainDeployerInterface
from nucypher.blockchain.eth.registry import InMemoryEthereumContractRegistry
# Prevents TesterBlockchain to be picked up by py.test as a test class
from nucypher.utilities.sandbox.blockchain import TesterBlockchain as _TesterBlockchain


@pytest.fixture()
def another_testerchain(solidity_compiler):
    memory_registry = InMemoryEthereumContractRegistry()
    deployer_interface = BlockchainDeployerInterface(compiler=solidity_compiler,
                                                     registry=memory_registry,
                                                     provider_uri=TEST_PROVIDER_URI)
    testerchain = _TesterBlockchain(interface=deployer_interface,
                                   test_accounts=2*NUMBER_OF_ETH_TEST_ACCOUNTS,
                                   eth_airdrop=True)
    deployer_interface.deployer_address = testerchain.etherbase_account
    yield testerchain
    testerchain.sever_connection()


def test_testerchain_creation(testerchain, another_testerchain):

    chains = (testerchain, another_testerchain)

    for chain in chains:

        # Ensure we are testing on the correct network...
        assert 'tester' in chain.interface.provider_uri

        # ... and that there are already some blocks mined
        assert chain.interface.w3.eth.blockNumber > 0

        # Check that we have enough test accounts
        assert len(chain.interface.w3.eth.accounts) >= NUMBER_OF_ETH_TEST_ACCOUNTS

        # Check that distinguished accounts are assigned
        etherbase = chain.etherbase_account
        assert etherbase == chain.interface.w3.eth.accounts[0]

        alice = chain.alice_account
        assert alice == chain.interface.w3.eth.accounts[1]

        bob = chain.bob_account
        assert bob == chain.interface.w3.eth.accounts[2]

        ursulas = [chain.ursula_account(i) for i in range(NUMBER_OF_URSULAS_IN_BLOCKCHAIN_TESTS)]
        assert ursulas == chain.ursulas_accounts

        # Check that the remaining accounts are different from the previous ones:
        assert set([etherbase, alice, bob] + ursulas).isdisjoint(set(chain.unassigned_accounts))

        # Check that accounts are funded
        for account in chain.interface.w3.eth.accounts:
            assert chain.interface.w3.eth.getBalance(account) >= DEVELOPMENT_ETH_AIRDROP_AMOUNT

        # Check that accounts can send transactions
        for account in chain.interface.w3.eth.accounts:
            balance = chain.interface.w3.eth.getBalance(account)
            assert balance

            tx = {'to': etherbase, 'from': account, 'value': 100}
            txhash = chain.interface.w3.eth.sendTransaction(tx)
            _receipt = chain.wait_for_receipt(txhash)