mavensmate/app/lib/commands/unit-test/get-coverage.js

Core APIs that drive MavensMate IDEs for Salesforce1/Force.com

/**
 * @file Returns current test coverage for the org or a given path
 * @author Joseph Ferraro <@joeferraro>
 */

'use strict';

var Promise         = require('bluebird');
var inherits        = require('inherits');
var BaseCommand     = require('../../command');
var ApexTest        = require('../../services/test');
var MavensMateFile  = require('../../file').MavensMateFile;

function Command() {
  Command.super_.call(this, Array.prototype.slice.call(arguments, 0));
}

inherits(Command, BaseCommand);

Command.prototype.execute = function() {
  var self = this;
  return new Promise(function(resolve, reject) {
    var test = new ApexTest({
      project: self.getProject()
    });
    var commandPromise;
    if (self.payload.global) {
      commandPromise = test.getOrgWideCoverage();
    } else {
      var mmFile = new MavensMateFile({ project: self.getProject(), path: self.payload.paths[0] });
      test.apexClassOrTriggerIdToName[mmFile.id] = mmFile.basename;
      commandPromise = test.getCoverage([ mmFile.id ]);
    }
    commandPromise
      .then(function(res) {
        resolve(res);
      })
      .catch(function(error) {
        reject(error);
      })
      .done();
  });
};

exports.command = Command;
exports.addSubCommand = function(program) {
	program
		.command('get-coverage [apexClassOrTriggerPath]')
    .option('-g, --global', 'Org-wide coverage')
		.description('Gets coverage for a specified class')
		.action(function(apexClassOrTriggerPath){
      if (this.global) {
        program.commandExecutor.execute({
          name: this._name,
          body: {
            global: true
          }
        });
      } else {
        program.commandExecutor.execute({
          name: this._name,
          body: {
            paths: [ apexClassOrTriggerPath ]
          }
        });
      }
		});
};