test(WebWorker): add test for WebSocket patching in WebWorker

jeffbcross · jeffbcross · commit 7c302bb0e8cf · 2016-02-12T21:43:11.000-08:00
diff --git a/karma.conf.js b/karma.conf.js
@@ -5,11 +5,15 @@ module.exports = function (config) {
     basePath: '',
     files: [
       'test/browser_entry_point.ts',
+      'test/zone_worker_entry_point.ts',
       'test/**/*.spec.ts',
       {pattern: 'test/assets/**/*.*', watched: true, served: true, included: false},
       // Autowatcch all files to trigger rerun
       {pattern: 'lib/**/*.ts', watched: true, served: false, included: false},
-      {pattern: 'test/**/*.ts', watched: true, served: false, included: false}
+      {pattern: 'test/**/*.ts', watched: true, served: false, included: false},
+      {pattern: 'test/ws-webworker-context.ts', watched: true, served: true, included: false},
+
+      {pattern: 'dist/**/*.js', watched: true, served: true, included: false}
     ],
 
     /*
@@ -23,7 +27,9 @@ module.exports = function (config) {
 
     preprocessors: {
       'test/browser_entry_point.ts': [ 'webpack', 'sourcemap' ],
+      'test/zone_worker_entry_point.ts': [ 'webpack', 'sourcemap' ],
       'test/**/*.spec.ts': [ 'webpack', 'sourcemap' ],
+      'test/ws-webworker-context.ts': ['webpack', 'sourcemap']
     },
     webpack: {
       devtool: 'source-map',
diff --git a/test/patch/WebSocket.spec.ts b/test/patch/WebSocket.spec.ts
@@ -24,6 +24,15 @@ describe('WebSocket', ifEnvSupports('WebSocket', function () {
   });
 
 
+  it('should be patched in a Web Worker', done => {
+    var worker = new Worker('/base/test/ws-webworker-context.ts');
+    worker.onmessage = (e:MessageEvent) => {
+      expect(e.data).toBe('pass');
+      done();
+    }
+  });
+
+
   it('should work with addEventListener', function (done) {
     testZone.run(function() {
       socket.addEventListener('message', function (event) {
diff --git a/test/ws-webworker-context.ts b/test/ws-webworker-context.ts
@@ -0,0 +1,17 @@
+declare var zone;
+importScripts('/base/test/zone_worker_entry_point.ts');
+
+zone.fork().run(() => {
+  var websocket = new WebSocket('ws://localhost:8001');
+  websocket.addEventListener('open', () => {
+    var outerZone = (<any>self).zone;
+    websocket.onmessage = () => {
+      if((<any>self).zone.parent === outerZone) {
+        (<any>self).postMessage('pass');
+      } else {
+        (<any>self).postMessage('fail');
+      }
+    };
+    websocket.send('text');
+  });
+});
diff --git a/test/zone_worker_entry_point.ts b/test/zone_worker_entry_point.ts
@@ -0,0 +1,2 @@
+// Setup tests for Zone without microtask support
+import '../lib/browser/zone';

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+// Setup tests for Zone without microtask support`
	`2`	`+import '../lib/browser/zone';`