| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium OS Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium OS 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 <string> | 5 #include <string> |
| 6 | 6 |
| 7 #include <gtest/gtest.h> | 7 #include <gtest/gtest.h> |
| 8 | 8 |
| 9 #include "update_engine/omaha_response_handler_action.h" | 9 #include "update_engine/omaha_response_handler_action.h" |
| 10 #include "update_engine/prefs_mock.h" | 10 #include "update_engine/prefs_mock.h" |
| 11 #include "update_engine/test_utils.h" | 11 #include "update_engine/test_utils.h" |
| 12 #include "update_engine/utils.h" | 12 #include "update_engine/utils.h" |
| 13 | 13 |
| 14 using std::string; | 14 using std::string; |
| 15 using testing::NiceMock; |
| 15 using testing::Return; | 16 using testing::Return; |
| 16 | 17 |
| 17 namespace chromeos_update_engine { | 18 namespace chromeos_update_engine { |
| 18 | 19 |
| 19 class OmahaResponseHandlerActionTest : public ::testing::Test { | 20 class OmahaResponseHandlerActionTest : public ::testing::Test { |
| 20 public: | 21 public: |
| 21 // Return true iff the OmahaResponseHandlerAction succeeded. | 22 // Return true iff the OmahaResponseHandlerAction succeeded. |
| 22 // If out is non-NULL, it's set w/ the response from the action. | 23 // If out is non-NULL, it's set w/ the response from the action. |
| 23 bool DoTest(const OmahaResponse& in, | 24 bool DoTest(const OmahaResponse& in, |
| 24 const string& boot_dev, | 25 const string& boot_dev, |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 57 | 58 |
| 58 bool OmahaResponseHandlerActionTest::DoTest(const OmahaResponse& in, | 59 bool OmahaResponseHandlerActionTest::DoTest(const OmahaResponse& in, |
| 59 const string& boot_dev, | 60 const string& boot_dev, |
| 60 InstallPlan* out) { | 61 InstallPlan* out) { |
| 61 ActionProcessor processor; | 62 ActionProcessor processor; |
| 62 OmahaResponseHandlerActionProcessorDelegate delegate; | 63 OmahaResponseHandlerActionProcessorDelegate delegate; |
| 63 processor.set_delegate(&delegate); | 64 processor.set_delegate(&delegate); |
| 64 | 65 |
| 65 ObjectFeederAction<OmahaResponse> feeder_action; | 66 ObjectFeederAction<OmahaResponse> feeder_action; |
| 66 feeder_action.set_obj(in); | 67 feeder_action.set_obj(in); |
| 67 PrefsMock prefs; | 68 NiceMock<PrefsMock> prefs; |
| 68 if (in.update_exists) { | 69 if (in.update_exists) { |
| 69 EXPECT_CALL(prefs, SetString(kPrefsUpdateCheckResponseHash, in.hash)) | 70 EXPECT_CALL(prefs, SetString(kPrefsUpdateCheckResponseHash, in.hash)) |
| 70 .WillOnce(Return(true)); | 71 .WillOnce(Return(true)); |
| 71 } | 72 } |
| 72 OmahaResponseHandlerAction response_handler_action(&prefs); | 73 OmahaResponseHandlerAction response_handler_action(&prefs); |
| 73 response_handler_action.set_boot_device(boot_dev); | 74 response_handler_action.set_boot_device(boot_dev); |
| 74 BondActions(&feeder_action, &response_handler_action); | 75 BondActions(&feeder_action, &response_handler_action); |
| 75 ObjectCollectorAction<InstallPlan> collector_action; | 76 ObjectCollectorAction<InstallPlan> collector_action; |
| 76 BondActions(&response_handler_action, &collector_action); | 77 BondActions(&response_handler_action, &collector_action); |
| 77 processor.EnqueueAction(&feeder_action); | 78 processor.EnqueueAction(&feeder_action); |
| (...skipping 93 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 171 in.update_exists = false; | 172 in.update_exists = false; |
| 172 InstallPlan install_plan; | 173 InstallPlan install_plan; |
| 173 EXPECT_FALSE(DoTest(in, "/dev/sda1", &install_plan)); | 174 EXPECT_FALSE(DoTest(in, "/dev/sda1", &install_plan)); |
| 174 EXPECT_FALSE(install_plan.is_full_update); | 175 EXPECT_FALSE(install_plan.is_full_update); |
| 175 EXPECT_EQ("", install_plan.download_url); | 176 EXPECT_EQ("", install_plan.download_url); |
| 176 EXPECT_EQ("", install_plan.download_hash); | 177 EXPECT_EQ("", install_plan.download_hash); |
| 177 EXPECT_EQ("", install_plan.install_path); | 178 EXPECT_EQ("", install_plan.install_path); |
| 178 } | 179 } |
| 179 | 180 |
| 180 } // namespace chromeos_update_engine | 181 } // namespace chromeos_update_engine |
| OLD | NEW |