From baaaf4ad883770851855f8c7ae8cebcc76ae2835 Mon Sep 17 00:00:00 2001
From: Thrilok Kumar <thrilok2000@gmail.com>
Date: Sat, 22 Aug 2020 09:32:46 +0530
Subject: [PATCH] Fixed test cases

---
 contracts/tests/MockCurveGaugeMapping.sol | 6 ++++++
 test/CurveGauge.js                        | 9 ++++++---
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/contracts/tests/MockCurveGaugeMapping.sol b/contracts/tests/MockCurveGaugeMapping.sol
index c80852c..0aa52ae 100644
--- a/contracts/tests/MockCurveGaugeMapping.sol
+++ b/contracts/tests/MockCurveGaugeMapping.sol
@@ -4,5 +4,11 @@ pragma experimental ABIEncoderV2;
 import { CurveGaugeMapping } from "../mapping/curve_gauge_mapping.sol";
 
 contract MockCurveGaugeMapping is CurveGaugeMapping {
+    constructor(
+    string[] memory gaugeNames,
+    address[] memory gaugeAddresses,
+    bool[] memory rewardTokens
+    ) public CurveGaugeMapping(gaugeNames, gaugeAddresses, rewardTokens) {
+  }
   modifier isChief override {_;}
 }
diff --git a/test/CurveGauge.js b/test/CurveGauge.js
index 96c2d26..e9ef1c5 100644
--- a/test/CurveGauge.js
+++ b/test/CurveGauge.js
@@ -18,7 +18,10 @@ contract("ConnectCurveGauge", async accounts => {
 
   before(async function () {
     mock = await MockContract.new();
-    mockCurveGaugeMapping = await MockCurveGaugeMapping.new();
+    let names = ["compound", "susd"]
+    let poolAddress = [mock.address, mock.address]
+    let rewardArr = [false, true]
+    mockCurveGaugeMapping = await MockCurveGaugeMapping.new(names, poolAddress, rewardArr);
     mockCurveGauge = await MockCurveGauge.new(mock.address, mockCurveGaugeMapping.address);
     // lp_token = new web3.eth.Contract(erc20ABI, mock.address);
     curveGauge = new web3.eth.Contract(gaugeABI, mock.address)
@@ -32,8 +35,8 @@ contract("ConnectCurveGauge", async accounts => {
     let rewarded_token = await curveGauge.methods.rewarded_token().encodeABI();
     await mock.givenMethodReturnAddress(rewarded_token, mock.address);
 
-    await mockCurveGaugeMapping.addGaugeMapping('compound', mock.address, false);
-    await mockCurveGaugeMapping.addGaugeMapping('susd', mock.address, true);
+    // await mockCurveGaugeMapping.addGaugeMapping('compound', mock.address, false);
+    // await mockCurveGaugeMapping.addGaugeMapping('susd', mock.address, true);
   })
 
   it('can deposit into compound gauge', async function() {