def test_linker_with_callback(escrow_deployer, w3):
sender = w3.eth.accounts[0]
recipient = w3.eth.accounts[5]
def callback_fn(package):
escrow_instance = package.deployments.get_instance("Escrow")
tx_hash = escrow_instance.functions.releaseFunds().transact(
{"from": sender})
w3.eth.waitForTransactionReceipt(tx_hash)
escrow_strategy = linker(
deploy("SafeSendLib", transaction={"from": sender}),
link("Escrow", "SafeSendLib"),
deploy(
"Escrow",
recipient,
transaction={
"from": sender,
"value": w3.toWei("1", "ether")
},
),
run_python(callback_fn),
)
escrow_deployer.register_strategy("Escrow", escrow_strategy)
assert w3.eth.getBalance(recipient) == w3.toWei("1000000", "ether")
linked_escrow_package = escrow_deployer.deploy("Escrow")
escrow_instance = linked_escrow_package.deployments.get_instance("Escrow")
assert escrow_instance.functions.sender().call() == sender
assert w3.eth.getBalance(recipient) == w3.toWei("1000001", "ether")
def test_linker_with_from(escrow_deployer, w3):
escrow_strategy = linker(
deploy("SafeSendLib"),
link("Escrow", "SafeSendLib"),
deploy("Escrow", w3.eth.accounts[0], transaction={"from": w3.eth.accounts[5]}),
)
escrow_deployer.register_strategy("Escrow", escrow_strategy)
linked_escrow_package = escrow_deployer.deploy("Escrow")
escrow_instance = linked_escrow_package.deployments.get_instance("Escrow")
assert escrow_instance.functions.sender().call() == w3.eth.accounts[5]
def escrow_package(w3, deployer):
manifest = ASSETS_DIR / "escrow" / "1.0.3.json"
escrow_deployer = deployer(manifest)
escrow_strategy = linker(
deploy("SafeSendLib"),
link("Escrow", "SafeSendLib"),
deploy("Escrow", w3.eth.accounts[0]),
)
escrow_deployer.register_strategy("Escrow", escrow_strategy)
escrow_package = escrow_deployer.deploy("Escrow")
return escrow_package, w3
def escrow_package(w3, deployer, ethpm_spec_dir):
manifest = ethpm_spec_dir / "examples" / "escrow" / "v3.json"
escrow_deployer = deployer(manifest)
escrow_strategy = linker(
deploy("SafeSendLib"),
link("Escrow", "SafeSendLib"),
deploy("Escrow", w3.eth.accounts[0]),
)
escrow_deployer.register_strategy("Escrow", escrow_strategy)
escrow_package = escrow_deployer.deploy("Escrow")
return escrow_package, w3
def deploy(self, contract_type: ContractName, *args: Any,
**kwargs: Any) -> Package:
factory = self.package.get_contract_factory(contract_type)
if contract_type in self.strategies:
strategy = self.strategies[contract_type]
return strategy(self.package)
if factory.needs_bytecode_linking:
raise DeployerError(
"Unable to deploy an unlinked factory. "
"Please register a strategy for this contract type.")
strategy = linker(deploy(contract_type, *args, **kwargs))
return strategy(self.package)
def test_linker(escrow_deployer, w3):
# todo test multiple links in one type
assert isinstance(escrow_deployer, Deployer)
with pytest.raises(DeployerError):
escrow_deployer.deploy("Escrow")
escrow_strategy = linker(
deploy("SafeSendLib"),
link("Escrow", "SafeSendLib"),
deploy("Escrow", w3.eth.accounts[0]),
)
assert hasattr(escrow_strategy, "__call__")
escrow_deployer.register_strategy("Escrow", escrow_strategy)
linked_escrow_package = escrow_deployer.deploy("Escrow")
assert isinstance(linked_escrow_package, Package)
linked_escrow_factory = linked_escrow_package.get_contract_factory("Escrow")
assert linked_escrow_factory.needs_bytecode_linking is False
