mirror of https://github.com/nucypher/nucypher.git
158 lines
5.4 KiB
Python
158 lines
5.4 KiB
Python
"""
|
|
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 contextlib
|
|
import json
|
|
import os
|
|
import shutil
|
|
|
|
import pytest
|
|
from click.testing import CliRunner
|
|
|
|
from nucypher.blockchain.eth.registry import AllocationRegistry
|
|
from nucypher.config.characters import UrsulaConfiguration
|
|
from nucypher.utilities.sandbox.constants import (
|
|
MOCK_CUSTOM_INSTALLATION_PATH,
|
|
MOCK_ALLOCATION_INFILE,
|
|
MOCK_REGISTRY_FILEPATH,
|
|
ONE_YEAR_IN_SECONDS,
|
|
MINERS_ESCROW_DEPLOYMENT_SECRET,
|
|
POLICY_MANAGER_DEPLOYMENT_SECRET,
|
|
MINING_ADJUDICATOR_DEPLOYMENT_SECRET,
|
|
USER_ESCROW_PROXY_DEPLOYMENT_SECRET)
|
|
from nucypher.utilities.sandbox.constants import MOCK_CUSTOM_INSTALLATION_PATH_2, INSECURE_DEVELOPMENT_PASSWORD
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def click_runner():
|
|
runner = CliRunner()
|
|
yield runner
|
|
|
|
|
|
@pytest.fixture(scope='session')
|
|
def deploy_user_input():
|
|
account_index = '0\n'
|
|
yes = 'Y\n'
|
|
node_password = f'{INSECURE_DEVELOPMENT_PASSWORD}\n'
|
|
deployment_secret = f'{INSECURE_DEVELOPMENT_PASSWORD}\n'
|
|
user_input = account_index + yes + node_password + (deployment_secret * 8) + 'DEPLOY'
|
|
return user_input
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def nominal_federated_configuration_fields():
|
|
config = UrsulaConfiguration(dev_mode=True, federated_only=True)
|
|
config_fields = config.static_payload
|
|
del config_fields['is_me']
|
|
yield tuple(config_fields.keys())
|
|
del config
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def mock_allocation_infile(testerchain, token_economics):
|
|
accounts = testerchain.unassigned_accounts
|
|
allocation_data = [{'address': addr,
|
|
'amount': token_economics.minimum_allowed_locked,
|
|
'duration': ONE_YEAR_IN_SECONDS}
|
|
for addr in accounts]
|
|
|
|
with open(MOCK_ALLOCATION_INFILE, 'w') as file:
|
|
file.write(json.dumps(allocation_data))
|
|
|
|
registry = AllocationRegistry(registry_filepath=MOCK_ALLOCATION_INFILE)
|
|
yield registry
|
|
os.remove(MOCK_ALLOCATION_INFILE)
|
|
|
|
|
|
@pytest.fixture(scope='module', autouse=True)
|
|
def mock_primary_registry_filepath():
|
|
yield MOCK_REGISTRY_FILEPATH
|
|
if os.path.isfile(MOCK_REGISTRY_FILEPATH):
|
|
os.remove(MOCK_REGISTRY_FILEPATH)
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def custom_filepath():
|
|
_custom_filepath = MOCK_CUSTOM_INSTALLATION_PATH
|
|
with contextlib.suppress(FileNotFoundError):
|
|
shutil.rmtree(_custom_filepath, ignore_errors=True)
|
|
try:
|
|
yield _custom_filepath
|
|
finally:
|
|
with contextlib.suppress(FileNotFoundError):
|
|
shutil.rmtree(_custom_filepath, ignore_errors=True)
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def custom_filepath_2():
|
|
_custom_filepath = MOCK_CUSTOM_INSTALLATION_PATH_2
|
|
with contextlib.suppress(FileNotFoundError):
|
|
shutil.rmtree(_custom_filepath, ignore_errors=True)
|
|
try:
|
|
yield _custom_filepath
|
|
finally:
|
|
with contextlib.suppress(FileNotFoundError):
|
|
shutil.rmtree(_custom_filepath, ignore_errors=True)
|
|
|
|
|
|
@pytest.fixture(scope='session')
|
|
def deployed_blockchain(token_economics):
|
|
|
|
# Interface
|
|
compiler = SolidityCompiler()
|
|
registry = InMemoryEthereumContractRegistry()
|
|
allocation_registry = InMemoryAllocationRegistry()
|
|
interface = BlockchainDeployerInterface(compiler=compiler,
|
|
registry=registry,
|
|
provider_uri=TEST_PROVIDER_URI)
|
|
|
|
# Blockchain
|
|
blockchain = TesterBlockchain(interface=interface, eth_airdrop=True, test_accounts=5, poa=True)
|
|
deployer_address = blockchain.etherbase_account
|
|
|
|
# Deployer
|
|
deployer = Deployer(blockchain=blockchain, deployer_address=deployer_address)
|
|
|
|
# The Big Three (+ Dispatchers)
|
|
deployer.deploy_network_contracts(miner_secret=MINERS_ESCROW_DEPLOYMENT_SECRET,
|
|
policy_secret=POLICY_MANAGER_DEPLOYMENT_SECRET,
|
|
adjudicator_secret=MINING_ADJUDICATOR_DEPLOYMENT_SECRET,
|
|
user_escrow_proxy_secret=USER_ESCROW_PROXY_DEPLOYMENT_SECRET)
|
|
|
|
# Start with some hard-coded cases...
|
|
all_yall = blockchain.unassigned_accounts
|
|
allocation_data = [{'address': all_yall[1],
|
|
'amount': token_economics.maximum_allowed_locked,
|
|
'duration': ONE_YEAR_IN_SECONDS}]
|
|
|
|
deployer.deploy_beneficiary_contracts(allocations=allocation_data, allocation_registry=allocation_registry)
|
|
|
|
yield blockchain, deployer_address, registry
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def custom_filepath_2():
|
|
_custom_filepath = MOCK_CUSTOM_INSTALLATION_PATH_2
|
|
with contextlib.suppress(FileNotFoundError):
|
|
shutil.rmtree(_custom_filepath, ignore_errors=True)
|
|
try:
|
|
yield _custom_filepath
|
|
finally:
|
|
with contextlib.suppress(FileNotFoundError):
|
|
shutil.rmtree(_custom_filepath, ignore_errors=True)
|