From cfb6f9841321ec325c9ae4af880663a5b3aadbc2 Mon Sep 17 00:00:00 2001
From: El De-dog-lo <3859395+fubuloubu@users.noreply.github.com>
Date: Tue, 18 Jul 2023 14:55:58 -0400
Subject: [PATCH] fix: use chain ID [APE-1201] (#31)

* refactor: use ManagerAccessMixin for access to contract cache

* fix: leverage chain_id when making requests for token info
---
 ape_tokens/managers.py | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/ape_tokens/managers.py b/ape_tokens/managers.py
index 8b5d141..831b5c6 100644
--- a/ape_tokens/managers.py
+++ b/ape_tokens/managers.py
@@ -1,6 +1,6 @@
-from ape.api import Address
+from ape.contracts import ContractInstance
 from ape.types import ContractType
-from ape.utils import cached_property
+from ape.utils import ManagerAccessMixin, cached_property
 from eth_utils import to_checksum_address
 from tokenlists import TokenListManager
 
@@ -105,7 +105,7 @@
 )
 
 
-class TokenManager(dict):
+class TokenManager(ManagerAccessMixin, dict):
     @cached_property
     def _manager(self) -> TokenListManager:
         return TokenListManager()
@@ -116,11 +116,15 @@ def _Contract(self):
 
         return Contract
 
-    def __getitem__(self, symbol: str) -> Address:
+    def __getitem__(self, symbol: str) -> ContractInstance:
         try:
-            token_info = self._manager.get_token_info(symbol)
+            token_info = self._manager.get_token_info(
+                symbol, chain_id=self.network_manager.network.chain_id
+            )
 
         except ValueError as e:
             raise KeyError(f"Symbol '{symbol}' is not a known token symbol") from e
 
-        return self._Contract(to_checksum_address(token_info.address), contract_type=ERC20)
+        return self.chain_manager.contracts.instance_at(
+            to_checksum_address(token_info.address), contract_type=ERC20
+        )