@harvest-finance/harvest-strategy-polygon 0.0.1-security → 1.0.0

package/test/quick-gamma/usdc-eth-wide.js

@@ -0,0 +1,197 @@
+// Utilities
+const Utils = require("../utilities/Utils.js");
+const { impersonates, setupCoreProtocol, depositVault } = require("../utilities/hh-utils.js");
+
+const addresses = require("../test-config.js");
+const BigNumber = require("bignumber.js");
+const IERC20 = artifacts.require("@openzeppelin/contracts/token/ERC20/IERC20.sol:IERC20");
+
+const Strategy = artifacts.require("QuickGammaStrategyMainnet_USDC_ETH_wide");
+
+//This test was developed at blockNumber 39013300
+
+// Vanilla Mocha test. Increased compatibility with tools that integrate Mocha.
+describe("Mainnet Quickswap-Gamma USDC-ETH (wide)", function() {
+  let accounts;
+
+  // external contracts
+  let underlying;
+
+  // external setup
+  let underlyingWhale = "0xCB6222f4Df04385ea08E8DA2A5871131FF5F6cBA";
+
+  // parties in the protocol
+  let governance;
+  let farmer1;
+
+  // numbers used in tests
+  let farmerBalance;
+
+  // Core protocol contracts
+  let controller;
+  let vault;
+  let strategy;
+
+  async function setupExternalContracts() {
+    underlying = await IERC20.at("0x6077177d4c41E114780D9901C9b5c784841C523f");
+    console.log("Fetching Underlying at: ", underlying.address);
+  }
+
+  async function setupBalance(){
+    let etherGiver = accounts[9];
+    // Give whale some ether to make sure the following actions are good
+    await web3.eth.sendTransaction({ from: etherGiver, to: underlyingWhale, value: 1e18});
+
+    farmerBalance = await underlying.balanceOf(underlyingWhale);
+    await underlying.transfer(farmer1, farmerBalance, { from: underlyingWhale });
+  }
+
+  before(async function() {
+    governance = "0xf00dD244228F51547f0563e60bCa65a30FBF5f7f";
+    accounts = await web3.eth.getAccounts();
+
+    farmer1 = accounts[1];
+
+    // impersonate accounts
+    await impersonates([governance, underlyingWhale]);
+
+    await setupExternalContracts();
+    [controller, vault, strategy] = await setupCoreProtocol({
+      "existingVaultAddress": null,
+      "strategyArtifact": Strategy,
+      "strategyArtifactIsUpgradable": true,
+      "underlying": underlying,
+      "governance": governance,
+    });
+
+    // whale send underlying to farmers
+    await setupBalance();
+  });
+
+  describe("Happy path", function() {
+    it("Farmer should earn money", async function() {
+      let farmerOldBalance = new BigNumber(await underlying.balanceOf(farmer1));
+      console.log("Farmer old balance:", farmerOldBalance.toFixed());
+      await depositVault(farmer1, underlying, vault, farmerBalance);
+      let fTokenBalance = new BigNumber(await vault.balanceOf(farmer1));
+      console.log("Farmer fToken balance:", fTokenBalance.toFixed());
+
+      // Using half days is to simulate how we doHardwork in the real world
+      let hours = 10;
+      let blocksPerHour = 1565*5;
+      let oldSharePrice;
+      let newSharePrice;
+      for (let i = 0; i < hours; i++) {
+        console.log("loop ", i);
+
+        oldSharePrice = new BigNumber(await vault.getPricePerFullShare());
+        await controller.doHardWork(vault.address, { from: governance });
+        newSharePrice = new BigNumber(await vault.getPricePerFullShare());
+
+        console.log("old shareprice: ", oldSharePrice.toFixed());
+        console.log("new shareprice: ", newSharePrice.toFixed());
+        console.log("growth: ", newSharePrice.toFixed() / oldSharePrice.toFixed());
+
+        apr = (newSharePrice.toFixed()/oldSharePrice.toFixed()-1)*(24/(blocksPerHour/1565))*365;
+        apy = ((newSharePrice.toFixed()/oldSharePrice.toFixed()-1)*(24/(blocksPerHour/1565))+1)**365;
+
+        console.log("instant APR:", apr*100, "%");
+        console.log("instant APY:", (apy-1)*100, "%");
+
+        await Utils.advanceNBlock(blocksPerHour);
+      }
+      await vault.withdraw(fTokenBalance, { from: farmer1 });
+      let farmerNewBalance = new BigNumber(await underlying.balanceOf(farmer1));
+      Utils.assertBNGt(farmerNewBalance, farmerOldBalance);
+
+      apr = (farmerNewBalance.toFixed()/farmerOldBalance.toFixed()-1)*(24/(blocksPerHour*hours/1565))*365;
+      apy = ((farmerNewBalance.toFixed()/farmerOldBalance.toFixed()-1)*(24/(blocksPerHour*hours/1565))+1)**365;
+
+      console.log("earned!");
+      console.log("Overall APR:", apr*100, "%");
+      console.log("Overall APY:", (apy-1)*100, "%");
+
+      await strategy.withdrawAllToVault({ from: governance }); // making sure can withdraw all for a next switch
+    });
+  });
+
+
+
+  describe("Basic strategy functionality checks", function () {
+    it("Deposit to vault", async function () {
+      // deposit to vault such that we have something to work with
+      await depositVault(farmer1, underlying, vault, farmerBalance);
+
+      // check vault has at least farmerBalance (there could be some dust etc., so no exact check)
+      let vaultUnderlying = new BigNumber(await vault.underlyingBalanceInVault());
+      Utils.assertBNGte(vaultUnderlying, farmerBalance)
+    });
+
+    it("withdrawAllToVault()", async function () {
+      let vaultUnderlying = new BigNumber(await vault.underlyingBalanceInVault());
+      // doHardwork invests all underlying from the vault
+      await controller.doHardWork(vault.address, { from: governance });
+      let investedUnderlyingBalance = new BigNumber(await strategy.investedUnderlyingBalance());
+
+      // strategy invested amount must be greater or equal than the initial vault underlying amount
+      Utils.assertBNGte(investedUnderlyingBalance, vaultUnderlying)
+
+      // move all invested underlying back to the vault
+      await strategy.withdrawAllToVault({ from: governance });
+
+      // strategy has no underlying left
+      let investedUnderlyingBalanceNew = new BigNumber(await strategy.investedUnderlyingBalance());
+      assert.equal(investedUnderlyingBalanceNew.eq(BigNumber(0)), true);
+
+      // vault has all of the underlying from the strategy
+      let vaultUnderlyingNew = new BigNumber(await vault.underlyingBalanceInVault());
+      Utils.assertBNGte(vaultUnderlyingNew, investedUnderlyingBalance)
+    });
+
+    it("withdrawToVault(uint256 _amount)", async function () {
+      let vaultUnderlying = new BigNumber(await vault.underlyingBalanceInVault());
+      // doHardwork invests all underlying from the vault
+      await controller.doHardWork(vault.address, { from: governance });
+      let investedUnderlyingBalance = new BigNumber(await strategy.investedUnderlyingBalance());
+
+      // strategy invested amount must be greater or equal than the initial vault underlying amount
+      Utils.assertBNGte(investedUnderlyingBalance, vaultUnderlying)
+
+      let amount = 1;
+      // move amount of underlying to vault
+      await strategy.withdrawToVault(amount, { from: governance });
+
+      // vault has exactly amount of underlying
+      let vaultUnderlyingNew = new BigNumber(await vault.underlyingBalanceInVault());
+      assert.equal(new BigNumber(amount).eq(vaultUnderlyingNew), true);
+    });
+
+    it("unsalvagableTokens(address _token)", async function () {
+      // underlying is unsalvageable
+      let isUnderlyingUnsalvagable = await strategy.unsalvagableTokens(underlying.address);
+      assert.equal(isUnderlyingUnsalvagable, true);
+
+      // rewardToken is unsalvageable
+      let isRewardTokenUnsalvagable = await strategy.unsalvagableTokens(await strategy.rewardToken());
+      assert.equal(isRewardTokenUnsalvagable, true);
+
+    });
+
+    it("Remove all deposited LP tokens", async function () {
+      // remove all deposited LP tokens
+      let fTokenBalance = await vault.balanceOf(farmer1);
+      await vault.withdraw(fTokenBalance, { from: farmer1 });
+
+      //check that there is nothing left in the strategy or vault
+      let strategyBalance = new BigNumber(await underlying.balanceOf(strategy.address));
+      assert.equal(strategyBalance.eq(BigNumber(0)), true);
+
+      let vaultBalance = new BigNumber(await underlying.balanceOf(vault.address));
+      assert.equal(vaultBalance.eq(BigNumber(0)), true);
+
+      let totalUnderlyingVaultAndStrategy = new BigNumber(await vault.underlyingBalanceWithInvestment());
+      assert.equal(totalUnderlyingVaultAndStrategy.eq(BigNumber(0)), true);
+    });
+  });
+
+});

