embark-area-51/lib/core/test.js

var async = require('async');
var Web3 = require('web3');

var Embark = require('../index.js');

var ABIGenerator = require('../contracts/abi.js');
var ContractsManager = require('../contracts/contracts.js');
var Deploy = require('../contracts/deploy.js');

var Config = require('./config.js');
var RunCode = require('./runCode.js');
var TestLogger = require('./test_logger.js');

var Test = function(options) {
  this.options = options || {};
  this.logger = new TestLogger({logLevel: this.options.logLevel || 'debug'});
  var simOptions = this.options.simulatorOptions || {};

  try {
    this.sim = require('ethereumjs-testrpc');
  } catch (e) {
    if (e.code === 'MODULE_NOT_FOUND') {
      console.log('Simulator not found; Please install it with "npm install ethereumjs-testrpc --save"');
      console.log('IMPORTANT: if you using a NodeJS version older than 6.9.1 then you need to install an older version of testrpc "npm install ethereumjs-testrpc@2.0 --save"');
      console.log('For more information see https://github.com/ethereumjs/testrpc');
      // TODO: should throw exception instead
      process.exit();
    } else {
      console.log("==============");
      console.log("Tried to load testrpc but an error occurred. This is a problem with testrpc");
      console.log('IMPORTANT: if you using a NodeJS version older than 6.9.1 then you need to install an older version of testrpc "npm install ethereumjs-testrpc@2.0 --save". Alternatively install node 6.9.1 and the testrpc 3.0');
      console.log("==============");
      throw e;
    }
  }

  this.web3 = new Web3();
  this.web3.setProvider(this.sim.provider(simOptions));
};

Test.prototype.deployAll = function(contractsConfig, cb) {
  var self = this;

  async.waterfall([
      function getConfig(callback) {
        var config = new Config({env: 'test', logger: self.logger});
        config.loadConfigFiles({
          embarkConfig: self.options.embarkConfig || 'embark.json', 
          interceptLogs: false
        });
        config.contractsConfig = {contracts: contractsConfig};
        callback(null, config);
      },
      function buildContracts(config, callback) {
        var contractsManager = new ContractsManager({
          contractFiles:  config.contractsFiles,
          contractsConfig: config.contractsConfig,
          logger: self.logger,
          plugins: config.plugins
        });
        contractsManager.build(function() {
          callback(null, contractsManager);
        });
      },
      function deployContracts(contractsManager, callback) {
        var deploy = new Deploy({
          web3: self.web3,
          contractsManager: contractsManager,
          logger: self.logger,
          chainConfig: false,
          env: 'test'
        });
        deploy.deployAll(function() {
          callback(null, contractsManager);
        });
      },
      function generateABI(contractsManager, callback) {
        var abiGenerator = new ABIGenerator({contractsManager: contractsManager});
        var ABI = abiGenerator.generateContracts(false);
        callback(null, ABI);
      }
  ], function(err, result) {
    if (err) {
      throw new Error(err);
    }
    self.web3.eth.getAccounts(function(err, accounts) { 
      if (err) {
        throw new Error(err);
      }
      self.web3.eth.defaultAccount = accounts[0];
      RunCode.doEval(result, self.web3); // jshint ignore:line
      cb();
    });
  });
};

module.exports = Test;