Add deployer preset

src/openzeppelin/utils/presets/UniversalDeployer.cairo

@@ -0,0 +1,72 @@
+// SPDX-License-Identifier: MIT
+// OpenZeppelin Contracts for Cairo v0.4.0 (utils/presets/UniversalDeployer.cairo)
+
+%lang starknet
+
+from starkware.starknet.common.syscalls import get_caller_address, deploy
+from starkware.cairo.common.cairo_builtins import HashBuiltin
+from starkware.cairo.common.hash import hash2
+from starkware.cairo.common.bool import FALSE, TRUE
+
+@event
+func ContractDeployed(
+    address: felt,
+    deployer: felt,
+    unique: felt,
+    classHash: felt,
+    calldata_len: felt,
+    calldata: felt*,
+    salt: felt
+) {
+}
+
+@external
+func deployContract{
+    syscall_ptr: felt*,
+    pedersen_ptr: HashBuiltin*,
+    range_check_ptr
+}(
+    classHash: felt,
+    salt: felt,
+    unique: felt,
+    calldata_len: felt,
+    calldata: felt*
+) -> (address: felt) {
+    alloc_locals;
+    let (deployer) = get_caller_address();
+
+    local _salt;
+    local from_zero;
+    if (unique == TRUE) {
+        let (unique_salt) = hash2{hash_ptr=pedersen_ptr}(deployer, salt);
+        _salt = unique_salt;
+        from_zero = FALSE;
+        tempvar _pedersen = pedersen_ptr;
+    } else {
+        _salt = salt;
+        from_zero = TRUE;
+        tempvar _pedersen = pedersen_ptr;
+    }
+
+    let pedersen_ptr = _pedersen;
+
+    let (address) = deploy(
+        class_hash=classHash,
+        contract_address_salt=_salt,
+        constructor_calldata_size=calldata_len,
+        constructor_calldata=calldata,
+        deploy_from_zero=from_zero,
+    );
+
+    ContractDeployed.emit(
+        address=address,
+        deployer=deployer,
+        unique=unique,
+        classHash=classHash,
+        calldata_len=calldata_len,
+        calldata=calldata,
+        salt=salt
+    );
+
+    return (address=address);
+}

tests/account/test_Account.py

@@ -1,14 +1,12 @@
 import pytest
 from signers import MockSigner, get_raw_invoke
 from nile.utils import TRUE, assert_revert
-from utils import get_contract_class, cached_contract, State, Account
+from utils import get_contract_class, cached_contract, State, Account, IACCOUNT_ID
 
 
 signer = MockSigner(123456789987654321)
 other = MockSigner(987654321123456789)
 
-IACCOUNT_ID = 0xa66bd575
-
 
 @pytest.fixture(scope='module')
 def contract_classes():
@@ -113,7 +111,7 @@ async def test_return_value(account_factory):
     read_info = await signer.send_transactions(account, [(initializable.contract_address, 'initialized', [])])
     call_info = await initializable.initialized().call()
     (call_result, ) = call_info.result
-    assert read_info.call_info.retdata[1] == call_result #1
+    assert read_info.call_info.retdata[1] == call_result  # 1
 
 
 @ pytest.mark.asyncio