package/test/quick-gamma/usdc-usdt.js

@@ -0,0 +1,209 @@
+// Utilities
+const Utils = require("../utilities/Utils.js");
+const { impersonates, setupCoreProtocol, depositVault } = require("../utilities/hh-utils.js");
+
+const addresses = require("../test-config.js");
+const BigNumber = require("bignumber.js");
+const IERC20 = artifacts.require("@openzeppelin/contracts/token/ERC20/IERC20.sol:IERC20");
+
+const Strategy = artifacts.require("QuickGammaStrategyV2Mainnet_USDC_USDT");
+
+//This test was developed at blockNumber 48397851
+
+// Vanilla Mocha test. Increased compatibility with tools that integrate Mocha.
+describe("Mainnet Quickswap-Gamma USDC-USDT", function() {
+  let accounts;
+
+  // external contracts
+  let underlying;
+
+  // external setup
+  let underlyingWhale = "0x7fC687e66397D38B5dee913Af5C7daE7E9AF67B6";
+  let quick = "0xB5C064F955D8e7F38fE0460C556a72987494eE17";
+  let wmatic = "0x0d500B1d8E8eF31E21C99d1Db9A6444d3ADf1270";
+  let usdc = "0x2791Bca1f2de4661ED88A30C99A7a9449Aa84174";
+  let usdt = "0xc2132D05D31c914a87C6611C10748AEb04B58e8F";
+
+  // parties in the protocol
+  let governance;
+  let farmer1;
+
+  // numbers used in tests
+  let farmerBalance;
+
+  // Core protocol contracts
+  let controller;
+  let vault;
+  let strategy;
+
+  async function setupExternalContracts() {
+    underlying = await IERC20.at("0x795f8c9B0A0Da9Cd8dea65Fc10f9B57AbC532E58");
+    console.log("Fetching Underlying at: ", underlying.address);
+  }
+
+  async function setupBalance(){
+    let etherGiver = accounts[9];
+    // Give whale some ether to make sure the following actions are good
+    await web3.eth.sendTransaction({ from: etherGiver, to: underlyingWhale, value: 100e18});
+
+    farmerBalance = await underlying.balanceOf(underlyingWhale);
+    await underlying.transfer(farmer1, farmerBalance, { from: underlyingWhale });
+  }
+
+  before(async function() {
+    governance = addresses.Governance;
+    accounts = await web3.eth.getAccounts();
+
+    farmer1 = accounts[1];
+
+    // impersonate accounts
+    await impersonates([governance, underlyingWhale]);
+
+    let etherGiver = accounts[9];
+    await web3.eth.sendTransaction({ from: etherGiver, to: governance, value: 100e18});
+
+    await setupExternalContracts();
+    [controller, vault, strategy] = await setupCoreProtocol({
+      "existingVaultAddress": null,
+      "strategyArtifact": Strategy,
+      "strategyArtifactIsUpgradable": true,
+      "underlying": underlying,
+      "governance": governance,
+      "liquidation": [
+        {"quickswap": [quick, wmatic]},
+        {"uniV3": [wmatic, usdc]},
+        {"uniV3": [wmatic, usdt]},
+      ],
+    });
+
+    // whale send underlying to farmers
+    await setupBalance();
+  });
+
+  describe("Happy path", function() {
+    it("Farmer should earn money", async function() {
+      let farmerOldBalance = new BigNumber(await underlying.balanceOf(farmer1));
+      console.log("Farmer old balance:", farmerOldBalance.toFixed());
+      await depositVault(farmer1, underlying, vault, farmerBalance);
+      let fTokenBalance = new BigNumber(await vault.balanceOf(farmer1));
+      console.log("Farmer fToken balance:", fTokenBalance.toFixed());
+
+      // Using half days is to simulate how we doHardwork in the real world
+      let hours = 10;
+      let blocksPerHour = 1565*5;
+      let oldSharePrice;
+      let newSharePrice;
+      for (let i = 0; i < hours; i++) {
+        console.log("loop ", i);
+
+        oldSharePrice = new BigNumber(await vault.getPricePerFullShare());
+        await controller.doHardWork(vault.address, { from: governance });
+        newSharePrice = new BigNumber(await vault.getPricePerFullShare());
+
+        console.log("old shareprice: ", oldSharePrice.toFixed());
+        console.log("new shareprice: ", newSharePrice.toFixed());
+        console.log("growth: ", newSharePrice.toFixed() / oldSharePrice.toFixed());
+
+        apr = (newSharePrice.toFixed()/oldSharePrice.toFixed()-1)*(24/(blocksPerHour/1565))*365;
+        apy = ((newSharePrice.toFixed()/oldSharePrice.toFixed()-1)*(24/(blocksPerHour/1565))+1)**365;
+
+        console.log("instant APR:", apr*100, "%");
+        console.log("instant APY:", (apy-1)*100, "%");
+
+        await Utils.advanceNBlock(blocksPerHour);
+      }
+      await vault.withdraw(fTokenBalance, { from: farmer1 });
+      let farmerNewBalance = new BigNumber(await underlying.balanceOf(farmer1));
+      Utils.assertBNGt(farmerNewBalance, farmerOldBalance);
+
+      apr = (farmerNewBalance.toFixed()/farmerOldBalance.toFixed()-1)*(24/(blocksPerHour*hours/1565))*365;
+      apy = ((farmerNewBalance.toFixed()/farmerOldBalance.toFixed()-1)*(24/(blocksPerHour*hours/1565))+1)**365;
+
+      console.log("earned!");
+      console.log("Overall APR:", apr*100, "%");
+      console.log("Overall APY:", (apy-1)*100, "%");
+
+      await strategy.withdrawAllToVault({ from: governance }); // making sure can withdraw all for a next switch
+    });
+  });
+
+
+
+  describe("Basic strategy functionality checks", function () {
+    it("Deposit to vault", async function () {
+      // deposit to vault such that we have something to work with
+      await depositVault(farmer1, underlying, vault, farmerBalance);
+
+      // check vault has at least farmerBalance (there could be some dust etc., so no exact check)
+      let vaultUnderlying = new BigNumber(await vault.underlyingBalanceInVault());
+      Utils.assertBNGte(vaultUnderlying, farmerBalance)
+    });
+
+    it("withdrawAllToVault()", async function () {
+      let vaultUnderlying = new BigNumber(await vault.underlyingBalanceInVault());
+      // doHardwork invests all underlying from the vault
+      await controller.doHardWork(vault.address, { from: governance });
+      let investedUnderlyingBalance = new BigNumber(await strategy.investedUnderlyingBalance());
+
+      // strategy invested amount must be greater or equal than the initial vault underlying amount
+      Utils.assertBNGte(investedUnderlyingBalance, vaultUnderlying)
+
+      // move all invested underlying back to the vault
+      await strategy.withdrawAllToVault({ from: governance });
+
+      // strategy has no underlying left
+      let investedUnderlyingBalanceNew = new BigNumber(await strategy.investedUnderlyingBalance());
+      assert.equal(investedUnderlyingBalanceNew.eq(BigNumber(0)), true);
+
+      // vault has all of the underlying from the strategy
+      let vaultUnderlyingNew = new BigNumber(await vault.underlyingBalanceInVault());
+      Utils.assertBNGte(vaultUnderlyingNew, investedUnderlyingBalance)
+    });
+
+    it("withdrawToVault(uint256 _amount)", async function () {
+      let vaultUnderlying = new BigNumber(await vault.underlyingBalanceInVault());
+      // doHardwork invests all underlying from the vault
+      await controller.doHardWork(vault.address, { from: governance });
+      let investedUnderlyingBalance = new BigNumber(await strategy.investedUnderlyingBalance());
+
+      // strategy invested amount must be greater or equal than the initial vault underlying amount
+      Utils.assertBNGte(investedUnderlyingBalance, vaultUnderlying)
+
+      let amount = 1;
+      // move amount of underlying to vault
+      await strategy.withdrawToVault(amount, { from: governance });
+
+      // vault has exactly amount of underlying
+      let vaultUnderlyingNew = new BigNumber(await vault.underlyingBalanceInVault());
+      assert.equal(new BigNumber(amount).eq(vaultUnderlyingNew), true);
+    });
+
+    it("unsalvagableTokens(address _token)", async function () {
+      // underlying is unsalvageable
+      let isUnderlyingUnsalvagable = await strategy.unsalvagableTokens(underlying.address);
+      assert.equal(isUnderlyingUnsalvagable, true);
+
+      // rewardToken is unsalvageable
+      let isRewardTokenUnsalvagable = await strategy.unsalvagableTokens(await strategy.rewardToken());
+      assert.equal(isRewardTokenUnsalvagable, true);
+
+    });
+
+    it("Remove all deposited LP tokens", async function () {
+      // remove all deposited LP tokens
+      let fTokenBalance = await vault.balanceOf(farmer1);
+      await vault.withdraw(fTokenBalance, { from: farmer1 });
+
+      //check that there is nothing left in the strategy or vault
+      let strategyBalance = new BigNumber(await underlying.balanceOf(strategy.address));
+      assert.equal(strategyBalance.eq(BigNumber(0)), true);
+
+      let vaultBalance = new BigNumber(await underlying.balanceOf(vault.address));
+      assert.equal(vaultBalance.eq(BigNumber(0)), true);
+
+      let totalUnderlyingVaultAndStrategy = new BigNumber(await vault.underlyingBalanceWithInvestment());
+      assert.equal(totalUnderlyingVaultAndStrategy.eq(BigNumber(0)), true);
+    });
+  });
+
+});

