Index: chrome/browser/ui/webui/options/password_manager_handler_unittest.cc |
diff --git a/chrome/browser/ui/webui/options/password_manager_handler_unittest.cc b/chrome/browser/ui/webui/options/password_manager_handler_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..b24532166b85646e8d7aeb75beeabea86e913252 |
--- /dev/null |
+++ b/chrome/browser/ui/webui/options/password_manager_handler_unittest.cc |
@@ -0,0 +1,230 @@ |
+// Copyright 2016 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 "base/metrics/histogram.h" |
+#include "base/metrics/statistics_recorder.h" |
+#include "base/strings/utf_string_conversions.h" |
+#include "chrome/browser/password_manager/password_store_factory.h" |
+#include "chrome/browser/ui/passwords/password_manager_presenter.h" |
+#include "chrome/browser/ui/webui/options/password_manager_handler.h" |
+#include "chrome/test/base/testing_profile.h" |
+#include "components/password_manager/core/browser/mock_password_store.h" |
+#include "components/password_manager/core/browser/password_manager_test_utils.h" |
+#include "content/public/test/test_browser_thread_bundle.h" |
+#include "content/public/test/test_web_ui.h" |
+#include "content/public/test/web_contents_tester.h" |
+#include "testing/gmock/include/gmock/gmock.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+#include "ui/shell_dialogs/select_file_dialog.h" |
+#include "ui/shell_dialogs/select_file_dialog_factory.h" |
+#include "ui/shell_dialogs/select_file_policy.h" |
+ |
+using password_manager::MockPasswordStore; |
+ |
+namespace { |
+class TestSelectFileDialogFactory final : public ui::SelectFileDialogFactory { |
+ public: |
+ TestSelectFileDialogFactory() {} |
+ ~TestSelectFileDialogFactory() override {} |
+ ui::SelectFileDialog* Create(ui::SelectFileDialog::Listener* listener, |
+ ui::SelectFilePolicy* policy) override { |
+ return new TestSelectFileDialog(listener, new TestSelectFilePolicy); |
+ } |
+ |
+ private: |
+ class TestSelectFilePolicy : public ui::SelectFilePolicy { |
+ public: |
+ bool CanOpenSelectFileDialog() override { return true; } |
+ void SelectFileDenied() override {} |
+ }; |
+ |
+ class TestSelectFileDialog : public ui::SelectFileDialog { |
+ public: |
+ TestSelectFileDialog(Listener* listener, ui::SelectFilePolicy* policy) |
+ : ui::SelectFileDialog(listener, policy) {} |
+ |
+ protected: |
+ void SelectFileImpl(Type type, |
+ const base::string16& title, |
+ const base::FilePath& default_path, |
+ const FileTypeInfo* file_types, |
+ int file_type_index, |
+ const base::FilePath::StringType& default_extension, |
+ gfx::NativeWindow owning_window, |
+ void* params) override { |
+ listener_->FileSelected(default_path, file_type_index, params); |
+ } |
+ bool IsRunning(gfx::NativeWindow owning_window) const override { |
+ return false; |
+ } |
+ void ListenerDestroyed() override {} |
+ bool HasMultipleFileTypeChoicesImpl() override { return false; } |
+ ~TestSelectFileDialog() override {} |
+ }; |
+}; |
+ |
+class CallbackTestWebUI : public content::TestWebUI { |
+ public: |
+ CallbackTestWebUI() {} |
+ ~CallbackTestWebUI() override {} |
+ void RegisterMessageCallback(const std::string& message, |
+ const MessageCallback& callback) override; |
+ void ProcessWebUIMessage(const GURL& source_url, |
+ const std::string& message, |
+ const base::ListValue& args) override; |
+ |
+ MOCK_CONST_METHOD0(GetWebContents, content::WebContents*()); |
+ |
+ private: |
+ std::map<std::string, MessageCallback> message_callbacks_; |
+}; |
+ |
+void CallbackTestWebUI::RegisterMessageCallback( |
+ const std::string& message, |
+ const MessageCallback& callback) { |
+ message_callbacks_.insert(std::make_pair(message, callback)); |
+} |
+ |
+void CallbackTestWebUI::ProcessWebUIMessage(const GURL& source_url, |
+ const std::string& message, |
+ const base::ListValue& args) { |
+ std::map<std::string, MessageCallback>::const_iterator callback = |
+ message_callbacks_.find(message); |
+ if (callback != message_callbacks_.end()) { |
+ callback->second.Run(&args); |
+ } |
+} |
+ |
+class TestPasswordManagerHandler : public options::PasswordManagerHandler { |
+ public: |
+ TestPasswordManagerHandler(scoped_ptr<PasswordManagerPresenter> presenter, |
+ CallbackTestWebUI* web_ui) |
+ : PasswordManagerHandler(std::move(presenter)) { |
+ set_web_ui(web_ui); |
+ } |
+ ~TestPasswordManagerHandler() override {} |
+#if !defined(OS_ANDROID) |
+ gfx::NativeWindow GetNativeWindow() const override; |
+#endif |
+ |
+ MOCK_METHOD3(FileSelected, void(const base::FilePath& path, int, void*)); |
+}; |
+#if !defined(OS_ANDROID) |
+gfx::NativeWindow TestPasswordManagerHandler::GetNativeWindow() const { |
+ return NULL; |
+} |
+#endif |
+ |
+class MockPasswordManagerPresenter : public PasswordManagerPresenter { |
+ public: |
+ explicit MockPasswordManagerPresenter(PasswordUIView* handler) |
+ : PasswordManagerPresenter(handler) {} |
+ ~MockPasswordManagerPresenter() override {} |
+ |
+ MOCK_METHOD0(IsUserAuthenticated, bool()); |
+ |
+ private: |
+ DISALLOW_COPY_AND_ASSIGN(MockPasswordManagerPresenter); |
+}; |
+ |
+class DummyPasswordManagerHandler : public PasswordUIView { |
+ public: |
+ DummyPasswordManagerHandler() : password_manager_presenter_(this) { |
+ password_manager_presenter_.Initialize(); |
+ } |
+ ~DummyPasswordManagerHandler() override {} |
+ Profile* GetProfile() override; |
+ |
+ void ShowPassword(size_t, |
+ const std::string&, |
+ const std::string&, |
+ const base::string16&) override {} |
+ void SetPasswordList( |
+ const std::vector<scoped_ptr<autofill::PasswordForm>>&) override {} |
+ void SetPasswordExceptionList( |
+ const std::vector<scoped_ptr<autofill::PasswordForm>>&) override {} |
+ |
+#if !defined(OS_ANDROID) |
+ gfx::NativeWindow GetNativeWindow() const override; |
+#endif |
+ private: |
+ TestingProfile profile_; |
+ PasswordManagerPresenter password_manager_presenter_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(DummyPasswordManagerHandler); |
+}; |
+ |
+#if !defined(OS_ANDROID) |
+gfx::NativeWindow DummyPasswordManagerHandler::GetNativeWindow() const { |
+ return NULL; |
+} |
+#endif |
+ |
+Profile* DummyPasswordManagerHandler::GetProfile() { |
+ return &profile_; |
+} |
+ |
+} // namespace |
+ |
+class PasswordManagerHandlerTest : public testing::Test { |
+ protected: |
+ PasswordManagerHandlerTest() { |
+ dummy_handler_.reset(new DummyPasswordManagerHandler()); |
+ presenter_raw_ = new MockPasswordManagerPresenter(dummy_handler_.get()); |
+ web_contents_ = |
+ content::WebContentsTester::CreateTestWebContents(&profile_, NULL); |
+ web_ui_.set_web_contents(web_contents_); |
+ handler_.reset(new TestPasswordManagerHandler( |
+ make_scoped_ptr(presenter_raw_), &web_ui_)); |
+ handler_->RegisterMessages(); |
+ ui::SelectFileDialog::SetFactory(new TestSelectFileDialogFactory); |
+ handler_->InitializeHandler(); |
+ } |
+ |
+ ~PasswordManagerHandlerTest() override {} |
+ |
+ void ExportPassword() { |
+ base::ListValue tmp; |
+ web_ui_.ProcessWebUIMessage(GURL(), "exportPassword", tmp); |
+ } |
+ |
+ void ImportPassword() { |
+ base::ListValue tmp; |
+ web_ui_.ProcessWebUIMessage(GURL(), "importPassword", tmp); |
+ } |
+ |
+ PasswordManagerPresenter* presenter_raw_; |
+ CallbackTestWebUI web_ui_; |
+ content::WebContents* web_contents_; |
+ scoped_ptr<DummyPasswordManagerHandler> dummy_handler_; |
+ scoped_ptr<TestPasswordManagerHandler> handler_; |
+ |
+ private: |
+ content::TestBrowserThreadBundle thread_bundle_; |
+ TestingProfile profile_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(PasswordManagerHandlerTest); |
+}; |
+ |
+MATCHER(IsEmptyPath, "") { |
+ return arg.empty(); |
+} |
+ |
+TEST_F(PasswordManagerHandlerTest, PasswordImport) { |
+ EXPECT_CALL(web_ui_, GetWebContents()) |
+ .WillRepeatedly(testing::Return(web_contents_)); |
+ EXPECT_CALL(*(handler_.get()), |
Evan Stade
2016/04/15 20:52:42
do you need this many parens? In fact I think *han
xunlu
2016/04/18 02:09:13
Done.
|
+ FileSelected(IsEmptyPath(), 1, reinterpret_cast<void*>(0))); |
vabr (Chromium)
2016/04/12 08:55:39
Here and on line 228 you are referencing FileSelec
xunlu
2016/04/18 02:09:13
Done.
|
+ ImportPassword(); |
+} |
+ |
+TEST_F(PasswordManagerHandlerTest, PasswordExport) { |
+ const base::FilePath file_path; |
+ EXPECT_CALL(*(static_cast<MockPasswordManagerPresenter*>(presenter_raw_)), |
Evan Stade
2016/04/15 20:52:42
do you need this many parens?
xunlu
2016/04/18 02:09:13
Casting is needed here, otherwise the code won't c
|
+ IsUserAuthenticated()) |
+ .WillRepeatedly(testing::Return(true)); |
vabr (Chromium)
2016/04/12 08:55:39
Please also use Times(AtLeast(1)), to ensure that
xunlu
2016/04/18 02:09:13
Done.
|
+ EXPECT_CALL(*(handler_.get()), |
Evan Stade
2016/04/15 20:52:42
ditto
xunlu
2016/04/18 02:09:13
Done.
|
+ FileSelected(IsEmptyPath(), 1, reinterpret_cast<void*>(1))); |
+ ExportPassword(); |
+} |