Fix

Author: rohan-agarwal-coinbase

tests/test_wallet.py

@@ -12,6 +12,7 @@
 from cdp.client.models.wallet import Wallet as WalletModel
 from cdp.contract_invocation import ContractInvocation
 from cdp.payload_signature import PayloadSignature
+from cdp.smart_contract import SmartContract
 from cdp.trade import Trade
 from cdp.transfer import Transfer
 from cdp.wallet import Wallet
@@ -535,3 +536,172 @@ def test_wallet_fetch(mock_api_clients, wallet_factory):
     assert isinstance(fetched_wallet, Wallet)
     assert fetched_wallet.id == "fetched-wallet-id"
     mock_get_wallet.assert_called_once_with("fetched-wallet-id")
+
+
+@patch("cdp.Cdp.use_server_signer", True)
+def test_wallet_deploy_token(wallet_factory):
+    """Test the deploy_token method of a Wallet."""
+    wallet = wallet_factory()
+    mock_default_address = Mock(spec=WalletAddress)
+    mock_smart_contract = Mock(spec=SmartContract)
+    mock_default_address.deploy_token.return_value = mock_smart_contract
+
+    with patch.object(
+        Wallet, "default_address", new_callable=PropertyMock
+    ) as mock_default_address_prop:
+        mock_default_address_prop.return_value = mock_default_address
+
+        smart_contract = wallet.deploy_token(name="TestToken", symbol="TT", total_supply="1000000")
+
+        assert isinstance(smart_contract, SmartContract)
+        mock_default_address.deploy_token.assert_called_once_with("TestToken", "TT", "1000000")
+
+
+@patch("cdp.Cdp.use_server_signer", True)
+def test_wallet_deploy_nft(wallet_factory):
+    """Test the deploy_nft method of a Wallet."""
+    wallet = wallet_factory()
+    mock_default_address = Mock(spec=WalletAddress)
+    mock_smart_contract = Mock(spec=SmartContract)
+    mock_default_address.deploy_nft.return_value = mock_smart_contract
+
+    with patch.object(
+        Wallet, "default_address", new_callable=PropertyMock
+    ) as mock_default_address_prop:
+        mock_default_address_prop.return_value = mock_default_address
+
+        smart_contract = wallet.deploy_nft(
+            name="TestNFT", symbol="TNFT", base_uri="https://example.com/nft/"
+        )
+
+        assert isinstance(smart_contract, SmartContract)
+        mock_default_address.deploy_nft.assert_called_once_with(
+            "TestNFT", "TNFT", "https://example.com/nft/"
+        )
+
+
+@patch("cdp.Cdp.use_server_signer", True)
+def test_wallet_deploy_multi_token(wallet_factory):
+    """Test the deploy_multi_token method of a Wallet."""
+    wallet = wallet_factory()
+    mock_default_address = Mock(spec=WalletAddress)
+    mock_smart_contract = Mock(spec=SmartContract)
+    mock_default_address.deploy_multi_token.return_value = mock_smart_contract
+
+    with patch.object(
+        Wallet, "default_address", new_callable=PropertyMock
+    ) as mock_default_address_prop:
+        mock_default_address_prop.return_value = mock_default_address
+
+        smart_contract = wallet.deploy_multi_token(uri="https://example.com/multi-token/{id}.json")
+
+        assert isinstance(smart_contract, SmartContract)
+        mock_default_address.deploy_multi_token.assert_called_once_with(
+            "https://example.com/multi-token/{id}.json"
+        )
+
+
+@patch("cdp.Cdp.use_server_signer", True)
+def test_wallet_deploy_token_no_default_address(wallet_factory):
+    """Test the deploy_token method of a Wallet with no default address."""
+    wallet = wallet_factory()
+
+    with patch.object(
+        Wallet, "default_address", new_callable=PropertyMock
+    ) as mock_default_address_prop:
+        mock_default_address_prop.return_value = None
+
+        with pytest.raises(ValueError, match="Default address does not exist"):
+            wallet.deploy_token(name="TestToken", symbol="TT", total_supply="1000000")
+
+
+# Add similar tests for deploy_nft and deploy_multi_token with no default address
+@patch("cdp.Cdp.use_server_signer", True)
+def test_wallet_deploy_nft_no_default_address(wallet_factory):
+    """Test the deploy_nft method of a Wallet with no default address."""
+    wallet = wallet_factory()
+
+    with patch.object(
+        Wallet, "default_address", new_callable=PropertyMock
+    ) as mock_default_address_prop:
+        mock_default_address_prop.return_value = None
+
+        with pytest.raises(ValueError, match="Default address does not exist"):
+            wallet.deploy_nft(name="TestNFT", symbol="TNFT", base_uri="https://example.com/nft/")
+
+
+@patch("cdp.Cdp.use_server_signer", True)
+def test_wallet_deploy_multi_token_no_default_address(wallet_factory):
+    """Test the deploy_multi_token method of a Wallet with no default address."""
+    wallet = wallet_factory()
+
+    with patch.object(
+        Wallet, "default_address", new_callable=PropertyMock
+    ) as mock_default_address_prop:
+        mock_default_address_prop.return_value = None
+
+        with pytest.raises(ValueError, match="Default address does not exist"):
+            wallet.deploy_multi_token(uri="https://example.com/multi-token/{id}.json")
+
+
+@patch("cdp.Cdp.use_server_signer", True)
+def test_wallet_deploy_token_with_server_signer(wallet_factory):
+    """Test the deploy_token method of a Wallet with server-signer."""
+    wallet = wallet_factory()
+    mock_default_address = Mock(spec=WalletAddress)
+    mock_smart_contract = Mock(spec=SmartContract)
+    mock_default_address.deploy_token.return_value = mock_smart_contract
+
+    with patch.object(
+        Wallet, "default_address", new_callable=PropertyMock
+    ) as mock_default_address_prop:
+        mock_default_address_prop.return_value = mock_default_address
+
+        smart_contract = wallet.deploy_token(name="TestToken", symbol="TT", total_supply="1000000")
+
+        assert isinstance(smart_contract, SmartContract)
+        mock_default_address.deploy_token.assert_called_once_with("TestToken", "TT", "1000000")
+
+
+@patch("cdp.Cdp.use_server_signer", True)
+def test_wallet_deploy_nft_with_server_signer(wallet_factory):
+    """Test the deploy_nft method of a Wallet with server-signer."""
+    wallet = wallet_factory()
+    mock_default_address = Mock(spec=WalletAddress)
+    mock_smart_contract = Mock(spec=SmartContract)
+    mock_default_address.deploy_nft.return_value = mock_smart_contract
+
+    with patch.object(
+        Wallet, "default_address", new_callable=PropertyMock
+    ) as mock_default_address_prop:
+        mock_default_address_prop.return_value = mock_default_address
+
+        smart_contract = wallet.deploy_nft(
+            name="TestNFT", symbol="TNFT", base_uri="https://example.com/nft/"
+        )
+
+        assert isinstance(smart_contract, SmartContract)
+        mock_default_address.deploy_nft.assert_called_once_with(
+            "TestNFT", "TNFT", "https://example.com/nft/"
+        )
+
+
+@patch("cdp.Cdp.use_server_signer", True)
+def test_wallet_deploy_multi_token_with_server_signer(wallet_factory):
+    """Test the deploy_multi_token method of a Wallet with server-signer."""
+    wallet = wallet_factory()
+    mock_default_address = Mock(spec=WalletAddress)
+    mock_smart_contract = Mock(spec=SmartContract)
+    mock_default_address.deploy_multi_token.return_value = mock_smart_contract
+
+    with patch.object(
+        Wallet, "default_address", new_callable=PropertyMock
+    ) as mock_default_address_prop:
+        mock_default_address_prop.return_value = mock_default_address
+
+        smart_contract = wallet.deploy_multi_token(uri="https://example.com/multi-token/{id}.json")
+
+        assert isinstance(smart_contract, SmartContract)
+        mock_default_address.deploy_multi_token.assert_called_once_with(
+            "https://example.com/multi-token/{id}.json"
+        )

