AMO Balancing Fork Tests

amo_balancing.ipynb

amo_tools_eth_results.csv

amo_tools_oeth_results.csv

brownie_fork.py

import json
import random
import hexbytes
from world import *
import pandas as pd

deploy = json.load(open("contracts/deployments/localhost/ConvexEthMetaStrategy.json"))
strat = Contract.from_abi("new_oeth_strat", OETH_CONVEX_OETH_ETH_STRAT, deploy["abi"], owner=STRATEGIST, persist=False)
pool = Contract.from_explorer("0x94b17476a93b3262d87b9a326965d1e91f9c13e7")
zapper = Contract.from_explorer(OETH_ZAPPER)





# Imbalance Pool
# Pick too much OETH
filename = "amo_tools_oeth_results.csv"
zapper.deposit({'from': BINANCE, 'value': 10000e18})
oeth.approve(pool, 1e70, {'from': BINANCE})
pool.add_liquidity([0, 10000e18], 0, {'from': BINANCE})
# Or too much ETH
filename = "amo_tools_eth_results.csv"
pool.add_liquidity([20000e18,0], 0, {'value': 20000e18, 'from': BINANCE})






## The goal is to test the new tools under a varity of randomized circumstances
ZERO_HEX = hexbytes.HexBytes('0x0000000000000000000000000000000000000000000000000000000000000000')
TRANSFER_HEX = hexbytes.HexBytes('0xddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef')
STRAT_HEX = hexbytes.HexBytes('0x0000000000000000000000001827f9ea98e0bf96550b2fc20f7233277fcd7e63')


eth_balance = pool.balances(0)
oeth_balance = pool.balances(1)
virtual_price = pool.get_virtual_price()

NUM_RUNS = 400
rows = []
for i in range(0,NUM_RUNS):
    with TemporaryFork():
        correction_factor = random.random() * 1.10 - 0.05
        print(i, correction_factor)

        diff = oeth_balance - eth_balance
        negative_change_amount = int(diff * correction_factor)


        row = {}

        row['before_pool_eth'] = pool.balances(0)
        row['before_pool_oeth'] = pool.balances(1)
        row['before_vault_oeth'] = weth.balanceOf(oeth_vault_core)
        row['before_strat_balance'] = strat.checkBalance(weth)
        o_token_change = 0

        if negative_change_amount > 0:
            try:
                tx = strat.removeAndBurnOTokens(negative_change_amount * 1e18 / virtual_price)
                for x in tx.logs:
                        if (
                            len(x.topics) == 3
                            and x.address == oeth.address
                            and x.topics[0] == TRANSFER_HEX
                            and x.topics[1] == STRAT_HEX
                            and x.topics[2] == ZERO_HEX
                            ):
                            o_token_change = int(x.data, 16) * -1
                            break
            except:
                pass
        else:
            if random.randint(0,1) == 1:
                try:
                    tx = strat.mintAndAddOTokens(0-negative_change_amount)
                    for x in tx.logs:
                        if (
                            len(x.topics) == 3
                            and x.address == oeth.address
                            and x.topics[0] == TRANSFER_HEX
                            and x.topics[1] == ZERO_HEX
                            and x.topics[2] == STRAT_HEX
                            ):
                            o_token_change = int(x.data, 16)
                            break

                except:
                    pass
            else:
                try:
                    strat.removeOnlyAssets((0-negative_change_amount) * 1e18 / virtual_price)
                except:
                    pass

        row['after_pool_eth'] = pool.balances(0)
        row['after_pool_oeth'] = pool.balances(1)
        row['after_vault_oeth'] = weth.balanceOf(oeth_vault_core)
        row['after_strat_balance'] = strat.checkBalance(weth)
        row['o_token_change'] = o_token_change

        rows.append(row)


print(pd.DataFrame.from_records(rows).to_csv(filename))