Added tests to wxdai deposit/withdraw random account (#460)

tests_integration_with_local_chain/markets/omen/test_omen.py

@@ -14,6 +14,7 @@
     HexStr,
     OutcomeStr,
     Wei,
+    private_key_type,
     xDai,
     xdai_type,
 )
@@ -59,6 +60,7 @@
 from prediction_market_agent_tooling.tools.hexbytes_custom import HexBytes
 from prediction_market_agent_tooling.tools.utils import utcnow
 from prediction_market_agent_tooling.tools.web3_utils import wei_to_xdai, xdai_to_wei
+from tests_integration_with_local_chain.conftest import create_and_fund_random_account
 
 DEFAULT_REASON = "Test logic need to be rewritten for usage of local chain, see ToDos"
 
@@ -355,6 +357,35 @@ def get_market_outcome_tokens() -> TokenAmount:
         assert tx["from"] == api_keys.bet_from_address
 
 
+def test_deposit_and_withdraw_wxdai(local_web3: Web3, test_keys: APIKeys) -> None:
+    deposit_amount = xDai(10)
+    fresh_account = create_and_fund_random_account(
+        private_key=test_keys.bet_from_private_key,
+        web3=local_web3,
+        deposit_amount=xDai(deposit_amount * 2),  # 2* for safety
+    )
+
+    api_keys = APIKeys(
+        BET_FROM_PRIVATE_KEY=private_key_type(fresh_account.key.hex()),
+        SAFE_ADDRESS=None,
+    )
+    wxdai = WrappedxDaiContract()
+    wxdai.deposit(
+        api_keys=api_keys, amount_wei=xdai_to_wei(deposit_amount), web3=local_web3
+    )
+    balance = get_balances(address=fresh_account.address, web3=local_web3)
+    assert balance.wxdai == deposit_amount
+
+    wxdai.withdraw(
+        api_keys=api_keys,
+        amount_wei=xdai_to_wei(balance.wxdai),
+        web3=local_web3,
+    )
+
+    balance = get_balances(address=fresh_account.address, web3=local_web3)
+    assert balance.wxdai == xDai(0)
+
+
 @pytest.mark.parametrize(
     "collateral_token_address, expected_symbol",
     [