feat(tests): add '--suite' option to 'ng e2e' command

packages/angular-cli/commands/e2e.ts

@@ -1,12 +1,21 @@
 const Command = require('../ember-cli/lib/models/command');
-import {E2eTask} from '../tasks/e2e';
-import {CliConfig} from '../models/config';
+import { E2eTask } from '../tasks/e2e';
+import { CliConfig } from '../models/config';
+
+export interface E2eOptions {
+  suite?: string;
+}
 
 const E2eCommand = Command.extend({
   name: 'e2e',
   description: 'Run e2e tests in existing project',
   works: 'insideProject',
-  run: function () {
+
+  availableOptions: [
+    { name: 'suite', type: String, default: null },
+  ],
+
+  run: function (commandOptions: E2eOptions) {
     this.project.ngConfig = this.project.ngConfig || CliConfig.fromProject();
 
     const e2eTask = new E2eTask({
@@ -15,7 +24,7 @@ const E2eCommand = Command.extend({
       project: this.project
     });
 
-    return e2eTask.run();
+    return e2eTask.run(commandOptions);
   }
 });
 

packages/angular-cli/tasks/e2e.ts

@@ -1,15 +1,21 @@
 const Task = require('../ember-cli/lib/models/task');
 import * as chalk from 'chalk';
-import {exec} from 'child_process';
-
+import { exec } from 'child_process';
+import { E2eOptions } from '../commands/e2e';
 
 export const E2eTask = Task.extend({
-  run: function () {
+  run: function (options: E2eOptions) {
+    const commandArgs: string[] = [];
     const ui = this.ui;
     let exitCode = 0;
 
+    if (options.suite) {
+      commandArgs.push(`--suite=${options.suite}`);
+    }
+
     return new Promise((resolve) => {
-      exec(`npm run e2e -- ${this.project.ngConfig.config.e2e.protractor.config}`,
+      const protractorConfig = this.project.ngConfig.config.e2e.protractor.config;
+      exec(`npm run e2e -- ${protractorConfig} ${commandArgs.join(' ')}`,
         (err: NodeJS.ErrnoException, stdout: string, stderr: string) => {
           ui.writeLine(stdout);
           if (err) {

tests/helpers/mock-project.js

@@ -0,0 +1,59 @@
+'use strict';
+
+var Project = require('angular-cli/ember-cli/lib/models/project');
+var MockUI = require('./mock-ui');
+
+function MockProject() {
+  var root = process.cwd();
+  var pkg  = {};
+  var ui = new MockUI();
+  Project.apply(this, [root, pkg, ui]);
+}
+
+MockProject.prototype.require = function(file) {
+  if (file === './server') {
+    return function() {
+      return {
+        listen: function() {
+          arguments[arguments.length - 1]();
+        }
+      };
+    };
+  }
+};
+
+MockProject.prototype.config = function() {
+  return this._config || {
+    baseURL: '/',
+    locationType: 'auto'
+  };
+};
+
+MockProject.prototype.has = function(key) {
+  return (/server/.test(key));
+};
+
+MockProject.prototype.name = function() {
+  return 'mock-project';
+};
+
+MockProject.prototype.initializeAddons = Project.prototype.initializeAddons;
+MockProject.prototype.hasDependencies = function() {
+  return true;
+};
+MockProject.prototype.discoverAddons = Project.prototype.discoverAddons;
+MockProject.prototype.addIfAddon = Project.prototype.addIfAddon;
+MockProject.prototype.supportedInternalAddonPaths = Project.prototype.supportedInternalAddonPaths;
+MockProject.prototype.setupBowerDirectory = Project.prototype.setupBowerDirectory;
+MockProject.prototype.setupNodeModulesPath = Project.prototype.setupNodeModulesPath;
+MockProject.prototype.isEmberCLIProject = Project.prototype.isEmberCLIProject;
+MockProject.prototype.isEmberCLIAddon = Project.prototype.isEmberCLIAddon;
+MockProject.prototype.findAddonByName = Project.prototype.findAddonByName;
+MockProject.prototype.dependencies = function() {
+  return [];
+};
+MockProject.prototype.isEmberCLIAddon = function() {
+  return false;
+};
+
+module.exports = MockProject;

tests/runner.js

@@ -7,7 +7,7 @@ var Mocha = require('mocha');
 var glob = require('glob');
 var path = require('path');
 
-var root = 'tests/{acceptance,models}';
+var root = 'tests/{acceptance,unit}';
 var specFiles = glob.sync(root + '/**/*.spec.*');
 var mocha = new Mocha({ timeout: 5000, reporter: 'spec' });
 

tests/unit/commands/e2e.spec.ts

@@ -0,0 +1,48 @@
+import E2eCommand from 'angular-cli/commands/e2e';
+import { CliConfig } from 'angular-cli/models/config/config';
+import { stub } from 'sinon';
+import { expect } from 'chai';
+import * as proc from 'child_process';
+
+const MockUI = require('../../helpers/mock-ui');
+const MockAnalytics = require('../../helpers/mock-analytics');
+const MockProject = require('../../helpers/mock-project');
+
+function createProject() {
+  const project = new MockProject();
+  project.isEmberCLIProject = () => true;
+  project.ngConfig = CliConfig.fromJson({
+    e2e: {
+      protractor: {
+        config: 'some-config'
+      }
+    }
+  });
+
+  return project;
+}
+
+describe('e2e command', () => {
+  let command: E2eCommand;
+
+  beforeEach(() => {
+    command = new E2eCommand({
+      settings: {},
+      project: createProject(),
+      ui: new MockUI(),
+      analytics: new MockAnalytics(),
+    });
+  });
+
+  beforeEach(() => {
+    stub(proc, 'exec').callsArg(1);
+  });
+
+  it('passes through the suite option', () => {
+    const commandOption = '--suite=suiteA';
+    return command.validateAndRun([commandOption]).then(() => {
+      expect(proc.exec.calledOnce).to.be.true;
+      expect(proc.exec.firstCall.args[0]).to.have.string(` ${commandOption}`);
+    });
+  });
+});