Add TGEDeployer contract with tests

Author: AugustoL

contracts/deploy/TGEDeployer.sol

@@ -0,0 +1,52 @@
+pragma solidity ^0.4.18;
+
+import "../LifCrowdsale.sol";
+
+/**
+   @title TGEDeployer, A deployer contract for the Winding Tree TGE
+
+   This contract is used to create a crowdsale and issue presale tokens in batches
+   it will also set the weiPerUSD and transfer ownership, after that everything is
+   ready for the TGE to succed.
+ */
+contract TGEDeployer {
+
+  LifCrowdsale public crowdsale;
+  address public wallet;
+  address public owner;
+
+  function TGEDeployer(
+    uint256 startTimestamp,
+    uint256 end1Timestamp,
+    uint256 end2Timestamp,
+    uint256 rate1,
+    uint256 rate2,
+    uint256 setWeiLockSeconds,
+    address foundationWallet,
+    address foundersWallet
+  ) public {
+    crowdsale = new LifCrowdsale(
+      startTimestamp, end1Timestamp, end2Timestamp, rate1, rate2,
+      setWeiLockSeconds, foundationWallet, foundersWallet
+    );
+    wallet = foundationWallet;
+    owner = msg.sender;
+  }
+
+  // Mint a batch of presale tokens
+  function addPresaleTokens(address[] contributors, uint256[] values, uint256 rate) public {
+    require(msg.sender == owner);
+    require(contributors.length == values.length);
+    for (uint32 i = 0; i < contributors.length; i ++) {
+      crowdsale.addPrivatePresaleTokens(contributors[i], values[i], rate);
+    }
+  }
+
+  // Set the wei per USD in the crowdsale and then transfer ownership to foundation
+  function finish(uint256 weiPerUSDinTGE) public {
+    require(msg.sender == owner);
+    crowdsale.setWeiPerUSDinTGE(weiPerUSDinTGE);
+    crowdsale.transferOwnership(wallet);
+  }
+
+}

test/deploy/TGEDeployer.js

@@ -0,0 +1,108 @@
+var LifCrowdsale = artifacts.require('./LifCrowdsale.sol'),
+  LifToken = artifacts.require('./LifToken.sol'),
+  TGEDeployer = artifacts.require('./deploy/TGEDeployer.sol');
+
+let help = require('../helpers');
+
+var BigNumber = web3.BigNumber;
+
+require('chai')
+  .use(require('chai-bignumber')(BigNumber))
+  .should();
+
+var { duration } = require('../helpers/increaseTime');
+const _ = require('lodash');
+
+const TGEDistribution = require('./TGEDistribution');
+
+contract('TGE Deployer', function ([deployAddress, foundationWallet, foundersWallet]) {
+  it('Deploy a TGE correctly', async function () {
+    const startTimestamp = new Date('Sun, 28 Jan 2018 8:0:00 GMT').getTime() / 1000;
+    const end1Timestamp = new Date('Sun, 28 Jan 2018 9:0:00 GMT').getTime() / 1000;
+    const end2Timestamp = new Date('Sun, 28 Jan 2018 10:0:00 GMT').getTime() / 1000;
+    const rate1 = 1000;
+    const rate2 = 900;
+    const setWeiLockSeconds = duration.hours(1);
+    var totalSupply = new BigNumber(0);
+    const weiPerUSDinTGE = web3.toWei(1 / 1000);
+
+    const deployer = await TGEDeployer.new(
+      startTimestamp, end1Timestamp, end2Timestamp, rate1,
+      rate2, setWeiLockSeconds, foundationWallet, foundersWallet,
+      { from: deployAddress }
+    );
+
+    const crowdsale = LifCrowdsale.at(await deployer.crowdsale());
+    const token = LifToken.at(await crowdsale.token());
+
+    help.debug('Data to create Deployer contract:');
+    help.debug(deployer.deployedBytecode);
+    help.debug('--------------------------------------------------');
+
+    // Check values
+    assert.equal(startTimestamp, parseInt(await crowdsale.startTimestamp.call()));
+    assert.equal(end1Timestamp, parseInt(await crowdsale.end1Timestamp.call()));
+    assert.equal(end2Timestamp, parseInt(await crowdsale.end2Timestamp.call()));
+    assert.equal(rate1, parseInt(await crowdsale.rate1.call()));
+    assert.equal(rate2, parseInt(await crowdsale.rate2.call()));
+    assert.equal(foundationWallet, parseInt(await crowdsale.foundationWallet.call()));
+    assert.equal(foundersWallet, parseInt(await crowdsale.foundersWallet.call()));
+
+    var processStage = async function (stage) {
+      // Check right amount of contributos and values
+      assert.equal(stage.contributors.length, stage.values.length);
+      var stageETH = new BigNumber(0);
+
+      // Parse ETH to wei
+      stage.values.map(function (value, i) {
+        stageETH = stageETH.add(stage.values[i]);
+        stage.values[i] = web3.toWei(stage.values[i]);
+      });
+
+      // Add TGE stage
+      const contributorsChunks = _.chunk(stage.contributors, 150);
+      const valuesChunks = _.chunk(stage.values, 150);
+      var txs = [];
+      for (var i = 0; i < contributorsChunks.length; i++) {
+        const data = await deployer.contract.addPresaleTokens.getData(
+          contributorsChunks[i], valuesChunks[i], stage.rate
+        );
+        await deployer.addPresaleTokens(
+          contributorsChunks[i], valuesChunks[i], stage.rate,
+          { from: deployAddress }
+        );
+        txs.push(data);
+      }
+
+      // Calculate tokens and check total supply
+      const stageTokens = new BigNumber(stageETH).mul(stage.rate);
+      totalSupply = totalSupply.add(stageTokens);
+      help.debug('TXs for stage', stage.name);
+      txs.map(function (tx, i) { help.debug('TX [', i, ']', tx); });
+      help.debug('--------------------------------------------------');
+      totalSupply.should.be.bignumber
+        .equal(help.lifWei2Lif(await token.totalSupply()), 2);
+    };
+
+    await processStage(TGEDistribution[0]);
+    await processStage(TGEDistribution[1]);
+    await processStage(TGEDistribution[2]);
+    await processStage(TGEDistribution[3]);
+    await processStage(TGEDistribution[4]);
+
+    const finalizeData = await deployer.contract.finish.getData(weiPerUSDinTGE);
+
+    await deployer.finish(weiPerUSDinTGE, { from: deployAddress });
+
+    help.debug('Data to finalize Deployer:');
+    help.debug(finalizeData);
+    help.debug('--------------------------------------------------');
+
+    assert.equal(foundationWallet, parseInt(await crowdsale.owner.call()));
+    assert.equal(weiPerUSDinTGE, parseInt(await crowdsale.weiPerUSDinTGE.call()));
+
+    // Check final total supply
+    totalSupply.should.be.bignumber
+      .equal(help.lifWei2Lif(await token.totalSupply()), 2);
+  });
+});