refactor: move angular source to /packages rather than modules/@angular

packages/benchpress/test/web_driver_extension_spec.ts

@@ -0,0 +1,56 @@
+/**
+ * @license
+ * Copyright Google Inc. All Rights Reserved.
+ *
+ * Use of this source code is governed by an MIT-style license that can be
+ * found in the LICENSE file at https://angular.io/license
+ */
+
+import {AsyncTestCompleter, describe, expect, inject, it} from '@angular/core/testing/testing_internal';
+
+import {Options, ReflectiveInjector, WebDriverExtension} from '../index';
+
+export function main() {
+  function createExtension(ids: any[], caps: any) {
+    return new Promise<any>((res, rej) => {
+      try {
+        res(ReflectiveInjector
+                .resolveAndCreate([
+                  ids.map((id) => ({provide: id, useValue: new MockExtension(id)})),
+                  {provide: Options.CAPABILITIES, useValue: caps},
+                  WebDriverExtension.provideFirstSupported(ids)
+                ])
+                .get(WebDriverExtension));
+      } catch (e) {
+        rej(e);
+      }
+    });
+  }
+
+  describe('WebDriverExtension.provideFirstSupported', () => {
+
+    it('should provide the extension that matches the capabilities',
+       inject([AsyncTestCompleter], (async: AsyncTestCompleter) => {
+         createExtension(['m1', 'm2', 'm3'], {'browser': 'm2'}).then((m) => {
+           expect(m.id).toEqual('m2');
+           async.done();
+         });
+       }));
+
+    it('should throw if there is no match',
+       inject([AsyncTestCompleter], (async: AsyncTestCompleter) => {
+         createExtension(['m1'], {'browser': 'm2'}).catch((err) => {
+           expect(err != null).toBe(true);
+           async.done();
+         });
+       }));
+  });
+}
+
+class MockExtension extends WebDriverExtension {
+  constructor(public id: string) { super(); }
+
+  supports(capabilities: {[key: string]: any}): boolean {
+    return capabilities['browser'] === this.id;
+  }
+}