| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/password_manager/chrome_password_manager_client.h" | 5 #include "chrome/browser/password_manager/chrome_password_manager_client.h" |
| 6 | 6 |
| 7 #include "chrome/test/base/chrome_render_view_host_test_harness.h" | 7 #include "chrome/test/base/chrome_render_view_host_test_harness.h" |
| 8 #include "components/autofill/content/common/autofill_messages.h" |
| 8 #include "components/password_manager/core/browser/password_manager_logger.h" | 9 #include "components/password_manager/core/browser/password_manager_logger.h" |
| 9 #include "content/public/browser/browser_context.h" | 10 #include "content/public/browser/browser_context.h" |
| 10 #include "content/public/browser/web_contents.h" | 11 #include "content/public/browser/web_contents.h" |
| 12 #include "content/public/test/mock_render_process_host.h" |
| 11 #include "testing/gmock/include/gmock/gmock.h" | 13 #include "testing/gmock/include/gmock/gmock.h" |
| 12 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
| 13 | 15 |
| 14 using content::BrowserContext; | 16 using content::BrowserContext; |
| 15 using content::WebContents; | 17 using content::WebContents; |
| 16 | 18 |
| 17 namespace { | 19 namespace { |
| 18 | 20 |
| 19 const char kTestText[] = "abcd1234"; | 21 const char kTestText[] = "abcd1234"; |
| 20 | 22 |
| 21 class MockPasswordManagerLogger | 23 class MockPasswordManagerLogger |
| 22 : public password_manager::PasswordManagerLogger { | 24 : public password_manager::PasswordManagerLogger { |
| 23 public: | 25 public: |
| 24 MockPasswordManagerLogger() {} | 26 MockPasswordManagerLogger() {} |
| 25 | 27 |
| 26 MOCK_METHOD1(LogSavePasswordProgress, void(const std::string&)); | 28 MOCK_METHOD1(LogSavePasswordProgress, void(const std::string&)); |
| 27 }; | 29 }; |
| 28 | 30 |
| 29 } // namespace | 31 } // namespace |
| 30 | 32 |
| 31 class ChromePasswordManagerClientTest : public ChromeRenderViewHostTestHarness { | 33 class ChromePasswordManagerClientTest : public ChromeRenderViewHostTestHarness { |
| 32 public: | 34 public: |
| 33 virtual void SetUp() OVERRIDE; | 35 virtual void SetUp() OVERRIDE; |
| 34 | 36 |
| 35 protected: | 37 protected: |
| 36 ChromePasswordManagerClient* GetClient(); | 38 ChromePasswordManagerClient* GetClient(); |
| 37 | 39 |
| 38 testing::StrictMock<MockPasswordManagerLogger> logger; | 40 // If the test IPC sink contains an AutofillMsg_ChangeLoggingState message, |
| 41 // then copies its argument into |activation_flag| and returns true. Otherwise |
| 42 // returns false. |
| 43 bool WasLoggingActivationMessageSent(bool* activation_flag); |
| 44 |
| 45 testing::StrictMock<MockPasswordManagerLogger> logger_; |
| 39 }; | 46 }; |
| 40 | 47 |
| 41 void ChromePasswordManagerClientTest::SetUp() { | 48 void ChromePasswordManagerClientTest::SetUp() { |
| 42 ChromeRenderViewHostTestHarness::SetUp(); | 49 ChromeRenderViewHostTestHarness::SetUp(); |
| 43 ChromePasswordManagerClient::CreateForWebContents(web_contents()); | 50 ChromePasswordManagerClient::CreateForWebContents(web_contents()); |
| 44 } | 51 } |
| 45 | 52 |
| 46 ChromePasswordManagerClient* ChromePasswordManagerClientTest::GetClient() { | 53 ChromePasswordManagerClient* ChromePasswordManagerClientTest::GetClient() { |
| 47 return ChromePasswordManagerClient::FromWebContents(web_contents()); | 54 return ChromePasswordManagerClient::FromWebContents(web_contents()); |
| 48 } | 55 } |
| 49 | 56 |
| 57 bool ChromePasswordManagerClientTest::WasLoggingActivationMessageSent( |
| 58 bool* activation_flag) { |
| 59 const uint32 kMsgID = AutofillMsg_ChangeLoggingState::ID; |
| 60 const IPC::Message* message = |
| 61 process()->sink().GetFirstMessageMatching(kMsgID); |
| 62 if (!message) |
| 63 return false; |
| 64 Tuple1<bool> param; |
| 65 AutofillMsg_ChangeLoggingState::Read(message, ¶m); |
| 66 *activation_flag = param.a; |
| 67 process()->sink().ClearMessages(); |
| 68 return true; |
| 69 } |
| 70 |
| 50 TEST_F(ChromePasswordManagerClientTest, LogSavePasswordProgressNoLogger) { | 71 TEST_F(ChromePasswordManagerClientTest, LogSavePasswordProgressNoLogger) { |
| 51 ChromePasswordManagerClient* client = GetClient(); | 72 ChromePasswordManagerClient* client = GetClient(); |
| 52 | 73 |
| 53 EXPECT_CALL(logger, LogSavePasswordProgress(kTestText)).Times(0); | 74 EXPECT_CALL(logger_, LogSavePasswordProgress(kTestText)).Times(0); |
| 54 // Before attaching the logger, no text should be passed. | 75 // Before attaching the logger, no text should be passed. |
| 55 client->LogSavePasswordProgress(kTestText); | 76 client->LogSavePasswordProgress(kTestText); |
| 56 EXPECT_FALSE(client->IsLoggingActive()); | 77 EXPECT_FALSE(client->IsLoggingActive()); |
| 57 } | 78 } |
| 58 | 79 |
| 59 TEST_F(ChromePasswordManagerClientTest, LogSavePasswordProgressAttachLogger) { | 80 TEST_F(ChromePasswordManagerClientTest, LogSavePasswordProgressAttachLogger) { |
| 60 ChromePasswordManagerClient* client = GetClient(); | 81 ChromePasswordManagerClient* client = GetClient(); |
| 61 | 82 |
| 62 // After attaching the logger, text should be passed. | 83 // After attaching the logger, text should be passed. |
| 63 client->SetLogger(&logger); | 84 client->SetLogger(&logger_); |
| 64 EXPECT_CALL(logger, LogSavePasswordProgress(kTestText)).Times(1); | 85 EXPECT_CALL(logger_, LogSavePasswordProgress(kTestText)).Times(1); |
| 65 client->LogSavePasswordProgress(kTestText); | 86 client->LogSavePasswordProgress(kTestText); |
| 66 EXPECT_TRUE(client->IsLoggingActive()); | 87 EXPECT_TRUE(client->IsLoggingActive()); |
| 67 } | 88 } |
| 68 | 89 |
| 69 TEST_F(ChromePasswordManagerClientTest, LogSavePasswordProgressDetachLogger) { | 90 TEST_F(ChromePasswordManagerClientTest, LogSavePasswordProgressDetachLogger) { |
| 70 ChromePasswordManagerClient* client = GetClient(); | 91 ChromePasswordManagerClient* client = GetClient(); |
| 71 | 92 |
| 72 client->SetLogger(&logger); | 93 client->SetLogger(&logger_); |
| 73 // After detaching the logger, no text should be passed. | 94 // After detaching the logger, no text should be passed. |
| 74 client->SetLogger(NULL); | 95 client->SetLogger(NULL); |
| 75 EXPECT_CALL(logger, LogSavePasswordProgress(kTestText)).Times(0); | 96 EXPECT_CALL(logger_, LogSavePasswordProgress(kTestText)).Times(0); |
| 76 client->LogSavePasswordProgress(kTestText); | 97 client->LogSavePasswordProgress(kTestText); |
| 77 EXPECT_FALSE(client->IsLoggingActive()); | 98 EXPECT_FALSE(client->IsLoggingActive()); |
| 78 } | 99 } |
| 100 |
| 101 TEST_F(ChromePasswordManagerClientTest, LogSavePasswordProgressNotifyRenderer) { |
| 102 ChromePasswordManagerClient* client = GetClient(); |
| 103 bool logging_active = false; |
| 104 |
| 105 // Initially, the logging should be off, so no IPC messages. |
| 106 EXPECT_FALSE(WasLoggingActivationMessageSent(&logging_active)); |
| 107 |
| 108 client->SetLogger(&logger_); |
| 109 EXPECT_TRUE(WasLoggingActivationMessageSent(&logging_active)); |
| 110 EXPECT_TRUE(logging_active); |
| 111 |
| 112 client->SetLogger(NULL); |
| 113 EXPECT_TRUE(WasLoggingActivationMessageSent(&logging_active)); |
| 114 EXPECT_FALSE(logging_active); |
| 115 } |
| OLD | NEW |