dsa-connectors-old/test/CurveSBTCProtocol.js

const {
  BN,           // Big Number support
  expectEvent,  // Assertions for emitted events
  expectRevert, // Assertions for transactions that should fail
  balance,
  ether
} = require('@openzeppelin/test-helpers');

const ConnectSBTCCurve = artifacts.require('ConnectSBTCCurve');
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");
const connectorsABI = require("./abi/connectors.json");
const accountABI = require("./abi/account.json");

contract('ConnectSBTCCurve', async accounts => {
  const [sender, receiver] =  accounts;
  let masterAddress = "0xfcd22438ad6ed564a1c26151df73f6b33b817b56";
  let accountID = 7;
  let dsrAddr = "0xEEB007bea2Bbb0cA6502217E8867f8f7b021B8D5";

  let connectorsAddr = "0xD6A602C01a023B98Ecfb29Df02FBA380d3B21E0c";
  let connectorInstance = new web3.eth.Contract(connectorsABI, connectorsAddr);

  // let accountAddr = "0x939Daad09fC4A9B8f8A9352A485DAb2df4F4B3F8";
  let accountInstance = new web3.eth.Contract(accountABI, dsrAddr);
  let connectSBTCCurve;

  beforeEach(async function () {
    connectSBTCCurve = await ConnectSBTCCurve.deployed();

    let wbtcContract = new web3.eth.Contract(erc20.wbtc.abi, erc20.wbtc.address);

    let uniswapFactory = new web3.eth.Contract(
      uniswap.factory.abi,
      uniswap.factory.address
    );

    const wbtcExchangeAddress = await uniswapFactory.methods.getExchange(
      erc20.wbtc.address,
    ).call();

    const wbtcExchange = new web3.eth.Contract(
      uniswap.exchange.abi,
      wbtcExchangeAddress
    );

    const wbtcBefore = await wbtcContract.methods.balanceOf(sender).call();
    console.log("WBTC Before: ", wbtcBefore.toString());

    const balanceBefore = await web3.eth.getBalance(sender);
    console.log("Balance Before: ", balanceBefore.toString());

    await wbtcExchange.methods.ethToTokenSwapInput(
      1, // min amount of token retrieved
      2525644800, // random timestamp in the future (year 2050)
    ).send(
      {
        gas: 4000000,
        value: ether("5"),
        from: sender
      }
    );

    let wbtcAfter = await wbtcContract.methods.balanceOf(sender).call();
    console.log("WBTC After: ", wbtcAfter.toString());
    const balanceAfter = await web3.eth.getBalance(sender);
    console.log("Balance After: ", balanceAfter.toString());

    expect(wbtcAfter - wbtcBefore).to.be.at.least(10000000);

    // send WBTC to master
    await wbtcContract.methods.transfer(dsrAddr, 10000000).send({from: sender});

    // Send ETH to master
    await web3.eth.sendTransaction({from: sender, to: masterAddress, value: ether("50")});

    let connectorID  = await connectSBTCCurve.connectorID();

    // Enable the the given connector address
    await connectorInstance.methods.enable(connectSBTCCurve.address).send({from: masterAddress});
    // check if the give connector address is enabled.
    let isEnabled = await connectorInstance.methods.connectors(connectSBTCCurve.address).call();
    assert.ok(isEnabled);
  });

  it('can sell WBTC for SBTC', async function () {
    const sbtcContract = new web3.eth.Contract(sbtcABI, "0xfe18be6b3bd88a2d2a7f928d00292e7a9963cfc6");

    const sbtcBefore = await sbtcContract.methods.balanceOf(dsrAddr).call();

    const encoded = await connectSBTCCurve.contract.methods.sell(
      "0xfe18be6b3bd88a2d2a7f928d00292e7a9963cfc6",
      erc20.wbtc.address,
      10000000,
      ( 0.09 / 0.1 * 1e18 ).toString(),
      0,
      0,
    ).encodeABI();

    //Inputs for `cast()` function of DSA Account.
    const castInputs = [
      [connectSBTCCurve.address],
      [encoded],
      masterAddress
    ]
    console.log("Cast Inputs: ", castInputs);

    // Execute `cast()` function
    const tx = await accountInstance.methods.cast(...castInputs).send({from: masterAddress});
    console.log(tx);

    const sbtcAfter = await sbtcContract.methods.balanceOf(sender).call();
    expect(sbtcAfter - sbtcBefore).to.be.at.least(ether("0.09"));
  });

  /*
  it('can add and remove liquidity for wbtc', async function() {
    const curveTokenContract = new web3.eth.Contract(
    erc20ABI,
      "0x075b1bb99792c9e1041ba13afef80c91a1e70fb3"
  )

    const txDeposit = await contract.deposit(
      erc20.wbtc.address,
      10000000,
      ( 0.09 / 0.1 * 1e18 ).toString(),
      0,
      0,
      {
        gas: 4000000,
        from: sender
      }
    );
    console.log(txDeposit);

    const balanceDeposit = await curveTokenContract.methods.balanceOf(sender);

    expect(balanceDeposit).to.be.at.least(ether("0.09"));

    const txWithdraw = await contract.withdraw(
      erc20.wbtc.address,
      10000000,
      ( 0.09 / 0.1 * 1e18 ).toString(),
      0,
      0,
      {
        gas: 4000000,
        from: sender
      }
    );
    console.log(txWithdraw);

    const balanceWithdraw = await curveTokenContract.methods.balanceOf(sender);

    expect(balanceWithdraw).to.equal(0);
  });
  */

});