| // 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 "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/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) { |
| registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_TEST_MESSAGE, |
| content::NotificationService::AllSources()); |
| chromeos::input_method::InputMethodManager::Get()-> |
| EnableLayouts(kLoginScreenUILanguage, kInitialInputMethodOnLoginScreen); |
| } |
| |
| virtual ~SetInputMethodListener() { |
| 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_; |
| }; |
| |
| 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_; |
| } |