@@ -139,7 +137,8 @@ async def test_nonce(account_factory):
 
     # higher nonce
     await assert_revert(
-        signer.send_transactions(account, [(initializable.contract_address, 'initialize', [])], nonce=current_nonce + 1),
+        signer.send_transactions(account, [(
+            initializable.contract_address, 'initialize', [])], nonce=current_nonce + 1),
         reverted_with="Invalid transaction nonce. Expected: {}, got: {}.".format(
             current_nonce, current_nonce + 1
         )
@@ -185,7 +184,8 @@ async def test_account_takeover_with_reentrant_call(account_factory):
     account, _, _, _, attacker = account_factory
 
     await assert_revert(
-        signer.send_transaction(account, attacker.contract_address, 'account_takeover', []),
+        signer.send_transaction(
+            account, attacker.contract_address, 'account_takeover', []),
         reverted_with="Account: no reentrant call"
     )
 

tests/account/test_EthAccount.py

@@ -1,14 +1,12 @@
 import pytest
 from nile.utils import assert_revert, TRUE, FALSE
-from utils import get_contract_class, cached_contract, State
+from utils import get_contract_class, cached_contract, State, IACCOUNT_ID
 from signers import MockEthSigner, get_raw_invoke
 
 private_key = b'\x01' * 32
 signer = MockEthSigner(b'\x01' * 32)
 other = MockEthSigner(b'\x02' * 32)
 
-IACCOUNT_ID = 0xa66bd575
-
 
 @pytest.fixture(scope='module')
 def contract_defs():

tests/utils.py

@@ -1,13 +1,19 @@
 """Utilities for testing Cairo contracts."""
 
-from pathlib import Path
 import os
+from pathlib import Path
+from starkware.starknet.public.abi import get_selector_from_name
+from starkware.starknet.business_logic.execution.objects import OrderedEvent
 from starkware.starknet.compiler.compile import compile_starknet_files
 from starkware.starknet.testing.starknet import StarknetContract
 from starkware.starknet.testing.starknet import Starknet
-from starkware.starknet.business_logic.execution.objects import OrderedEvent
-from starkware.starknet.public.abi import get_selector_from_name
 
+MAX_UINT256 = (2**128 - 1, 2**128 - 1)
+INVALID_UINT256 = (MAX_UINT256[0] + 1, MAX_UINT256[1])
+ZERO_ADDRESS = 0
+TRUE = 1
+FALSE = 0
+IACCOUNT_ID = 0xa66bd575
 
 _root = Path(__file__).parent.parent
 

tests/utils/test_UniversalDeployer.py

@@ -0,0 +1,102 @@
+import pytest
+from starkware.starknet.core.os.contract_address.contract_address import calculate_contract_address_from_hash
+from starkware.crypto.signature.fast_pedersen_hash import pedersen_hash
+from starkware.starknet.core.os.class_hash import compute_class_hash
+
+from signers import MockSigner
+from utils import (
+    State,
+    Account,
+    get_contract_class,
+    assert_event_emitted,
+    cached_contract,
+    IACCOUNT_ID,
+    FALSE,
+    TRUE,
+)
+
+signer = MockSigner(123456789987654321)
+
+
+@pytest.fixture(scope='module')
+def contract_classes():
+    account_cls = Account.get_class
+    deployer_cls = get_contract_class('UniversalDeployer')
+
+    return account_cls, deployer_cls
+
+
+@pytest.fixture(scope='module')
+async def deployer_init(contract_classes):
+    _, deployer_cls = contract_classes
+    starknet = await State.init()
+    account = await Account.deploy(signer.public_key)
+    deployer = await starknet.deploy(contract_class=deployer_cls)
+    return (
+        starknet.state,
+        account,
+        deployer
+    )
+
+
+@pytest.fixture
+def deployer_factory(contract_classes, deployer_init):
+    account_cls, deployer_cls = contract_classes
+    state, account, deployer = deployer_init
+    _state = state.copy()
+    _account = cached_contract(_state, account_cls, account)
+    deployer = cached_contract(_state, deployer_cls, deployer)
+
+    return _account, deployer
+
+
+@pytest.mark.asyncio
+@pytest.mark.parametrize('unique', [TRUE, FALSE])
+async def test_deployment(deployer_factory, unique):
+    account, deployer = deployer_factory
+    salt = 1234567875432  # random value
+    calldata = [signer.public_key]
+    class_hash = compute_class_hash(
+        contract_class=Account.get_class, hash_func=pedersen_hash)
+
+    # deploy contract
+    params = [class_hash, salt, unique, len(calldata), *calldata]
+    deploy_exec_info = await signer.send_transaction(account, deployer.contract_address, 'deployContract', params)
+    deployed_address = deploy_exec_info.call_info.retdata[1]
+
+    # check address
+    if unique:
+        actual_salt = pedersen_hash(account.contract_address, salt)
+        deployer_address = deployer.contract_address
+    else:
+        actual_salt = salt
+        deployer_address = 0
+
+    expected_address = calculate_contract_address_from_hash(
+        salt=actual_salt,
+        class_hash=class_hash,
+        constructor_calldata=calldata,
+        deployer_address=deployer_address
+    )
+
+    assert deployed_address == expected_address
+
+    # check deployment
+    tx_exec_info = await signer.send_transaction(account, deployed_address, 'supportsInterface', [IACCOUNT_ID])
+    is_account = tx_exec_info.call_info.retdata[1]
+    assert is_account == TRUE
+
+    assert_event_emitted(
+        deploy_exec_info,
+        from_address=deployer.contract_address,
+        name='ContractDeployed',
+        data=[
+            deployed_address,         # contractAddress
+            account.contract_address, # deployer
+            unique,                   # unique
+            class_hash,               # classHash
+            len(calldata),            # calldata_len
+            *calldata,                # calldata
+            salt,                     # salt
+        ]
+    )