added deposit method and test;

This commit is contained in:
Lecky Lao 2020-07-01 00:07:15 +10:00
parent b3da8dcba0
commit 0ff06b28f1
2 changed files with 70 additions and 0 deletions

View File

@ -112,4 +112,48 @@ contract CurveSBTCProtocol is CurveSBTCHelpers {
emitEvent(_eventCode, _eventParam);
}
/**
* @dev Deposit Token.
* @param token token address.
* @param amt token amount.
* @param unitAmt unit amount of curve_amt/token_amt with slippage.
* @param getId Get token amount at this ID from `InstaMemory` Contract.
* @param setId Set token amount at this ID in `InstaMemory` Contract.
*/
function deposit(
address token,
uint amt,
uint unitAmt,
uint getId,
uint setId
) external payable {
uint256 _amt = getUint(getId, amt);
ERC20 tokenContract = ERC20(token);
_amt = _amt == uint(-1) ? tokenContract.balanceOf(address(this)) : _amt;
uint[3] memory _amts;
_amts[uint(getTokenI(token))] = _amt;
tokenContract.approve(getCurveSwapAddr(), _amt);
uint _amt18 = convertTo18(tokenContract.decimals(), _amt);
uint _slippageAmt = wmul(unitAmt, _amt18);
ERC20 curveTokenContract = ERC20(getCurveTokenAddr());
uint initialCurveBal = curveTokenContract.balanceOf(address(this));
ICurve(getCurveSwapAddr()).add_liquidity(_amts, _slippageAmt);
uint finalCurveBal = curveTokenContract.balanceOf(address(this));
uint mintAmt = sub(finalCurveBal, initialCurveBal);
setUint(setId, mintAmt);
emit LogDeposit(token, _amt, mintAmt, getId, setId);
bytes32 _eventCode = keccak256("LogDeposit(address,uint256,uint256,uint256,uint256)");
bytes memory _eventParam = abi.encode(token, _amt, mintAmt, getId, setId);
emitEvent(_eventCode, _eventParam);
}
}

View File

@ -10,6 +10,7 @@ const CurveSBTCProtocol = artifacts.require('CurveSBTCProtocol');
const erc20 = require("@studydefi/money-legos/erc20");
const uniswap = require("@studydefi/money-legos/uniswap");
const sbtcABI = require("./abi/sbtc.json");
const erc20ABI = require("./abi/erc20.js");
contract('CurveSBTCProtocol', async accounts => {
const [sender, receiver] = accounts;
@ -74,4 +75,29 @@ contract('CurveSBTCProtocol', async accounts => {
const sbtcAfter = await sbtcContract.methods.balanceOf(sender).call();
expect(sbtcAfter - sbtcBefore).to.be.at.least(ether("0.09"));
});
it('can add liquidity for wbtc', async function() {
const curveTokenContract = new web3.eth.Contract(
erc20ABI,
"0x075b1bb99792c9e1041ba13afef80c91a1e70fb3"
)
const tx = await contract.deposit(
erc20.wbtc.address,
10000000,
( 0.09 / 0.1 * 1e18 ).toString(),
0,
0,
{
gas: 4000000,
from: sender
}
);
console.log(tx);
const balance = await curveTokenContract.methods.balanceOf(sender);
expect(balance).to.be.at.least(ether("0.09"));
});
});