Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1712)

Unified Diff: remoting/webapp/base/js/protocol_extension_manager_unittest.js

Issue 1067133002: Move ProtocolExtensionManager from SessionConnector into its own class (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 5 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: remoting/webapp/base/js/protocol_extension_manager_unittest.js
diff --git a/remoting/webapp/base/js/protocol_extension_manager_unittest.js b/remoting/webapp/base/js/protocol_extension_manager_unittest.js
new file mode 100644
index 0000000000000000000000000000000000000000..596dd2ca3ad5fafdbc8f24d6926cc20e68aabcd0
--- /dev/null
+++ b/remoting/webapp/base/js/protocol_extension_manager_unittest.js
@@ -0,0 +1,117 @@
+// Copyright 2015 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.
+
+/**
+ * @fileoverview
+ */
+
+(function() {
+
+'use strict';
+
+/** @type {remoting.ProtocolExtensionManager} */
+var extensionManager;
+
+/** @type {(sinon.Spy|function(string, string))} */
+var sendClientMessage;
+
+/**
+ * @constructor
+ * @param {Array<string>} types
+ * @implements {remoting.ProtocolExtension}
+ */
+var DummyExtension = function(types) {
+ /** @private {?function(string, string)} */
+ this.sendMessageToHost_ = null;
+ /** @private */
+ this.types_ = types;
+};
+
+DummyExtension.prototype.getExtensionTypes = function() {
+ return this.types_.slice(0);
+}
+
+/**
+ * @param {function(string,string)} sendMessageToHost Callback to send a message
+ * to the host.
+ */
+DummyExtension.prototype.startExtension = function(sendMessageToHost) {
+ this.sendMessageToHost_ = sendMessageToHost;
+};
+
+/**
+ * Called when an extension message of a matching type is received.
+ *
+ * @param {string} type The message type.
+ * @param {Object} message The parsed extension message data.
+ * @return {boolean} True if the extension message was handled.
+ */
+DummyExtension.prototype.onExtensionMessage = function(type, message){
+ return this.types_.indexOf(type) !== 1;
+};
+
+
+QUnit.module('ProtocolExtensionManager', {
+ beforeEach: function() {
+ sendClientMessage = /** @type {function(string, string)} */ (sinon.spy());
+ extensionManager = new remoting.ProtocolExtensionManager(sendClientMessage);
+ },
+ afterEach: function() {
+ }
+});
+
+QUnit.test('should route message to extension by type', function(assert) {
+ var extension = new DummyExtension(['type1', 'type2']);
+ var onExtensionMessage = /** @type {(sinon.Spy|function(string, string))} */ (
+ sinon.spy(extension, 'onExtensionMessage'));
+ extensionManager.register(extension);
+ extensionManager.start();
+
+ extensionManager.onProtocolExtensionMessage('type1', '{"message": "hello"}');
+ assert.ok(onExtensionMessage.called);
+ onExtensionMessage.reset();
+
+ extensionManager.onProtocolExtensionMessage('type2', '{"message": "hello"}');
+ assert.ok(onExtensionMessage.called);
+ onExtensionMessage.reset();
+
+ extensionManager.onProtocolExtensionMessage('type3', '{"message": "hello"}');
+ assert.ok(!onExtensionMessage.called);
+ onExtensionMessage.reset();
+});
+
+
+QUnit.test('should not register extensions of the same type', function(assert) {
garykac 2015/04/07 22:14:16 I assume that there's no way to detect that an err
kelvinp 2015/04/08 00:26:02 I modified register to return true when an extensi
+ var extension1 = new DummyExtension(['type1']);
+ var extension2 = new DummyExtension(['type1']);
+
+ var onExtensionMessage1 = /** @type {(sinon.Spy|function(string, string))} */(
+ sinon.spy(extension1, 'onExtensionMessage'));
+ var onExtensionMessage2 = /** @type {(sinon.Spy|function(string, string))} */(
+ sinon.spy(extension2, 'onExtensionMessage'));
+
+ extensionManager.register(extension1);
+ extensionManager.register(extension2);
+ extensionManager.start();
+
+ extensionManager.onProtocolExtensionMessage('type1', '{"message": "hello"}');
+ assert.ok(onExtensionMessage1.called);
+ assert.ok(!onExtensionMessage2.called);
+});
+
+QUnit.test('should handle extensions registration after it is started',
+ function(assert) {
+ var extension = new DummyExtension(['type']);
+
+ var onExtensionMessage = /** @type {(sinon.Spy|function(string, string))} */(
+ sinon.spy(extension, 'onExtensionMessage'));
+
+ extensionManager.start();
+ extensionManager.register(extension);
+
+ extensionManager.onProtocolExtensionMessage('type', '{"message": "hello"}');
+ assert.ok(onExtensionMessage.called);
+});
+
+})();

Powered by Google App Engine
This is Rietveld 408576698