remoting/webapp/base/js/xmpp_connection_unittest.js

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 (function() {
   6
   7 'use strict';
   8
   9 var testUsername = 'testUsername@gmail.com';
  10 var testToken = 'testToken';
  11 var socketId = 3;
  12
  13 var onStanzaStr = null;
  14
  15 /** @type {remoting.XmppConnection} */
  16 var connection = null;
  17
  18 /** @type {remoting.TcpSocket} */
  19 var socket = null;
  20
  21 var stateChangeHandler = function(/** remoting.SignalStrategy.State */ state) {}
  22
  23 function onStateChange(/** remoting.SignalStrategy.State */ state) {
  24   stateChangeHandler(state)
  25 };
  26
  27 /**
  28  * @param {QUnit.Assert} assert
  29  * @param {remoting.SignalStrategy.State} expectedState
  30  * @returns {Promise}
  31  */
  32 function expectNextState(assert, expectedState) {
  33   return new Promise(function(resolve, reject) {
  34     stateChangeHandler = function(/** remoting.SignalStrategy.State */ state) {
  35       assert.equal(state, expectedState);
  36       assert.equal(connection.getState(), expectedState);
  37       resolve(0);
  38     }
  39   });
  40 }
  41
  42 QUnit.module('XmppConnection', {
  43   beforeEach: function() {
  44     remoting.settings = new remoting.Settings();
  45
  46     onStanzaStr = sinon.spy();
  47     /** @param {Element} stanza */
  48     function onStanza(stanza) {
  49       onStanzaStr(new XMLSerializer().serializeToString(stanza));
  50     }
  51
  52     socket = /** @type{remoting.TcpSocket} */
  53         (sinon.createStubInstance(remoting.TcpSocket));
  54
  55     connection = new remoting.XmppConnection();
  56     connection.setSocketForTests(socket);
  57     connection.setStateChangedCallback(onStateChange);
  58     connection.setIncomingStanzaCallback(onStanza);
  59   },
  60   afterEach: function() {
  61     remoting.settings = null;
  62   }
  63 });
  64
  65 QUnit.test('should go to FAILED state when failed to connect',
  66     function(assert) {
  67   var done = assert.async();
  68   $testStub(socket.connect).withArgs("xmpp.example.com", 123)
  69       .returns(new Promise(function(resolve, reject) { reject(-1); }));
  70
  71   var deferredSend = new base.Deferred();
  72   $testStub(socket.send).onFirstCall().returns(deferredSend.promise());
  73
  74   expectNextState(assert, remoting.SignalStrategy.State.CONNECTING)
  75       .then(onConnecting);
  76   connection.connect('xmpp.example.com:123', 'testUsername@gmail.com',
  77                      'testToken');
  78
  79   function onConnecting() {
  80     expectNextState(assert, remoting.SignalStrategy.State.FAILED)
  81         .then(onFailed);
  82   }
  83
  84   function onFailed() {
  85     sinon.assert.calledWith(socket.dispose);
  86     assert.ok(connection.getError().hasTag(remoting.Error.Tag.NETWORK_FAILURE));
  87     done();
  88   }
  89 });
  90
  91 QUnit.test('should use XmppLoginHandler for handshake', function(assert) {
  92
  93   $testStub(socket.connect).withArgs("xmpp.example.com", 123)
  94       .returns(new Promise(function(resolve, reject) { resolve(0) }));
  95
  96   var deferredSend = new base.Deferred();
  97   $testStub(socket.send).onFirstCall().returns(deferredSend.promise());
  98
  99   var parser = new remoting.XmppStreamParser();
 100   var parserMock = sinon.mock(parser);
 101   var setCallbacksCalled = parserMock.expects('setCallbacks').once();
 102   var State = remoting.SignalStrategy.State;
 103
 104   var promise = expectNextState(assert, State.CONNECTING).then(function() {
 105     return expectNextState(assert, State.HANDSHAKE);
 106   }).then(function() {
 107     var handshakeDoneCallback =
 108         connection.loginHandler_.getHandshakeDoneCallbackForTesting();
 109     var onConnected = expectNextState(assert, State.CONNECTED);
 110     handshakeDoneCallback('test@example.com/123123', parser);
 111     return onConnected;
 112   }).then(function() {
 113     setCallbacksCalled.verify();
 114
 115     // Simulate read() callback with |data|. It should be passed to
 116     // the parser.
 117     var data = base.encodeUtf8('<iq id="1">hello</iq>');
 118     sinon.assert.calledWith(socket.startReceiving);
 119     var appendDataCalled =
 120         parserMock.expects('appendData').once().withArgs(data);
 121     $testStub(socket.startReceiving).getCall(0).args[0](data);
 122     appendDataCalled.verify();
 123   });
 124
 125   connection.connect(
 126       'xmpp.example.com:123', 'testUsername@gmail.com', 'testToken');
 127   return promise;
 128 });
 129
 130 })();