[chore] Fix Get Default Address on Newly Hydrated Wallet (#14)

coinbase · Sep 30, 2024 · cda596a · cda596a
1 parent 0013e00
commit cda596a
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -6,6 +6,10 @@
 
 - Contract invocation support.
 
+### Fixed
+
+- Fixed bug in `Wallet` `default_address` property for newly hydrated wallets.
+
 ## [0.0.3] - 2024-09-25
 
 ### Added

diff --git a/cdp/wallet.py b/cdp/wallet.py
@@ -606,7 +606,7 @@ def _address(self, address_id: str) -> WalletAddress | None:
 
         """
         return next(
-            (address for address in self._addresses if address.address_id == address_id),
+            (address for address in self.addresses if address.address_id == address_id),
             None,
         )
 

diff --git a/tests/test_wallet.py b/tests/test_wallet.py
@@ -48,7 +48,7 @@ def _create_wallet_model(
         network_id="base-sepolia",
         default_address=None,
         feature_set=None,
-        server_signer_status="active_seed",
+        server_signer_status="active_seed"
     ):
         if default_address is None:
             default_address = address_model_factory()
-Original file line number
+Diff line change
@@ Expand Up @@
             """
             return next(
-                (address for address in self._addresses if address.address_id == address_id),
+                (address for address in self.addresses if address.address_id == address_id),
                 None,
             )
@@ Expand Down @@