package/test/quick-gamma/wbtc-eth-narrow.js

@@ -0,0 +1,207 @@
+// Utilities
+const Utils = require("../utilities/Utils.js");
+const { impersonates, setupCoreProtocol, depositVault } = require("../utilities/hh-utils.js");
+
+const addresses = require("../test-config.js");
+const BigNumber = require("bignumber.js");
+const IERC20 = artifacts.require("@openzeppelin/contracts/token/ERC20/IERC20.sol:IERC20");
+
+const Strategy = artifacts.require("QuickGammaStrategyV2Mainnet_WBTC_ETH_narrow");
+
+//This test was developed at blockNumber 48397300
+
+// Vanilla Mocha test. Increased compatibility with tools that integrate Mocha.
+describe("Mainnet Quickswap-Gamma WBTC-ETH (narrow)", function() {
+  let accounts;
+
+  // external contracts
+  let underlying;
+
+  // external setup
+  let underlyingWhale = "0xa4C21Aaf21f03B9F090b6b7456DEAbdad6369556";
+  let quick = "0xB5C064F955D8e7F38fE0460C556a72987494eE17";
+  let weth = "0x7ceB23fD6bC0adD59E62ac25578270cFf1b9f619";
+  let wbtc = "0x1BFD67037B42Cf73acF2047067bd4F2C47D9BfD6";
+
+  // parties in the protocol
+  let governance;
+  let farmer1;
+
+  // numbers used in tests
+  let farmerBalance;
+
+  // Core protocol contracts
+  let controller;
+  let vault;
+  let strategy;
+
+  async function setupExternalContracts() {
+    underlying = await IERC20.at("0x4B9e26a02121a1C541403a611b542965Bd4b68Ce");
+    console.log("Fetching Underlying at: ", underlying.address);
+  }
+
+  async function setupBalance(){
+    let etherGiver = accounts[9];
+    // Give whale some ether to make sure the following actions are good
+    await web3.eth.sendTransaction({ from: etherGiver, to: underlyingWhale, value: 100e18});
+
+    farmerBalance = await underlying.balanceOf(underlyingWhale);
+    await underlying.transfer(farmer1, farmerBalance, { from: underlyingWhale });
+  }
+
+  before(async function() {
+    governance = addresses.Governance;
+    accounts = await web3.eth.getAccounts();
+
+    farmer1 = accounts[1];
+
+    // impersonate accounts
+    await impersonates([governance, underlyingWhale]);
+
+    let etherGiver = accounts[9];
+    await web3.eth.sendTransaction({ from: etherGiver, to: governance, value: 100e18});
+
+    await setupExternalContracts();
+    [controller, vault, strategy] = await setupCoreProtocol({
+      "existingVaultAddress": null,
+      "strategyArtifact": Strategy,
+      "strategyArtifactIsUpgradable": true,
+      "underlying": underlying,
+      "governance": governance,
+      "liquidation": [
+        {"quickswap": [quick, weth]},
+        {"quickswap": [quick, weth, wbtc]},
+      ],
+    });
+
+    // whale send underlying to farmers
+    await setupBalance();
+  });
+
+  describe("Happy path", function() {
+    it("Farmer should earn money", async function() {
+      let farmerOldBalance = new BigNumber(await underlying.balanceOf(farmer1));
+      console.log("Farmer old balance:", farmerOldBalance.toFixed());
+      await depositVault(farmer1, underlying, vault, farmerBalance);
+      let fTokenBalance = new BigNumber(await vault.balanceOf(farmer1));
+      console.log("Farmer fToken balance:", fTokenBalance.toFixed());
+
+      // Using half days is to simulate how we doHardwork in the real world
+      let hours = 10;
+      let blocksPerHour = 1565*5;
+      let oldSharePrice;
+      let newSharePrice;
+      for (let i = 0; i < hours; i++) {
+        console.log("loop ", i);
+
+        oldSharePrice = new BigNumber(await vault.getPricePerFullShare());
+        await controller.doHardWork(vault.address, { from: governance });
+        newSharePrice = new BigNumber(await vault.getPricePerFullShare());
+
+        console.log("old shareprice: ", oldSharePrice.toFixed());
+        console.log("new shareprice: ", newSharePrice.toFixed());
+        console.log("growth: ", newSharePrice.toFixed() / oldSharePrice.toFixed());
+
+        apr = (newSharePrice.toFixed()/oldSharePrice.toFixed()-1)*(24/(blocksPerHour/1565))*365;
+        apy = ((newSharePrice.toFixed()/oldSharePrice.toFixed()-1)*(24/(blocksPerHour/1565))+1)**365;
+
+        console.log("instant APR:", apr*100, "%");
+        console.log("instant APY:", (apy-1)*100, "%");
+
+        await Utils.advanceNBlock(blocksPerHour);
+      }
+      await vault.withdraw(fTokenBalance, { from: farmer1 });
+      let farmerNewBalance = new BigNumber(await underlying.balanceOf(farmer1));
+      Utils.assertBNGt(farmerNewBalance, farmerOldBalance);
+
+      apr = (farmerNewBalance.toFixed()/farmerOldBalance.toFixed()-1)*(24/(blocksPerHour*hours/1565))*365;
+      apy = ((farmerNewBalance.toFixed()/farmerOldBalance.toFixed()-1)*(24/(blocksPerHour*hours/1565))+1)**365;
+
+      console.log("earned!");
+      console.log("Overall APR:", apr*100, "%");
+      console.log("Overall APY:", (apy-1)*100, "%");
+
+      await strategy.withdrawAllToVault({ from: governance }); // making sure can withdraw all for a next switch
+    });
+  });
+
+
+
+  describe("Basic strategy functionality checks", function () {
+    it("Deposit to vault", async function () {
+      // deposit to vault such that we have something to work with
+      await depositVault(farmer1, underlying, vault, farmerBalance);
+
+      // check vault has at least farmerBalance (there could be some dust etc., so no exact check)
+      let vaultUnderlying = new BigNumber(await vault.underlyingBalanceInVault());
+      Utils.assertBNGte(vaultUnderlying, farmerBalance)
+    });
+
+    it("withdrawAllToVault()", async function () {
+      let vaultUnderlying = new BigNumber(await vault.underlyingBalanceInVault());
+      // doHardwork invests all underlying from the vault
+      await controller.doHardWork(vault.address, { from: governance });
+      let investedUnderlyingBalance = new BigNumber(await strategy.investedUnderlyingBalance());
+
+      // strategy invested amount must be greater or equal than the initial vault underlying amount
+      Utils.assertBNGte(investedUnderlyingBalance, vaultUnderlying)
+
+      // move all invested underlying back to the vault
+      await strategy.withdrawAllToVault({ from: governance });
+
+      // strategy has no underlying left
+      let investedUnderlyingBalanceNew = new BigNumber(await strategy.investedUnderlyingBalance());
+      assert.equal(investedUnderlyingBalanceNew.eq(BigNumber(0)), true);
+
+      // vault has all of the underlying from the strategy
+      let vaultUnderlyingNew = new BigNumber(await vault.underlyingBalanceInVault());
+      Utils.assertBNGte(vaultUnderlyingNew, investedUnderlyingBalance)
+    });
+
+    it("withdrawToVault(uint256 _amount)", async function () {
+      let vaultUnderlying = new BigNumber(await vault.underlyingBalanceInVault());
+      // doHardwork invests all underlying from the vault
+      await controller.doHardWork(vault.address, { from: governance });
+      let investedUnderlyingBalance = new BigNumber(await strategy.investedUnderlyingBalance());
+
+      // strategy invested amount must be greater or equal than the initial vault underlying amount
+      Utils.assertBNGte(investedUnderlyingBalance, vaultUnderlying)
+
+      let amount = 1;
+      // move amount of underlying to vault
+      await strategy.withdrawToVault(amount, { from: governance });
+
+      // vault has exactly amount of underlying
+      let vaultUnderlyingNew = new BigNumber(await vault.underlyingBalanceInVault());
+      assert.equal(new BigNumber(amount).eq(vaultUnderlyingNew), true);
+    });
+
+    it("unsalvagableTokens(address _token)", async function () {
+      // underlying is unsalvageable
+      let isUnderlyingUnsalvagable = await strategy.unsalvagableTokens(underlying.address);
+      assert.equal(isUnderlyingUnsalvagable, true);
+
+      // rewardToken is unsalvageable
+      let isRewardTokenUnsalvagable = await strategy.unsalvagableTokens(await strategy.rewardToken());
+      assert.equal(isRewardTokenUnsalvagable, true);
+
+    });
+
+    it("Remove all deposited LP tokens", async function () {
+      // remove all deposited LP tokens
+      let fTokenBalance = await vault.balanceOf(farmer1);
+      await vault.withdraw(fTokenBalance, { from: farmer1 });
+
+      //check that there is nothing left in the strategy or vault
+      let strategyBalance = new BigNumber(await underlying.balanceOf(strategy.address));
+      assert.equal(strategyBalance.eq(BigNumber(0)), true);
+
+      let vaultBalance = new BigNumber(await underlying.balanceOf(vault.address));
+      assert.equal(vaultBalance.eq(BigNumber(0)), true);
+
+      let totalUnderlyingVaultAndStrategy = new BigNumber(await vault.underlyingBalanceWithInvestment());
+      assert.equal(totalUnderlyingVaultAndStrategy.eq(BigNumber(0)), true);
+    });
+  });
+
+});