tests/test_wallet_address.py

@@ -1,5 +1,5 @@
 from decimal import Decimal
-from unittest.mock import Mock, patch
+from unittest.mock import ANY, Mock, patch
 
 import pytest
 from eth_account.datastructures import SignedMessage
@@ -14,6 +14,7 @@
 from cdp.contract_invocation import ContractInvocation
 from cdp.errors import InsufficientFundsError
 from cdp.payload_signature import PayloadSignature
+from cdp.smart_contract import SmartContract
 from cdp.trade import Trade
 from cdp.transfer import Transfer
 from cdp.wallet_address import WalletAddress
@@ -629,3 +630,126 @@ def test_repr(wallet_address):
     """Test the repr representation of a WalletAddress."""
     expected_repr = "WalletAddress: (address_id: 0x1234567890123456789012345678901234567890, wallet_id: test-wallet-id-1, network_id: base-sepolia)"
     assert repr(wallet_address) == expected_repr
+
+
+@patch("cdp.Cdp.api_clients")
+@patch("cdp.Cdp.use_server_signer", True)
+def test_wallet_address_deploy_token(mock_api_clients, wallet_address):
+    """Test the deploy_token method of a WalletAddress."""
+    mock_smart_contract = Mock(spec=SmartContract)
+    mock_api_clients.smart_contracts.create_smart_contract.return_value = mock_smart_contract
+
+    smart_contract = wallet_address.deploy_token(
+        name="TestToken", symbol="TT", total_supply="1000000"
+    )
+
+    assert isinstance(smart_contract, SmartContract)
+    mock_api_clients.smart_contracts.create_smart_contract.assert_called_once_with(
+        wallet_id=wallet_address.wallet_id,
+        address_id=wallet_address.address_id,
+        create_smart_contract_request=ANY,
+    )
+
+
+@patch("cdp.Cdp.api_clients")
+@patch("cdp.Cdp.use_server_signer", True)
+def test_wallet_address_deploy_nft(mock_api_clients, wallet_address):
+    """Test the deploy_nft method of a WalletAddress."""
+    mock_smart_contract = Mock(spec=SmartContract)
+    mock_api_clients.smart_contracts.create_smart_contract.return_value = mock_smart_contract
+
+    smart_contract = wallet_address.deploy_nft(
+        name="TestNFT", symbol="TNFT", base_uri="https://example.com/nft/"
+    )
+
+    assert isinstance(smart_contract, SmartContract)
+    mock_api_clients.smart_contracts.create_smart_contract.assert_called_once_with(
+        wallet_id=wallet_address.wallet_id,
+        address_id=wallet_address.address_id,
+        create_smart_contract_request=ANY,
+    )
+
+
+@patch("cdp.Cdp.api_clients")
+@patch("cdp.Cdp.use_server_signer", True)
+def test_wallet_address_deploy_multi_token(mock_api_clients, wallet_address):
+    """Test the deploy_multi_token method of a WalletAddress."""
+    mock_smart_contract = Mock(spec=SmartContract)
+    mock_api_clients.smart_contracts.create_smart_contract.return_value = mock_smart_contract
+
+    smart_contract = wallet_address.deploy_multi_token(
+        uri="https://example.com/multi-token/{id}.json"
+    )
+
+    assert isinstance(smart_contract, SmartContract)
+    mock_api_clients.smart_contracts.create_smart_contract.assert_called_once_with(
+        wallet_id=wallet_address.wallet_id,
+        address_id=wallet_address.address_id,
+        create_smart_contract_request=ANY,
+    )
+
+
+@patch("cdp.Cdp.api_clients")
+@patch("cdp.Cdp.use_server_signer", True)
+def test_wallet_address_deploy_token_with_server_signer(mock_api_clients, wallet_address):
+    """Test the deploy_token method of a WalletAddress with server signer."""
+    mock_smart_contract = Mock(spec=SmartContract)
+    mock_api_clients.smart_contracts.create_smart_contract.return_value = mock_smart_contract
+
+    smart_contract = wallet_address.deploy_token(
+        name="TestToken", symbol="TT", total_supply="1000000"
+    )
+
+    assert isinstance(smart_contract, SmartContract)
+    mock_api_clients.smart_contracts.create_smart_contract.assert_called_once_with(
+        wallet_id=wallet_address.wallet_id,
+        address_id=wallet_address.address_id,
+        create_smart_contract_request=ANY,
+    )
+    # Verify that sign and broadcast methods are not called when using server signer
+    mock_smart_contract.sign.assert_not_called()
+    mock_smart_contract.broadcast.assert_not_called()
+
+
+@patch("cdp.Cdp.api_clients")
+@patch("cdp.Cdp.use_server_signer", True)
+def test_wallet_address_deploy_nft_with_server_signer(mock_api_clients, wallet_address):
+    """Test the deploy_nft method of a WalletAddress with server signer."""
+    mock_smart_contract = Mock(spec=SmartContract)
+    mock_api_clients.smart_contracts.create_smart_contract.return_value = mock_smart_contract
+
+    smart_contract = wallet_address.deploy_nft(
+        name="TestNFT", symbol="TNFT", base_uri="https://example.com/nft/"
+    )
+
+    assert isinstance(smart_contract, SmartContract)
+    mock_api_clients.smart_contracts.create_smart_contract.assert_called_once_with(
+        wallet_id=wallet_address.wallet_id,
+        address_id=wallet_address.address_id,
+        create_smart_contract_request=ANY,
+    )
+    # Verify that sign and broadcast methods are not called when using server signer
+    mock_smart_contract.sign.assert_not_called()
+    mock_smart_contract.broadcast.assert_not_called()
+
+
+@patch("cdp.Cdp.api_clients")
+@patch("cdp.Cdp.use_server_signer", True)
+def test_wallet_address_deploy_multi_token_with_server_signer(mock_api_clients, wallet_address):
+    """Test the deploy_multi_token method of a WalletAddress with server signer."""
+    mock_smart_contract = Mock(spec=SmartContract)
+    mock_api_clients.smart_contracts.create_smart_contract.return_value = mock_smart_contract
+
+    smart_contract = wallet_address.deploy_multi_token(
+        uri="https://example.com/multi-token/{id}.json"
+    )
+
+    assert isinstance(smart_contract, SmartContract)
+    mock_api_clients.smart_contracts.create_smart_contract.assert_called_once_with(
+        wallet_id=wallet_address.wallet_id,
+        address_id=wallet_address.address_id,
+        create_smart_contract_request=ANY,
+    )
+    # Verify that sign and broadcast methods are not called when using server signer
+    mock_smart_contract.sign.assert_not_called()
+    mock_smart_contract.broadcast.assert_not_called()