check balances before and after

proxy/testing/test_neon_token.py

@@ -4,6 +4,11 @@
 import os
 from .testing_helpers import request_airdrop
 from solana.account import Account as SolanaAccount
+from solana.rpc.api import Client as SolanaClient
+from spl.token.client import Token as SplToken
+from proxy.environment import ETH_TOKEN_MINT_ID
+from spl.token.constants import TOKEN_PROGRAM_ID
+from solana.rpc.commitment import Confirmed
 
 NEON_TOKEN_CONTRACT = '''
 // SPDX-License-Identifier: MIT
@@ -21,25 +26,31 @@
 
 
 proxy_url = os.environ.get('PROXY_URL', 'http://127.0.0.1:9090/solana')
+solana_url = os.environ.get('SOLANA_URL', 'http://solana:8899/')
 proxy = Web3(Web3.HTTPProvider(proxy_url))
+solana = SolanaClient(solana_url)
 SEED = 'TestNeonToken'
 eth_account = proxy.eth.account.create(SEED)
 proxy.eth.default_account = eth_account.address
 
 class TestNeonToken(unittest.TestCase):
     @classmethod
     def setUpClass(cls) -> None:
-        artifacts = compile_source(NEON_TOKEN_CONTRACT)
-        _, cls.neon_token = artifacts.popitem()
+        cls.sol_payer = SolanaAccount()
         cls.deploy_contract(cls)
+        cls.spl_neon_token = SplToken(solana, ETH_TOKEN_MINT_ID, TOKEN_PROGRAM_ID, cls.sol_payer)
         print(f"default eth account: {eth_account.address}")
         request_airdrop(eth_account.address)
 
     def deploy_contract(self):
-        erc20 = proxy.eth.contract(abi=self.neon_token['abi'], bytecode=self.neon_token['bin'])
+        artifacts = compile_source(NEON_TOKEN_CONTRACT)
+        _, self.neon_token_iface = artifacts.popitem()
+
+        self.neon_contract = proxy.eth.contract(abi=self.neon_token_iface['abi'], 
+                                                bytecode=self.neon_token_iface['bin'])
         nonce = proxy.eth.get_transaction_count(eth_account.address)
         tx = {'nonce': nonce}
-        tx_constructor = erc20.constructor().buildTransaction(tx)
+        tx_constructor = self.neon_contract.constructor().buildTransaction(tx)
         tx_deploy = proxy.eth.account.sign_transaction(tx_constructor, eth_account.key)
         tx_deploy_hash = proxy.eth.send_raw_transaction(tx_deploy.rawTransaction)
         print(f'tx_deploy_hash: {tx_deploy_hash.hex()}')
@@ -48,13 +59,17 @@ def deploy_contract(self):
         print(f'deploy status: {tx_deploy_receipt.status}')
         self.neon_token_address = tx_deploy_receipt.contractAddress
         print(f'NeonToken contract address is: {self.neon_token_address}')
-        self.neon_contract = proxy.eth.contract(address=self.neon_token_address, abi=self.neon_token['abi'])
+        self.neon_contract = proxy.eth.contract(address=self.neon_token_address, 
+                                                abi=self.neon_token_iface['abi'])
 
     def test_success_call_withdraw(self):
         dest_acc = SolanaAccount()
         print(f"Try to withdraw NEON tokens to solana account {dest_acc.public_key()}")
         amount = pow(10, 18) # 1 NEON
 
+        print(f'Source account balance before: {proxy.eth.get_balance(eth_account.address)}')
+        print(f'Destination account balance before: {self.spl_neon_token.get_balance(dest_acc.public_key(), commitment=Confirmed)}')
+
         nonce = proxy.eth.get_transaction_count(eth_account.address)
         tx = {'value': amount, 'nonce': nonce}
         withdraw_tx_dict = self.neon_contract.functions.withdraw(bytes(dest_acc.public_key())).buildTransaction(tx)
@@ -63,4 +78,7 @@ def test_success_call_withdraw(self):
         print(f'withdraw_tx_hash: {withdraw_tx_hash.hex()}')
         withdraw_tx_receipt = proxy.eth.wait_for_transaction_receipt(withdraw_tx_hash)
         print(f'withdraw_tx_receipt: {withdraw_tx_receipt}')
-        print(f'deploy status: {withdraw_tx_receipt.status}')
+        print(f'deploy status: {withdraw_tx_receipt.status}')
+
+        print(f'Source account balance after: {proxy.eth.get_balance(eth_account.address)}')
+        print(f'Destination account balance after: {self.spl_neon_token.get_balance(dest_acc.public_key(), commitment=Confirmed)}')