[IME] migrate the xkb ID to extension based xkb ID.

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/extensions/api/test/test_api.h"
 #include "chrome/common/chrome_switches.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 "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";
 
 // Class that listens for the JS message then changes input method and replies
 // back.
 class SetInputMethodListener : public content::NotificationObserver {
  public:
   // Creates listener, which should reply exactly |count_| times.
   explicit SetInputMethodListener(int count) : count_(count) {
+    use_ext_xkb_flag_ = chromeos::extension_ime_util::
+        SetUseWrappedExtensionKeyboardLayoutsFlagForTesting(false);
     registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_TEST_MESSAGE,
                    content::NotificationService::AllSources());
     std::vector<std::string> keyboard_layouts;
     keyboard_layouts.push_back(kInitialInputMethodOnLoginScreen);
     chromeos::input_method::InputMethodManager::Get()->EnableLoginLayouts(
         kLoginScreenUILanguage, keyboard_layouts);
   }
 
   virtual ~SetInputMethodListener() {
+    chromeos::extension_ime_util::
+        SetUseWrappedExtensionKeyboardLayoutsFlagForTesting(use_ext_xkb_flag_);
     EXPECT_EQ(0, count_);
   }
 
   // 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();
     const std::string expected_message =
         base::StringPrintf("%s:%s", kSetInputMethodMessage, kNewInputMethod);
     if (content == expected_message) {
       chromeos::input_method::InputMethodManager::Get()->
           ChangeInputMethod(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_;
+  bool use_ext_xkb_flag_;
 };
 
 class ExtensionInputMethodApiTest : public ExtensionApiTest {
   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
     ExtensionApiTest::SetUpCommandLine(command_line);
     command_line->AppendSwitchASCII(
         switches::kWhitelistedExtensionID, "ilanclmaeigfpnmdlgelmhkpkegdioip");
   }
 };
 
 }  // namespace
 
 IN_PROC_BROWSER_TEST_F(ExtensionInputMethodApiTest, Basic) {
   // Two test, two calls. See JS code for more info.
   SetInputMethodListener listener(2);
 
   ASSERT_TRUE(RunExtensionTest("input_method")) << message_;
 }