def test_emergency_exit(web3, strategy, vault, whale, chain, dai, cdai, gov,
comp):
amount1 = Wei('5000 ether')
deposit(amount1, whale, dai, vault)
amount1 = Wei('500 ether')
deposit(amount1, gov, dai, vault)
strategy.harvest({'from': gov})
wait(30, chain)
assert vault.emergencyShutdown() == False
vault.setEmergencyShutdown(True, {"from": gov})
assert vault.emergencyShutdown()
stateOfStrat(strategy, dai, comp)
stateOfVault(vault, strategy)
strategy.harvest({'from': gov})
print('\n Emergency shut down + harvest done')
stateOfStrat(strategy, dai, comp)
stateOfVault(vault, strategy)
print('\n Withdraw All')
vault.withdraw(vault.balanceOf(gov), {'from': gov})
stateOfStrat(strategy, dai, comp)
stateOfVault(vault, strategy)
def test_harvest_trigger(web3, chain, comp, vault, largerunningstrategy, whale,
gov, dai):
largerunningstrategy.setMinCompToSell(Wei('0.01 ether'), {"from": gov})
assert largerunningstrategy.harvestTrigger(Wei('1 ether')) == False
#sleep a day
chain.sleep(86401)
chain.mine(1)
assert largerunningstrategy.harvestTrigger(Wei('1 ether')) == True
largerunningstrategy.harvest({"from": gov})
assert largerunningstrategy.harvestTrigger(Wei('0.0002 ether')) == False
deposit(Wei('100 ether'), whale, dai, vault)
assert largerunningstrategy.harvestTrigger(Wei('0.0002 ether')) == True
assert largerunningstrategy.harvestTrigger(Wei('0.006 ether')) == False
largerunningstrategy.harvest({"from": gov})
times = 0
while largerunningstrategy.harvestTrigger(Wei('0.0002 ether')) == False:
wait(50, chain)
print(largerunningstrategy.predictCompAccrued().to('ether'),
' comp prediction')
times = times + 1
assert times < 10
assert times > 3
largerunningstrategy.harvest({"from": gov})
def test_live_status(web3, chain, live_vault, live_strategy, comp,dai, samdev):
stateOfStrat(live_strategy, dai, comp)
stateOfVault(live_vault, live_strategy)
genericStateOfStrat(live_strategy, dai, live_vault)
genericStateOfVault(live_vault, dai)
#withdraw(100,samdev, dai, live_vault)
genericStateOfStrat(live_strategy, dai, live_vault)
genericStateOfVault(live_vault, dai)
wait(10, chain)
print('\n Harvest')
live_strategy.harvest({'from': samdev})
genericStateOfStrat(live_strategy, dai, live_vault)
genericStateOfVault(live_vault, dai)
stateOfStrat(live_strategy, dai, comp)
stateOfVault(live_vault, live_strategy)
wait(10, chain)
print('\n Harvest')
live_strategy.harvest({'from': samdev})
genericStateOfStrat(live_strategy, dai, live_vault)
genericStateOfVault(live_vault, dai)
stateOfStrat(live_strategy, dai, comp)
stateOfVault(live_vault, live_strategy)
def test_profit_is_registered(web3, chain, comp, vault, largerunningstrategy,
whale, gov, dai):
stateOfStrat(largerunningstrategy, dai, comp)
stateOfVault(vault, largerunningstrategy)
#1m deposit
amount = Wei('1000000 ether')
deposit(amount, whale, dai, vault)
harvest(largerunningstrategy, gov)
#all money in vault
assert largerunningstrategy.estimatedTotalAssets() > amount * 0.99
stateOfStrat(largerunningstrategy, dai, comp)
stateOfVault(vault, largerunningstrategy)
sample = 500
wait(sample, chain)
harvest(largerunningstrategy, gov)
stateOfStrat(largerunningstrategy, dai, comp)
stateOfVault(vault, largerunningstrategy)
debt = vault.strategies(largerunningstrategy)[5]
returns = vault.strategies(largerunningstrategy)[6]
assert returns > 0
blocks_per_year = 2_300_000
apr = returns / debt * (blocks_per_year / sample)
print(f'implied apr: {apr:.8%}')
assert apr > 0
def test_vault_shares_generic(strategy_changeable, web3, chain, Vault,
currency, whale, strategist):
gov = strategist
vault = strategist.deploy(Vault, currency, strategist, strategist,
"TestVault", "Amount")
deposit_limit = Wei('1000 ether')
#set limit to the vault
vault.setDepositLimit(deposit_limit, {"from": strategist})
#deploy strategy
strategy = strategist.deploy(strategy_changeable, vault)
vault.addStrategy(strategy, deposit_limit, deposit_limit, 50,
{"from": strategist})
amount1 = Wei('50 ether')
deposit(amount1, whale, currency, vault)
whale_share = vault.balanceOf(whale)
deposit(amount1, gov, currency, vault)
gov_share = vault.balanceOf(gov)
assert gov_share == whale_share
assert vault.pricePerShare() == 1e18
assert vault.pricePerShare() * whale_share / 1e18 == amount1
assert vault.pricePerShare() * whale_share / 1e18 == vault.totalAssets(
) / 2
assert gov_share == whale_share
strategy.harvest({'from': gov})
#no profit yet
whale_share = vault.balanceOf(whale)
gov_share = vault.balanceOf(gov)
assert gov_share > whale_share
wait(100, chain)
whale_share = vault.balanceOf(whale)
gov_share = vault.balanceOf(gov)
# no profit just aum fee. meaning total balance should be the same
assert (gov_share +
whale_share) * vault.pricePerShare() / 1e18 == 100 * 1e18
strategy.harvest({'from': gov})
whale_share = vault.balanceOf(whale)
gov_share = vault.balanceOf(gov)
#add strategy return
assert vault.totalSupply() == whale_share + gov_share
value = (gov_share + whale_share) * vault.pricePerShare() / 1e18
assert value == 100 * 1e18 + vault.strategies(strategy)[6]
#check we are within 0.1% of expected returns
assert value < strategy.estimatedTotalAssets(
) * 1.001 and value > strategy.estimatedTotalAssets() * 0.999
assert gov_share > whale_share
def test_live_apr_dai(live_vault_dai_030, live_strategy_dai_030, Contract,
whale, web3, live_gov, accounts, chain, cdai, comp, usdc,
currency, samdev):
strategist = samdev
strategy = live_strategy_dai_030
vault = live_vault_dai_030
strategy.setMinCompToSell(0, {'from': strategist})
strategy.harvest({'from': strategist})
startingBalance = vault.totalAssets()
for i in range(6):
waitBlock = 25
print(f'\n----wait {waitBlock} blocks----')
wait(waitBlock, chain)
ppsBefore = vault.pricePerShare()
print(ppsBefore / 1e18)
cdai.mint(0, {'from': strategist})
strategy.harvest({'from': strategist})
#stateOfStrat(strategy, usdc, comp)
#genericStateOfVault(vault, usdc)
ppsAfter = vault.pricePerShare()
print(ppsAfter / 1e6)
profit = (vault.totalAssets() - startingBalance) / 1e18
strState = vault.strategies(strategy)
totalReturns = strState[6]
totaleth = totalReturns / 1e18
print(f'Real Profit: {profit:.5f}')
difff = profit - totaleth
# print(f'Diff: {difff}')
blocks_per_year = 2_300_000
assert startingBalance != 0
time = (i + 1) * waitBlock
assert time != 0
ppsProfit = (ppsAfter -
ppsBefore) / ppsBefore / waitBlock * blocks_per_year
apr = (profit / (startingBalance / 1e18)) * (blocks_per_year / time)
print(f'implied apr assets: {apr:.8%}')
print(f'implied apr pps: {ppsProfit:.8%}')
def test_apr_dai(web3, chain, comp, vault, enormousrunningstrategy, whale, gov,
dai, strategist):
enormousrunningstrategy.setProfitFactor(1, {"from": strategist})
assert enormousrunningstrategy.profitFactor() == 1
enormousrunningstrategy.setMinCompToSell(1, {"from": gov})
enormousrunningstrategy.setMinWant(0, {"from": gov})
assert enormousrunningstrategy.minCompToSell() == 1
startingBalance = vault.totalAssets()
stateOfStrat(enormousrunningstrategy, dai, comp)
stateOfVault(vault, enormousrunningstrategy)
for i in range(6):
waitBlock = 25
print(f"\n----wait {waitBlock} blocks----")
chain.sleep(21600)
wait(waitBlock, chain)
ppsBefore = vault.pricePerShare()
harvest(enormousrunningstrategy, strategist, vault)
ppsAfter = vault.pricePerShare()
# stateOfStrat(enormousrunningstrategy, dai, comp)
# stateOfVault(vault, enormousrunningstrategy)
profit = (vault.totalAssets() - startingBalance).to("ether")
strState = vault.strategies(enormousrunningstrategy)
totalReturns = strState[6]
totaleth = totalReturns.to("ether")
print(f"Real Profit: {profit:.5f}")
difff = profit - totaleth
print(f"Diff: {difff}")
blocks_per_year = 2_300_000
assert startingBalance != 0
time = (i + 1) * waitBlock
assert time != 0
ppsProfit = (ppsAfter - ppsBefore) / 1e18 / waitBlock * blocks_per_year
apr = (totalReturns / startingBalance) * (blocks_per_year / time)
print(f"implied apr assets: {apr:.8%}")
print(f"implied apr pps: {ppsProfit:.8%}")
vault.withdraw(vault.balanceOf(whale), {"from": whale})
def test_comp_dis(web3, chain, comp, vault, largerunningstrategy, dai, gov):
#stateOfStrat(largerunningstrategy, dai)
# stateOfVault(vault, largerunningstrategy)
wait(100, chain)
# Claim COMP and check balance in strategy -> for testing _claimComp() needs to be public
print('\n----checking comp----')
balanceBefore = comp.balanceOf(largerunningstrategy)
comp_prediction = largerunningstrategy.predictCompAccrued()
print(comp_prediction.to('ether'), 'comp accrued')
largerunningstrategy.claimComp({'from': gov})
comp_balance = comp.balanceOf(largerunningstrategy) - balanceBefore
print(comp_balance.to('ether'), 'comp claimed')
assert comp_balance < comp_prediction * 1.1 and comp_balance > comp_prediction * 0.9
def test_good_migration(strategy_changeable, web3, chain, Vault, currency,
whale, rando, strategist):
# Call this once to seed the strategy with debt
gov = strategist
vault = strategist.deploy(Vault, currency, strategist, strategist,
"TestVault", "Amount")
deposit_limit = Wei('1000000 ether')
#set limit to the vault
vault.setDepositLimit(deposit_limit, {"from": strategist})
#deploy strategy
strategy = strategist.deploy(strategy_changeable, vault)
vault.addStrategy(strategy, deposit_limit, deposit_limit, 50,
{"from": strategist})
amount1 = Wei('500 ether')
deposit(amount1, whale, currency, vault)
amount1 = Wei('50 ether')
deposit(amount1, gov, currency, vault)
strategy.harvest({'from': gov})
wait(30, chain)
strategy.harvest({'from': gov})
strategy_debt = vault.strategies(strategy)[4] # totalDebt
prior_position = strategy.estimatedTotalAssets()
assert strategy_debt > 0
new_strategy = strategist.deploy(strategy_changeable, vault)
assert vault.strategies(new_strategy)[4] == 0
assert currency.balanceOf(new_strategy) == 0
# Only Governance can migrate
with brownie.reverts():
vault.migrateStrategy(strategy, new_strategy, {"from": rando})
vault.migrateStrategy(strategy, new_strategy, {"from": gov})
assert vault.strategies(strategy)[4] == 0
assert vault.strategies(new_strategy)[4] == strategy_debt
assert new_strategy.estimatedTotalAssets(
) > prior_position * 0.999 or new_strategy.estimatedTotalAssets(
) < prior_position * 1.001
def test_emergency_exit_generic(strategy_changeable, web3, chain, Vault,
currency, whale, strategist):
gov = strategist
vault = strategist.deploy(Vault, currency, strategist, strategist,
"TestVault", "Amount")
deposit_limit = Wei('1000000 ether')
#set limit to the vault
vault.setDepositLimit(deposit_limit, {"from": strategist})
#deploy strategy
strategy = strategist.deploy(strategy_changeable, vault)
vault.addStrategy(strategy, deposit_limit, deposit_limit, 50,
{"from": strategist})
amount1 = Wei('500 ether')
deposit(amount1, whale, currency, vault)
amount1 = Wei('50 ether')
deposit(amount1, gov, currency, vault)
strategy.harvest({'from': gov})
wait(30, chain)
assert vault.emergencyShutdown() == False
vault.setEmergencyShutdown(True, {"from": gov})
assert vault.emergencyShutdown()
genericStateOfStrat(strategy, currency, vault)
genericStateOfVault(vault, currency)
strategy.harvest({'from': gov})
assert currency.balanceOf(strategy) == 0
strategy.harvest({'from': gov})
assert currency.balanceOf(strategy) == strategy.estimatedTotalAssets()
print('\n Emergency shut down + harvest done')
genericStateOfStrat(strategy, currency, vault)
genericStateOfVault(vault, currency)
print('\n Withdraw All')
vault.withdraw(vault.balanceOf(gov), {'from': gov})
genericStateOfStrat(strategy, currency, vault)
genericStateOfVault(vault, currency)
def test_getting_too_close_to_liq(web3, chain, cdai, comp, vault,
largerunningstrategy, whale, gov, dai):
stateOfStrat(largerunningstrategy, dai, comp)
stateOfVault(vault, largerunningstrategy)
largerunningstrategy.setCollateralTarget(Wei('0.7498 ether'),
{"from": gov})
deposit(Wei('1000 ether'), whale, dai, vault)
balanceBefore = vault.totalAssets()
collat = 0
assert largerunningstrategy.tendTrigger(1e18) == False
largerunningstrategy.harvest({'from': gov})
deposits, borrows = largerunningstrategy.getCurrentPosition()
collat = borrows / deposits
print(collat)
stateOfStrat(largerunningstrategy, dai, comp)
stateOfVault(vault, largerunningstrategy)
assertCollateralRatio(largerunningstrategy)
lastCol = collat
while largerunningstrategy.tendTrigger(1e18) == False:
cdai.mint(0, {"from": gov})
waitBlock = 100
wait(waitBlock, chain)
deposits, borrows = largerunningstrategy.getCurrentPosition()
collat = borrows / deposits
assert collat > lastCol
lastCol = collat
print("Collat ratio: ", collat)
print("Blocks to liq: ",
largerunningstrategy.getblocksUntilLiquidation())
largerunningstrategy.tend({'from': gov})
largerunningstrategy.setCollateralTarget(Wei('0.73 ether'), {"from": gov})
assert largerunningstrategy.tendTrigger(1e18) == False
largerunningstrategy.tend({'from': gov})
assertCollateralRatio(largerunningstrategy)
stateOfStrat(largerunningstrategy, dai, comp)
stateOfVault(vault, largerunningstrategy)
def test_apr_creth(web3, chain, cream, vault, enormousrunningstrategy, whale,
gov, currency, strategist):
enormousrunningstrategy.setProfitFactor(1, {"from": strategist})
assert (enormousrunningstrategy.profitFactor() == 1)
enormousrunningstrategy.setMinCompToSell(1, {"from": gov})
enormousrunningstrategy.setMinWant(0, {"from": gov})
assert enormousrunningstrategy.minCompToSell() == 1
startingBalance = vault.totalAssets()
stateOfStrat(enormousrunningstrategy, currency, cream)
stateOfVault(vault, enormousrunningstrategy)
for i in range(10):
waitBlock = 100
print(f'\n----wait {waitBlock} blocks----')
wait(waitBlock, chain)
harvest(enormousrunningstrategy, strategist, vault)
stateOfStrat(enormousrunningstrategy, currency, cream)
stateOfVault(vault, enormousrunningstrategy)
#print(vault.creditAvailable(enormousrunningstrategy)/1e18)
profit = (vault.totalAssets() - startingBalance).to('ether')
strState = vault.strategies(enormousrunningstrategy)
totalReturns = strState[6]
totaleth = totalReturns.to('ether')
print(f'Real Profit: {profit:.5f}')
difff = profit - totaleth
print(f'Diff: {difff}')
blocks_per_year = 2_300_000
assert startingBalance != 0
time = (i + 1) * waitBlock
assert time != 0
apr = (totalReturns / startingBalance) * (blocks_per_year / time)
print(f'implied apr: {apr:.8%}')
vault.withdraw(vault.balanceOf(whale), {'from': whale})
stateOfStrat(enormousrunningstrategy, currency, cream)
stateOfVault(vault, enormousrunningstrategy)
def test_withdraw_all(web3, chain, comp, vault, largerunningstrategy, whale,
gov, dai, strategist, isolation):
balance_before = dai.balanceOf(strategist)
stateOfStrat(largerunningstrategy, dai, comp)
stateOfVault(vault, largerunningstrategy)
amount = Wei('10000 ether')
deposit(amount, strategist, dai, vault)
harvest(largerunningstrategy, gov, vault)
wait(50, chain)
harvest(largerunningstrategy, gov, vault)
stateOfStrat(largerunningstrategy, dai, comp)
stateOfVault(vault, largerunningstrategy)
withdraw(1, strategist, dai, vault)
profitW = dai.balanceOf(strategist) - balance_before
profit = profitW.to('ether')
print(f'profit: {profit:.5f}')
def test_good_migration(usdc, chain, whale, gov, strategist, rando, Vault,
LenderYieldOptimiser, GenericCompound, GenericCream,
GenericDyDx, interface):
currency = usdc
solo = interface.ISoloMargin('0x1E0447b19BB6EcFdAe1e4AE1694b0C3659614e4e')
cUsdc = interface.CErc20I('0x39AA39c021dfbaE8faC545936693aC917d5E7563')
crUsdc = interface.CErc20I('0x44fbeBd2F576670a6C33f6Fc0B00aA8c5753b322')
vault = strategist.deploy(Vault, usdc, strategist, strategist, "", "")
usdc.approve(vault, 2**256 - 1, {"from": whale})
usdc.approve(vault, 2**256 - 1, {"from": strategist})
#deploy strategy
strategy = strategist.deploy(LenderYieldOptimiser, vault)
#cDai = interface.CErc20I('0x5d3a536e4d6dbd6114cc1ead35777bab948e3643')
solo = interface.ISoloMargin('0x1E0447b19BB6EcFdAe1e4AE1694b0C3659614e4e')
cUsdc = interface.CErc20I('0x39AA39c021dfbaE8faC545936693aC917d5E7563')
assert cUsdc.underlying() == vault.token()
crUsdc = interface.CErc20I('0x44fbeBd2F576670a6C33f6Fc0B00aA8c5753b322')
assert crUsdc.underlying() == vault.token()
compoundPlugin = strategist.deploy(GenericCompound, strategy, "Compound",
cUsdc)
creamPlugin = strategist.deploy(GenericCream, strategy, "Cream", crUsdc)
dydxPlugin = strategist.deploy(GenericDyDx, strategy, "DyDx")
strategy.addLender(compoundPlugin, {"from": strategist})
assert strategy.numLenders() == 1
strategy.addLender(creamPlugin, {"from": strategist})
assert strategy.numLenders() == 2
strategy.addLender(dydxPlugin, {"from": strategist})
assert strategy.numLenders() == 3
deposit_limit = 1_000_000_000 * 1e6
vault.addStrategy(strategy, deposit_limit, deposit_limit, 500,
{"from": strategist})
amount1 = 500 * 1e6
vault.deposit(amount1, {"from": whale})
amount1 = 50 * 1e6
vault.deposit(amount1, {"from": strategist})
gov = strategist
strategy.harvest({'from': gov})
wait(30, chain)
strategy.harvest({'from': gov})
strategy_debt = vault.strategies(strategy)[4] # totalDebt
prior_position = strategy.estimatedTotalAssets()
assert strategy_debt > 0
new_strategy = strategist.deploy(LenderYieldOptimiser, vault)
assert vault.strategies(new_strategy)[4] == 0
assert currency.balanceOf(new_strategy) == 0
# Only Governance can migrate
with brownie.reverts():
vault.migrateStrategy(strategy, new_strategy, {"from": rando})
vault.migrateStrategy(strategy, new_strategy, {"from": gov})
assert vault.strategies(strategy)[4] == 0
assert vault.strategies(new_strategy)[4] == strategy_debt
assert new_strategy.estimatedTotalAssets(
) > prior_position * 0.999 or new_strategy.estimatedTotalAssets(
) < prior_position * 1.001
def test_strat_sam(accounts, interface, web3, chain, vault, comp, whale,
YearnDaiCompStratV2, dai, gov):
strategist_and_keeper = accounts[1]
print(strategist_and_keeper)
assert vault.governance() == gov
assert vault.guardian() == gov
assert vault.rewards() == gov
assert vault.token() == dai
# Deploy the Strategy
strategy = strategist_and_keeper.deploy(YearnDaiCompStratV2, vault)
# Addresses
assert strategy.strategist() == strategist_and_keeper
assert strategy.keeper() == strategist_and_keeper
assert strategy.want() == vault.token()
stateOfStrat(strategy, dai, comp)
# Add strategy to the Vault
assert vault.strategies(strategy) == [0, 0, 0, 0, 0, 0, 0]
_debtLimit = Wei('1000000 ether')
_rateLimit = Wei('1000000 ether')
vault.addStrategy(strategy, _debtLimit, _rateLimit, 50, {"from": gov})
assert vault.strategies(strategy) == [
50,
web3.eth.blockNumber,
_debtLimit,
_rateLimit,
web3.eth.blockNumber,
0,
0,
]
print(strategy._predictCompAccrued(), ' comp prediction')
# Nothing was reported yet from the strategy
assert vault.expectedReturn(strategy) == 0
depositLimit = Wei('500000 ether')
vault.setDepositLimit(depositLimit, {"from": gov})
assert vault.depositLimit() == depositLimit
#set compToSell
# Provide funds to the Vault from whale
# Test first with simply 5k as it is the current rate DAI/block
amount = Wei('100000 ether')
deposit(amount, whale, dai, vault)
stateOfStrat(strategy, dai, comp)
stateOfVault(vault, strategy)
# Call harvest in Strategy only when harvestTrigger() --> (true)
harvest(strategy, strategist_and_keeper)
# assert( !strategy.harvestTrigger(0, {'from': strategist_and_keeper}))
stateOfStrat(strategy, dai, comp)
stateOfVault(vault, strategy)
# now lets see 90k
amount = Wei('320000 ether')
deposit(amount, whale, dai, vault)
stateOfStrat(strategy, dai, comp)
stateOfVault(vault, strategy)
harvest(strategy, strategist_and_keeper)
stateOfStrat(strategy, dai, comp)
stateOfVault(vault, strategy)
# Claim COMP and check balance in strategy -> for testing _claimComp() needs to be public
#strategy._claimComp({'from': strategist_and_keeper})
#comp_balance = comp.balanceOf(strategy)
#print(comp_balance.to('ether'), 'comp claimed')
# Call harvest in Strategy only when harvestTrigger() --> (true)
#harvest(strategy, strategist_and_keeper)
stateOfStrat(strategy, dai, comp)
stateOfVault(vault, strategy)
#withdraw(1, strategy, whale, dai, vault)
#vault.setEmergencyShutdown(True, {"from": gov})
while strategy.harvestTrigger(0) == False:
wait(25, chain)
print(strategy._predictCompAccrued().to('ether'), ' comp prediction')
#print('Emergency Exit')
#strategy.setEmergencyExit({"from": gov})
stateOfStrat(strategy, dai, comp)
stateOfVault(vault, strategy)
harvest(strategy, strategist_and_keeper)
stateOfStrat(strategy, dai, comp)
stateOfVault(vault, strategy)
##test migration
print('test migration')
strategy2 = strategist_and_keeper.deploy(YearnDaiCompStratV2, vault)
vault.migrateStrategy(strategy, strategy2, {"from": gov})
print('old strat')
stateOfStrat(strategy, dai, comp)
stateOfVault(vault, strategy)
print('new strat')
stateOfStrat(strategy2, dai, comp)
stateOfVault(vault, strategy2)
withdraw(1, whale, dai, vault)
stateOfStrat(strategy2, dai, comp)
stateOfVault(vault, strategy2)
amount = Wei('320000 ether')
deposit(amount, whale, dai, vault)
stateOfStrat(strategy2, dai, comp)
stateOfVault(vault, strategy2)
harvest(strategy2, strategist_and_keeper)
stateOfStrat(strategy2, dai, comp)
stateOfVault(vault, strategy2)
withdraw(1, whale, dai, vault)
stateOfStrat(strategy2, dai, comp)
stateOfVault(vault, strategy2)
def test_strat_sam(accounts, interface, web3, chain, Vault,
YearnDaiCompStratV2):
gov = accounts[0]
print(gov)
strategist_and_keeper = accounts[1]
print(strategist_and_keeper)
dai = interface.ERC20('0x6b175474e89094c44da98b954eedeac495271d0f')
cdai = interface.ERC20('0x5d3a536E4D6DbD6114cc1Ead35777bAB948E3643')
comp = interface.ERC20('0xc00e94Cb662C3520282E6f5717214004A7f26888')
comptroller = interface.ComptrollerI(
'0x3d9819210A31b4961b30EF54bE2aeD79B9c9Cd3B')
# Current COMP speeds
print('COMP speed: ', comptroller.compSpeeds(cdai))
ydai = interface.ERC20('0x16de59092dae5ccf4a1e6439d611fd0653f0bd01')
whale = accounts.at("0xBE0eB53F46cd790Cd13851d5EFf43D12404d33E8",
force=True)
# Deploy the Vault
vault = gov.deploy(Vault, dai, gov, gov, "Yearn DAI v2", "y2DAI")
assert vault.governance() == gov
assert vault.guardian() == gov
assert vault.rewards() == gov
assert vault.token() == dai
# Deploy the Strategy
strategy = strategist_and_keeper.deploy(YearnDaiCompStratV2, vault)
# Addresses
assert strategy.strategist() == strategist_and_keeper
assert strategy.keeper() == strategist_and_keeper
assert strategy.want() == vault.token()
stateOfStrat(strategy, dai)
# Test current price for COMP and DAI using Chainlink
lastExchangeRate = strategy.getLatestExchangeRate()
print('Current exchange rate (COMP/DAI): ', lastExchangeRate)
# Add strategy to the Vault
assert vault.strategies(strategy) == [0, 0, 0, 0, 0, 0, 0]
_debtLimit = Wei('1000000 ether')
_rateLimit = Wei('1000000 ether')
vault.addStrategy(strategy, _debtLimit, _rateLimit, 50, {"from": gov})
assert vault.strategies(strategy) == [
50,
web3.eth.blockNumber,
_debtLimit,
_rateLimit,
web3.eth.blockNumber,
0,
0,
]
print(strategy._predictCompAccrued(), ' comp prediction')
# Nothing was reported yet from the strategy
assert vault.expectedReturn(strategy) == 0
stateOfStrat(strategy, dai)
depositLimit = Wei('500000 ether')
vault.setDepositLimit(depositLimit, {"from": gov})
assert vault.depositLimit() == depositLimit
# Provide funds to the Vault from whale
# Test first with simply 5k as it is the current rate DAI/block
amount = Wei('100000 ether')
deposit(amount, whale, dai, vault)
stateOfStrat(strategy, dai)
stateOfVault(vault, strategy)
# Call harvest in Strategy only when harvestTrigger() --> (true)
harvest(strategy, strategist_and_keeper)
# assert( !strategy.harvestTrigger(0, {'from': strategist_and_keeper}))
stateOfStrat(strategy, dai)
stateOfVault(vault, strategy)
# now lets see 90k
amount = Wei('320000 ether')
deposit(amount, whale, dai, vault)
stateOfStrat(strategy, dai)
stateOfVault(vault, strategy)
harvest(strategy, strategist_and_keeper)
stateOfStrat(strategy, dai)
stateOfVault(vault, strategy)
# Call harvest in Strategy only when harvestTrigger() --> (true)
#harvest(strategy, strategist_and_keeper)
stateOfStrat(strategy, dai)
stateOfVault(vault, strategy)
#withdraw(1, strategy, whale, dai, vault)
#vault.setEmergencyShutdown(True, {"from": gov})
while strategy.harvestTrigger(0) == False:
wait(25, chain)
print(strategy._predictCompAccrued().to('ether'), ' comp prediction')
#print('Emergency Exit')
#strategy.setEmergencyExit({"from": gov})
# Claim COMP and check balance in strategy -> for testing _claimComp() needs to be public
strategy._claimComp({'from': strategist_and_keeper})
comp_balance = comp.balanceOf(strategy)
print(comp_balance.to('ether'), 'comp claimed')
stateOfStrat(strategy, dai)
stateOfVault(vault, strategy)
harvest(strategy, strategist_and_keeper)
stateOfStrat(strategy, dai)
stateOfVault(vault, strategy)
##test migration
print('test migration')
strategy2 = strategist_and_keeper.deploy(YearnDaiCompStratV2, vault)
vault.migrateStrategy(strategy, strategy2, {"from": gov})
print('old strat')
stateOfStrat(strategy, dai)
stateOfVault(vault, strategy)
print('new strat')
stateOfStrat(strategy2, dai)
stateOfVault(vault, strategy2)
withdraw(1, whale, dai, vault)
stateOfStrat(strategy2, dai)
stateOfVault(vault, strategy2)
amount = Wei('320000 ether')
deposit(amount, whale, dai, vault)
stateOfStrat(strategy2, dai)
stateOfVault(vault, strategy2)
harvest(strategy2, strategist_and_keeper)
stateOfStrat(strategy2, dai)
stateOfVault(vault, strategy2)
withdraw(1, whale, dai, vault)
stateOfStrat(strategy2, dai)
stateOfVault(vault, strategy2)
