Store fee_token internally

0xSpaceShard · Jun 3, 2022 · 8db6e6c · 8db6e6c
1 parent 15c3e7e
commit 8db6e6c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/starknet_devnet/starknet_wrapper.py b/starknet_devnet/starknet_wrapper.py
@@ -75,7 +75,7 @@ async def initialize(self):
         if not self.__initialized:
             starknet = await self.__get_starknet()
 
-            await FeeToken.deploy(starknet)
+            await self.__deploy_fee_token()
             await self.__deploy_accounts()
 
             await self.__preserve_current_state(starknet.state.state)
@@ -159,6 +159,11 @@ async def __store_transaction(
 
         self.transactions.store(tx_hash, transaction)
 
+    async def __deploy_fee_token(self):
+        starknet = await self.__get_starknet()
+        await FeeToken.deploy(starknet)
+        self.contracts.store(FeeToken.ADDRESS, ContractWrapper(FeeToken.contract, FeeToken.get_definition()))
+
     async def __deploy_accounts(self):
         starknet = await self.__get_starknet()
         for account in self.accounts: