@gnosis.pm/pm-contracts/test/javascript/digix_tempo.js

Collection of smart contracts for the Gnosis prediction market platform

//This is a modified copy of: https://github.com/DigixGlobal/tempo/blob/master/lib/index.js
// It is not compatible with Web3 v1.0 (https://github.com/DigixGlobal/tempo/issues/5)
// A pull request has been created https://github.com/DigixGlobal/tempo/pull/4
// In the mean time, a local modified copy is used.

'use strict';

module.exports = function (web3) {
  function sendRpc(method, params) {
    return new Promise(function (resolve) {
      web3.currentProvider.send({
        jsonrpc: '2.0',
        method: method,
        params: params || [],
        id: new Date().getTime()
      }, function (err, res) {
        resolve(res);
      });
    });
  }
  function waitUntilBlock(seconds, targetBlock) {
    return new Promise(function (resolve) {
      var asyncIterator = function asyncIterator() {
        return web3.eth.getBlock('latest', function (e, _ref) {
          var number = _ref.number;

          if (number >= targetBlock - 1) {
            return sendRpc('evm_increaseTime', [seconds]).then(function () {
              return sendRpc('evm_mine');
            }).then(resolve);
          }
          return sendRpc('evm_mine').then(asyncIterator);
        });
      };
      asyncIterator();
    });
  }
  function wait() {
    var seconds = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : 20;
    var blocks = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : 1;

    return new Promise(function (resolve) {
      return web3.eth.getBlock('latest', function (e, _ref2) {
        var number = _ref2.number;

        resolve(blocks + number);
      });
    }).then(function (targetBlock) {
      return waitUntilBlock(seconds, targetBlock);
    });
  }
  return { wait: wait, waitUntilBlock: waitUntilBlock };
};