Adds IME switching private APIs.

chrome/browser/chromeos/extensions/input_method_apitest_chromeos.cc

 // Copyright (c) 2012 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.
 
 #include "chrome/browser/extensions/extension_apitest.h"
 
 #include <vector>
 
 #include "base/command_line.h"
 #include "base/memory/ref_counted.h"
 #include "base/strings/stringprintf.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/chromeos/extensions/input_method_event_router.h"
 #include "chrome/browser/chromeos/input_method/input_method_util.h"
 #include "chromeos/ime/extension_ime_util.h"
 #include "chromeos/ime/input_method_manager.h"
 #include "content/public/browser/notification_observer.h"
 #include "content/public/browser/notification_registrar.h"
 #include "content/public/browser/notification_service.h"
 #include "extensions/common/switches.h"
 #include "extensions/browser/api/test/test_api.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
 const char kLoginScreenUILanguage[] = "fr";
 const char kInitialInputMethodOnLoginScreen[] = "xkb:us::eng";
-const char kNewInputMethod[] = "fr::fra";
-const char kSetInputMethodMessage[] = "setInputMethod";
-const char kSetInputMethodDone[] = "done";
 const char kBackgroundReady[] = "ready";
 
-// Class that listens for the JS message then changes input method and replies
-// back.
-class SetInputMethodListener : public content::NotificationObserver {
+// Class that listens for the JS message.
+class TestListener : public content::NotificationObserver {
  public:
-  // Creates listener, which should reply exactly |count_| times.
-  explicit SetInputMethodListener(int count) : count_(count) {
+  explicit TestListener() {
     registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_TEST_MESSAGE,
                    content::NotificationService::AllSources());
   }
 
-  virtual ~SetInputMethodListener() {
-    EXPECT_EQ(0, count_);
-  }
+  virtual ~TestListener() {}
 
   // Implements the content::NotificationObserver interface.
   virtual void Observe(int type,
                        const content::NotificationSource& source,
                        const content::NotificationDetails& details) OVERRIDE {
     const std::string& content = *content::Details<std::string>(details).ptr();
     if (content == kBackgroundReady) {
       // Initializes IMF for testing when receives ready message from
       // background.
       chromeos::input_method::InputMethodManager* manager =
           chromeos::input_method::InputMethodManager::Get();
       manager->GetInputMethodUtil()->InitXkbInputMethodsForTesting();
 
       std::vector<std::string> keyboard_layouts;
       keyboard_layouts.push_back(
           chromeos::extension_ime_util::GetInputMethodIDByEngineID(
               kInitialInputMethodOnLoginScreen));
       manager->EnableLoginLayouts(kLoginScreenUILanguage, keyboard_layouts);
-      return;
-    }
-
-    const std::string expected_message =
-        base::StringPrintf("%s:%s", kSetInputMethodMessage, kNewInputMethod);
-    if (content == expected_message) {
-      chromeos::input_method::InputMethodManager::Get()->ChangeInputMethod(
-          chromeos::extension_ime_util::GetInputMethodIDByEngineID(
-              base::StringPrintf("xkb:%s", kNewInputMethod)));
-
-      scoped_refptr<extensions::TestSendMessageFunction> function =
-          content::Source<extensions::TestSendMessageFunction>(
-              source).ptr();
-      EXPECT_GT(count_--, 0);
-      function->Reply(kSetInputMethodDone);
     }
   }
 
  private:
   content::NotificationRegistrar registrar_;
-
-  int count_;
 };
 
 class ExtensionInputMethodApiTest : public ExtensionApiTest {
   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
     ExtensionApiTest::SetUpCommandLine(command_line);
     command_line->AppendSwitchASCII(
         extensions::switches::kWhitelistedExtensionID,
         "ilanclmaeigfpnmdlgelmhkpkegdioip");
   }
 };
 
 }  // namespace
 
 IN_PROC_BROWSER_TEST_F(ExtensionInputMethodApiTest, Basic) {
-  // Two test, two calls. See JS code for more info.
-  SetInputMethodListener listener(2);
+  // Listener for extension's background ready.
+  TestListener listener;
 
   ASSERT_TRUE(RunExtensionTest("input_method")) << message_;
 }