tests

Author: rohan-agarwal-coinbase

cdp/smart_wallet.py

@@ -97,6 +97,9 @@ def send_user_operation(
         """
         network = Network.from_chain_id(chain_id)
 
+        if not calls:
+            raise ValueError("Calls list cannot be empty")
+
         encoded_calls = []
         for call in calls:
             if isinstance(call, EVMAbiCallDict):

tests/factories/user_operation_factory.py

@@ -19,10 +19,11 @@ def _create_user_operation_model(
         status="pending",
     ):
         if calls is None:
-            calls = [Call(
-                to="0x1234567890123456789012345678901234567890",
-                value="1000000000000000000",
-                data="0x",
+            calls = [
+                Call(
+                    to="0x1234567890123456789012345678901234567890",
+                    value="1000000000000000000",
+                    data="0x",
                 )
             ]
         return UserOperationModel(

tests/test_smart_wallet.py

@@ -1,11 +1,12 @@
 from unittest.mock import Mock, patch
 
+import pytest
 from eth_account import Account
 
 from cdp.client.models.call import Call
 from cdp.client.models.create_smart_wallet_request import CreateSmartWalletRequest
-from cdp.evm_call_types import EVMCallDict
-from cdp.smart_wallet import SmartWallet
+from cdp.evm_call_types import EVMAbiCallDict, EVMCallDict
+from cdp.smart_wallet import SmartWallet, to_smart_wallet
 from cdp.user_operation import UserOperation
 
 
@@ -51,15 +52,14 @@ def test_smart_wallet_use_network(smart_wallet_factory):
 
 
 @patch("cdp.Cdp.api_clients")
-def test_smart_wallet_send_user_operation(
+def test_smart_wallet_send_user_operation_with_encoded_call(
     mock_api_clients, smart_wallet_factory, user_operation_model_factory, account_factory
 ):
     """Test SmartWallet send_user_operation method."""
     account = account_factory()
     smart_wallet_address = "0x1234567890123456789012345678901234567890"
     smart_wallet = smart_wallet_factory(smart_wallet_address, account)
 
-    # Create a mock user operation using UserOperationModel
     mock_user_operation = user_operation_model_factory(
         id="test-user-operation-id",
         network_id="base-sepolia",
@@ -70,7 +70,6 @@ def test_smart_wallet_send_user_operation(
         status="pending",
     )
 
-    # Setup the mock chain
     mock_create_user_operation = Mock(return_value=mock_user_operation)
     mock_api_clients.smart_wallets.create_user_operation = mock_create_user_operation
 
@@ -80,14 +79,170 @@ def test_smart_wallet_send_user_operation(
         calls=calls, chain_id=84532, paymaster_url="https://paymaster.com"
     )
 
-    # Verify the mock was called with correct arguments
     mock_create_user_operation.assert_called_once()
 
-    # Verify the user operation properties match expected values
     assert user_operation.user_operation_id == mock_user_operation.id
     assert user_operation.smart_wallet_address == smart_wallet_address
     assert user_operation.unsigned_payload == mock_user_operation.unsigned_payload
     assert user_operation.signature == mock_user_operation.signature
     assert user_operation.transaction_hash == mock_user_operation.transaction_hash
     assert user_operation.status == UserOperation.Status(mock_user_operation.status)
     assert not user_operation.terminal_state
+
+
+@patch("cdp.Cdp.api_clients")
+def test_send_user_operation_with_abi_call(
+    mock_api_clients, smart_wallet_factory, user_operation_model_factory, account_factory
+):
+    """Test sending user operation with EVMAbiCallDict."""
+    account = account_factory()
+    smart_wallet_address = "0x1234567890123456789012345678901234567890"
+    smart_wallet = smart_wallet_factory(smart_wallet_address, account)
+
+    mock_user_operation = user_operation_model_factory(
+        id="test-op-id",
+        network_id="base-sepolia",
+        calls=[Call(to=account.address, data="0x123", value="0")],
+        unsigned_payload="0x" + "0" * 64,
+        signature="0xe0a180fdd0fe38037cc878c03832861b40a29d32bd7b40b10c9e1efc8c1468a05ae06d1624896d0d29f4b31e32772ea3cb1b4d7ed4e077e5da28dcc33c0e78121c",
+        transaction_hash="0x4567890123456789012345678901234567890123",
+        status="pending",
+    )
+
+    mock_create_user_operation = Mock(return_value=mock_user_operation)
+    mock_api_clients.smart_wallets.create_user_operation = mock_create_user_operation
+
+    abi_call = EVMAbiCallDict(
+        to=account.address,
+        abi=[{"inputs": [], "name": "transfer", "type": "function"}],
+        function_name="transfer",
+        args=[],
+        value=None,
+    )
+
+    user_operation = smart_wallet.send_user_operation(calls=[abi_call], chain_id=84532)
+
+    assert user_operation.user_operation_id == mock_user_operation.id
+    assert user_operation.smart_wallet_address == smart_wallet_address
+    assert user_operation.unsigned_payload == mock_user_operation.unsigned_payload
+    assert user_operation.signature == mock_user_operation.signature
+    assert user_operation.transaction_hash == mock_user_operation.transaction_hash
+    assert user_operation.status == UserOperation.Status(mock_user_operation.status)
+    assert not user_operation.terminal_state
+    mock_create_user_operation.assert_called_once()
+
+
+def test_smart_wallet_multiple_calls(smart_wallet_factory, account_factory):
+    """Test sending multiple calls in one operation."""
+    account = account_factory()
+    smart_wallet_address = "0x1234567890123456789012345678901234567890"
+    smart_wallet = smart_wallet_factory(smart_wallet_address, account)
+
+    calls = [
+        EVMCallDict(to=account.address, value=1000000000000000000, data="0x"),
+        EVMCallDict(to=account.address, value=0, data="0x123"),
+        EVMCallDict(to=account.address, value=None, data=None),
+    ]
+
+    with patch("cdp.Cdp.api_clients") as mock_api_clients:
+        mock_user_operation = Mock(
+            id="test-op-id",
+            network_id="base-sepolia",
+            calls=[Call(to=c.to, value=str(c.value or 0), data=c.data or "0x") for c in calls],
+            unsigned_payload="0x" + "0" * 64,
+            signature="0x" + "0" * 130,
+            transaction_hash="0x" + "0" * 64,
+            status="pending",
+        )
+        mock_api_clients.smart_wallets.create_user_operation.return_value = mock_user_operation
+
+        user_operation = smart_wallet.send_user_operation(calls=calls, chain_id=84532)
+        assert user_operation.user_operation_id == "test-op-id"
+
+
+def test_network_scoped_wallet_initialization(smart_wallet_factory, account_factory):
+    """Test NetworkScopedSmartWallet initialization."""
+    account = account_factory()
+    smart_wallet_address = "0x1234567890123456789012345678901234567890"
+
+    network_wallet = smart_wallet_factory(smart_wallet_address, account).use_network(
+        84532, "https://paymaster.com"
+    )
+    assert network_wallet.chain_id == 84532
+    assert network_wallet.paymaster_url == "https://paymaster.com"
+
+    network_wallet = smart_wallet_factory(smart_wallet_address, account).use_network(84532)
+    assert network_wallet.chain_id == 84532
+    assert network_wallet.paymaster_url is None
+
+
+@patch("cdp.Cdp.api_clients")
+def test_network_scoped_wallet_send_operation(
+    mock_api_clients, smart_wallet_factory, user_operation_model_factory, account_factory
+):
+    """Test NetworkScopedSmartWallet send_user_operation method."""
+    account = account_factory()
+    smart_wallet_address = "0x1234567890123456789012345678901234567890"
+    network_wallet = smart_wallet_factory(smart_wallet_address, account).use_network(84532)
+
+    mock_user_operation = user_operation_model_factory(
+        id="test-op-id",
+        network_id="base-sepolia",
+        calls=[Call(to=account.address, value="1000000000000000000", data="0x")],
+        unsigned_payload="0x" + "0" * 64,
+        signature="0xe0a180fdd0fe38037cc878c03832861b40a29d32bd7b40b10c9e1efc8c1468a05ae06d1624896d0d29f4b31e32772ea3cb1b4d7ed4e077e5da28dcc33c0e78121c",
+        transaction_hash="0x4567890123456789012345678901234567890123",
+        status="pending",
+    )
+
+    mock_create_user_operation = Mock(return_value=mock_user_operation)
+    mock_api_clients.smart_wallets.create_user_operation = mock_create_user_operation
+
+    calls = [EVMCallDict(to=account.address, value=1000000000000000000, data="0x")]
+    user_operation = network_wallet.send_user_operation(calls=calls)
+
+    assert user_operation.user_operation_id == mock_user_operation.id
+    assert user_operation.smart_wallet_address == smart_wallet_address
+    assert user_operation.unsigned_payload == mock_user_operation.unsigned_payload
+    assert user_operation.signature == mock_user_operation.signature
+    assert user_operation.transaction_hash == mock_user_operation.transaction_hash
+    assert user_operation.status == UserOperation.Status(mock_user_operation.status)
+    assert not user_operation.terminal_state
+
+
+def test_network_scoped_wallet_string_representations(smart_wallet_factory, account_factory):
+    """Test NetworkScopedSmartWallet string representations."""
+    account = account_factory()
+    address = "0x1234567890123456789012345678901234567890"
+    network_wallet = smart_wallet_factory(address, account).use_network(84532)
+
+    assert str(network_wallet) == f"Network Scoped Smart Wallet: {address} (Chain ID: 84532)"
+    assert (
+        repr(network_wallet)
+        == f"Network Scoped Smart Wallet: (model=SmartWalletModel(address='{address}'), network=Network(chain_id=84532, paymaster_url=None))"
+    )
+
+
+def test_to_smart_wallet_creation():
+    """Test to_smart_wallet function."""
+    signer = Account.create()
+    address = "0x1234567890123456789012345678901234567890"
+
+    wallet = to_smart_wallet(address, signer)
+    assert wallet.address == address
+    assert wallet.owners == [signer]
+
+
+@patch("cdp.Cdp.api_clients")
+def test_send_user_operation_with_empty_calls(
+    mock_api_clients, smart_wallet_factory, account_factory
+):
+    """Test that sending empty calls list raises ValueError."""
+    account = account_factory()
+    smart_wallet_address = "0x1234567890123456789012345678901234567890"
+    smart_wallet = smart_wallet_factory(smart_wallet_address, account)
+
+    with pytest.raises(ValueError, match="Calls list cannot be empty"):
+        smart_wallet.send_user_operation(calls=[], chain_id=84532)
+
+    mock_api_clients.smart_wallets.create_user_operation.assert_not_called()