Hangouts remote desktop part III - It2MeService

remoting/webapp/unittests/it2me_service_unittest.js

Index: remoting/webapp/unittests/it2me_service_unittest.js
diff --git a/remoting/webapp/unittests/it2me_service_unittest.js b/remoting/webapp/unittests/it2me_service_unittest.js
new file mode 100644
index 0000000000000000000000000000000000000000..6b0b1ad10b30ad7796d50b7835f6c74bf8f9327c
--- /dev/null
+++ b/remoting/webapp/unittests/it2me_service_unittest.js
@@ -0,0 +1,145 @@
+// Copyright 2014 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+(function() {
+
+'use strict';
+
+var appLauncher = null;
+var hangoutPort = null;
+var webappPort = null;
+var it2meService = null;
+
+function createPort(name, senderId) {
+  var port = new chromeMocks.runtime.Port();
+  port.name = (senderId) ? name +'@' + senderId : name;
+  port.postMessage = sinon.spy(port, 'postMessage');
+  port.disconnect = sinon.spy(port, 'disconnect');
+
+  return port;
+}
+
+function promiseResolveSynchronous(value) {
+  return {
+    then: function(callback) {
+      callback(value);
+    }
+  };
+}
+
+module('It2MeService', {
+  setup: function() {
+    // App Launcher.
+    appLauncher = {
+      launch: function () {
+        return promiseResolveSynchronous('tabId');
+      },
+      close: function () {}
+    };
+    // HangoutPort.
+    hangoutPort = createPort('it2me.helper.hangout');
+    it2meService = new remoting.It2MeService(appLauncher);
+    it2meService.onConnectExternal_(hangoutPort);
+    webappPort = createPort('it2me.helper.webapp', 'tabId');
+  }
+});
+
+test('should establish a channel two way channel when the webapp connects',
+  function() {
+    // Hangout ---- connect ----> It2MeService.
+    hangoutPort.onMessage.mock$fire({
+      method: 'connect',
+      accessCode: "123412341234"
+    });
+
+    // Webapp ---- connect ----> It2MeService.
+    it2meService.onWebappConnect_(webappPort);
+
+    // Webapp ---- sessionStateChanged ----> It2MeService.
+    webappPort.onMessage.mock$fire({
+      method: 'sessionStateChanged',
+      state: remoting.ClientSession.State.CONNECTED
+    });
+
+    // verify that hangout can receive message events.
+    sinon.assert.calledWith(hangoutPort.postMessage, {
+      method: 'sessionStateChanged',
+      state: remoting.ClientSession.State.CONNECTED
+    });
+
+    hangoutPort.onDisconnect.mock$fire();
+    QUnit.equal(it2meService.helpers_.length, 0);
+});
+
+test('should handle multiple helper connections', function() {
+    // Hangout ---- connect ----> It2MeService.
+    hangoutPort.onMessage.mock$fire({
+      method: 'connect',
+      accessCode: "123412341234"
+    });
+
+    // Hangout2 ---- connect ----> It2MeService.
+    var hangoutPort2 = createPort('it2me.helper.hangout');
+    it2meService.onConnectExternal_(hangoutPort2);
+
+    appLauncher.launch = function () {
+      return promiseResolveSynchronous('tabId2');
+    };
+
+    hangoutPort2.onMessage.mock$fire({
+      method: 'connect',
+      accessCode: "123412341234"
+    });
+
+    it2meService.onWebappConnect_(webappPort);
+
+    var webappPort2 = createPort('it2me.helper.webapp', 'tabId2');
+    it2meService.onWebappConnect_(webappPort2);
+
+    webappPort.onMessage.mock$fire({
+      method: 'sessionStateChanged',
+      state: remoting.ClientSession.State.CONNECTED
+    });
+
+    // verify that hangout can receive message events from webapp 1
+    sinon.assert.calledWith(hangoutPort.postMessage, {
+      method: 'sessionStateChanged',
+      state: remoting.ClientSession.State.CONNECTED
+    });
+
+    webappPort2.onMessage.mock$fire({
+      method: 'sessionStateChanged',
+      state: remoting.ClientSession.State.CLOSED
+    });
+
+    // verify that hangout can receive message events from webapp 2.
+    sinon.assert.calledWith(hangoutPort2.postMessage, {
+      method: 'sessionStateChanged',
+      state: remoting.ClientSession.State.CLOSED
+    });
+});
+
+test('should reject unknown connection', function() {
+  it2meService.onWebappConnect_(webappPort);
+  sinon.assert.called(webappPort.disconnect);
+
+  var randomPort = createPort('unsupported.port.name');
+  it2meService.onConnectExternal_(randomPort);
+  sinon.assert.called(randomPort.disconnect);
+});
+
+test('messageExternal("hello") should return supportedFeatures', function() {
+  var response = null;
+  function callback(msg) {
+    response = msg;
+  }
+
+  it2meService.onMessageExternal_({
+    method: 'hello'
+  }, null, callback);
+
+  QUnit.ok(response.value.supportedFeatures instanceof Array);
+});
+
+})();