Handle host need upgrade in Pin browser test

remoting/webapp/browser_test/browser_test.js

 // 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.
 
 /**
  * @fileoverview
  * @suppress {checkTypes}  By default, JSCompile is not run on test files.
  *    However, you can modify |remoting_webapp_files.gypi| locally to include
  *    the test in the package to expedite local development.  This suppress
  *    is here so that JSCompile won't complain.
@@ skipping 138 matching lines @@
       }
     }
 
     if (opt_timeout != browserTest.Timeout.NONE) {
       timerId = window.setTimeout(onTimeout, opt_timeout);
     }
     remoting.testEvents.addEventListener(uiModeChanged, onUIModeChanged);
   });
 };
 
+browserTest.connectMe2Me = function() {

[garykac, 2014/07/11 23:25:39]

connectMe2Me doesn't feel like the right name for this function.
This function doesn't actually connect - it just takes the connection flow up to
the Pin dialog prompt.

Perhaps clickConnectButton?

+  var AppMode = remoting.AppMode;
+  browserTest.clickOnControl('this-host-connect');
+  return browserTest.onUIMode(AppMode.CLIENT_HOST_NEEDS_UPGRADE).then(
+    function() {
+      // On fulfilled.
+      browserTest.clickOnControl('host-needs-update-connect-button');
+    }, function() {
+      // On time out.
+      return Promise.resolve();
+    }).then(function() {
+      return browserTest.onUIMode(AppMode.CLIENT_PIN_PROMPT);
+    });
+}
+
 browserTest.expectMe2MeError = function(errorTag) {
   var AppMode = remoting.AppMode;
   var Timeout = browserTest.Timeout;
 
   var onConnected = browserTest.onUIMode(AppMode.IN_SESSION, Timeout.None);
   var onFailure = browserTest.onUIMode(AppMode.CLIENT_CONNECT_FAILED_ME2ME);
 
   onConnected = onConnected.then(function() {
     return Promise.reject(
         'Expected the Me2Me connection to fail.');
@@ skipping 35 matching lines @@
   try {
     var test = new testClass();
     browserTest.expect(typeof test.run == 'function');
     test.run(data);
   } catch (e) {
     browserTest.fail(e);
   }
 };
 
 browserTest.init();