Chromium Code Reviews| Index: chrome/browser/extensions/api/system_private/system_private_apitest.cc |
| diff --git a/chrome/browser/extensions/api/system_private/system_private_apitest.cc b/chrome/browser/extensions/api/system_private/system_private_apitest.cc |
| index 725b0504f01e47d9f713a0fc270ca26429865089..6e3d3419bcc2a2d3875589b4111e1cfb3eafdae7 100644 |
| --- a/chrome/browser/extensions/api/system_private/system_private_apitest.cc |
| +++ b/chrome/browser/extensions/api/system_private/system_private_apitest.cc |
| @@ -9,11 +9,10 @@ |
| #include "chrome/common/pref_names.h" |
| #if defined(OS_CHROMEOS) |
| -#include "chromeos/dbus/mock_dbus_thread_manager.h" |
| -#include "chromeos/dbus/mock_update_engine_client.h" |
| +#include "chromeos/dbus/fake_update_engine_client.h" |
| +#include "chromeos/dbus/mock_dbus_thread_manager_without_gmock.h" |
| using chromeos::UpdateEngineClient; |
| -using ::testing::Return; |
| #endif |
| IN_PROC_BROWSER_TEST_F(ExtensionApiTest, GetIncognitoModeAvailability) { |
| @@ -28,18 +27,15 @@ IN_PROC_BROWSER_TEST_F(ExtensionApiTest, GetIncognitoModeAvailability) { |
| class GetUpdateStatusApiTest : public ExtensionApiTest { |
| public: |
| - GetUpdateStatusApiTest() : mock_update_engine_client_(NULL) {} |
| + GetUpdateStatusApiTest() : fake_update_engine_client_(NULL) {} |
| virtual void SetUpInProcessBrowserTestFixture() OVERRIDE { |
| ExtensionApiTest::SetUpInProcessBrowserTestFixture(); |
| - chromeos::MockDBusThreadManager* mock_dbus_thread_manager = |
| - new chromeos::MockDBusThreadManager; |
| + chromeos::MockDBusThreadManagerWithoutGMock* mock_dbus_thread_manager = |
| + new chromeos::MockDBusThreadManagerWithoutGMock; |
| chromeos::DBusThreadManager::InitializeForTesting(mock_dbus_thread_manager); |
| - mock_update_engine_client_ = |
| - mock_dbus_thread_manager->mock_update_engine_client(); |
| - EXPECT_CALL(*mock_update_engine_client_, GetLastStatus()) |
| - .Times(1) |
|
bartfab (slow)
2013/05/14 09:35:06
This used to verify that the method gets called ex
hashimoto
2013/05/14 10:37:21
Do we need to check that GetLastStatus() is called
bartfab (slow)
2013/05/14 10:53:14
Fair enough. Let's leave out the check.
Haruki Sato
2013/05/14 13:20:14
Thank you guys for the review!
|
| - .WillOnce(Return(chromeos::MockUpdateEngineClient::Status())); |
| + fake_update_engine_client_ = |
| + mock_dbus_thread_manager->fake_update_engine_client(); |
| } |
| virtual void TearDownInProcessBrowserTestFixture() OVERRIDE { |
| @@ -48,7 +44,7 @@ class GetUpdateStatusApiTest : public ExtensionApiTest { |
| } |
| protected: |
| - chromeos::MockUpdateEngineClient* mock_update_engine_client_; |
| + chromeos::FakeUpdateEngineClient* fake_update_engine_client_; |
| private: |
| DISALLOW_COPY_AND_ASSIGN(GetUpdateStatusApiTest); |
| @@ -64,10 +60,11 @@ IN_PROC_BROWSER_TEST_F(GetUpdateStatusApiTest, Progress) { |
| status_boot_needed.status = |
| UpdateEngineClient::UPDATE_STATUS_UPDATED_NEED_REBOOT; |
| - EXPECT_CALL(*mock_update_engine_client_, GetLastStatus()) |
| - .WillOnce(Return(status_not_available)) |
| - .WillOnce(Return(status_updating)) |
| - .WillOnce(Return(status_boot_needed)); |
| + // The fake client returns the last status in this order. |
| + fake_update_engine_client_->PushLastStatus(status_not_available); |
| + fake_update_engine_client_->PushLastStatus(status_updating); |
| + fake_update_engine_client_->PushLastStatus(status_boot_needed); |
| + |
| ASSERT_TRUE(RunComponentExtensionTest( |
| "system/get_update_status")) << message_; |
| } |