[chore] Fix Get Default Address on Newly Hydrated Wallet

coinbase · Sep 29, 2024 · 6286af6 · 6286af6
1 parent 105bc8c
commit 6286af6
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,10 @@
 
 ## Unreleased
 
+### 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
@@ -569,7 +569,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
@@ -47,7 +47,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 @@