def test_single_dependencies():
deps = {
'Test-1': set(('Dep1', 'Dep2')),
'Test-2': set(('Dep1', )),
}
assert get_dependencies('Test-1', deps) == {'Dep1', 'Dep2'}
assert get_dependencies('Test-2', deps) == {'Dep1'}
def test_single_dependencies():
deps = {
'Test-1': set(('Dep1', 'Dep2')),
'Test-2': set(('Dep1',)),
}
assert get_dependencies('Test-1', deps) == {'Dep1', 'Dep2'}
assert get_dependencies('Test-2', deps) == {'Dep1'}
def test_multiple_dependencies():
deps = {
'Test-1': set(('Dep1', 'Dep2', 'Test-3')),
'Test-2': set(('Dep1', 'Test-1')),
'Test-3': set(('Dep3',))
}
assert get_dependencies('Test-2', deps) == {'Dep1', 'Dep2', 'Dep3', 'Test-1', 'Test-3'}
def test_multiple_dependencies():
deps = {
'Test-1': set(('Dep1', 'Dep2', 'Test-3')),
'Test-2': set(('Dep1', 'Test-1')),
'Test-3': set(('Dep3', ))
}
assert get_dependencies(
'Test-2', deps) == {'Dep1', 'Dep2', 'Dep3', 'Test-1', 'Test-3'}
def deploy_contracts(deploy_client,
contracts,
deploy_at_block=0,
max_wait_for_deploy=0,
from_address=None,
max_wait=0,
contracts_to_deploy=None,
dependencies=None,
constructor_args=None,
deploy_gas=None):
_deployed_contracts = {}
_receipts = {}
if constructor_args is None:
constructor_args = {}
if dependencies is None:
dependencies = {}
# Potentiall wait until we've reached a specific block.
deploy_client.wait_for_block(deploy_at_block, max_wait_for_deploy)
# Extract and dependencies that exist due to library linking.
linker_dependencies = get_linker_dependencies(contracts)
deploy_dependencies = merge_dependencies(
dependencies,
linker_dependencies,
)
# If a subset of contracts have been specified to be deployed, compute
# their dependencies as well.
contracts_to_deploy = set(
itertools.chain.from_iterable(
get_dependencies(contract_name, deploy_dependencies)
for contract_name in (
contracts_to_deploy or []))).union(contracts_to_deploy)
# If there are any dependencies either explicit or from libraries, sort the
# contracts by their dependencies.
if deploy_dependencies:
dependencies = copy.copy(deploy_dependencies)
for contract_name, _ in contracts:
if contract_name not in deploy_dependencies:
dependencies[contract_name] = set()
sorted_contract_names = toposort.toposort_flatten(dependencies)
contracts = sorted(contracts,
key=lambda c: sorted_contract_names.index(c[0]))
if from_address is None:
from_address = deploy_client.get_coinbase()
for contract_name, contract_class in contracts:
# If a subset of contracts have been specified, only deploy those or
# the contracts they depend upon.
if contracts_to_deploy and contract_name not in contracts_to_deploy:
continue
args = constructor_args.get(contract_name, None)
if callable(args):
args = args(_deployed_contracts)
if deploy_gas is None:
deploy_gas_limit = int(deploy_client.get_max_gas() * 0.98)
elif callable(deploy_gas):
deploy_gas_limit = deploy_gas(contract_name, contract_class)
else:
deploy_gas_limit = deploy_gas
if contract_name in linker_dependencies:
for dependency_name in linker_dependencies[contract_name]:
deployed_contract = _deployed_contracts[dependency_name]
link_contract_dependency(contract_class, deployed_contract)
txn_hash = deploy_contract(
deploy_client,
contract_class,
constructor_args=args,
_from=from_address,
gas=deploy_gas_limit,
)
contract_addr = get_contract_address_from_txn(
deploy_client,
txn_hash,
max_wait=max_wait,
)
_receipts[contract_name] = deploy_client.wait_for_transaction(
txn_hash,
max_wait,
)
_deployed_contracts[contract_name] = contract_class(
contract_addr,
deploy_client,
)
_dict = {
'_deploy_receipts': _receipts,
'__len__': lambda s: len(_deployed_contracts),
'__iter__': lambda s: iter(_deployed_contracts.items()),
'__getitem__': lambda s, k: _deployed_contracts.__getitem__[k],
}
_dict.update(_deployed_contracts)
return type('deployed_contracts', (object, ), _dict)()
