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 <base/file_util.h> | 5 #include <base/file_util.h> |
6 #include <gtest/gtest.h> | 6 #include <gtest/gtest.h> |
7 | 7 |
8 #include "update_engine/action_mock.h" | 8 #include "update_engine/action_mock.h" |
9 #include "update_engine/action_processor_mock.h" | 9 #include "update_engine/action_processor_mock.h" |
10 #include "update_engine/filesystem_copier_action.h" | 10 #include "update_engine/filesystem_copier_action.h" |
11 #include "update_engine/postinstall_runner_action.h" | 11 #include "update_engine/postinstall_runner_action.h" |
12 #include "update_engine/set_bootable_flag_action.h" | 12 #include "update_engine/set_bootable_flag_action.h" |
13 #include "update_engine/update_attempter.h" | 13 #include "update_engine/update_attempter.h" |
14 | 14 |
15 using std::string; | 15 using std::string; |
16 using testing::InSequence; | 16 using testing::InSequence; |
17 using testing::Property; | 17 using testing::Property; |
18 using testing::Return; | 18 using testing::Return; |
19 | 19 |
20 namespace chromeos_update_engine { | 20 namespace chromeos_update_engine { |
21 | 21 |
22 // Test a subclass rather than the main class directly so that we can mock out | 22 // Test a subclass rather than the main class directly so that we can mock out |
23 // methods within the class. There're explicit unit test for the mocked out | 23 // methods within the class. There're explicit unit tests for the mocked out |
24 // methods. | 24 // methods. |
25 class UpdateAttempterUnderTest : public UpdateAttempter { | 25 class UpdateAttempterUnderTest : public UpdateAttempter { |
26 public: | 26 public: |
27 UpdateAttempterUnderTest() | 27 UpdateAttempterUnderTest() |
28 : UpdateAttempter(NULL, NULL) {} | 28 : UpdateAttempter(NULL, NULL) {} |
29 }; | 29 }; |
30 | 30 |
31 class UpdateAttempterTest : public ::testing::Test { | 31 class UpdateAttempterTest : public ::testing::Test { |
32 protected: | 32 protected: |
33 virtual void SetUp() { | 33 virtual void SetUp() { |
(...skipping 11 matching lines...) Expand all Loading... |
45 EXPECT_EQ("0.0.0.0", attempter_.new_version_); | 45 EXPECT_EQ("0.0.0.0", attempter_.new_version_); |
46 EXPECT_EQ(0, attempter_.new_size_); | 46 EXPECT_EQ(0, attempter_.new_size_); |
47 processor_ = new ActionProcessorMock(); | 47 processor_ = new ActionProcessorMock(); |
48 attempter_.processor_.reset(processor_); // Transfers ownership. | 48 attempter_.processor_.reset(processor_); // Transfers ownership. |
49 } | 49 } |
50 | 50 |
51 UpdateAttempterUnderTest attempter_; | 51 UpdateAttempterUnderTest attempter_; |
52 ActionProcessorMock* processor_; | 52 ActionProcessorMock* processor_; |
53 }; | 53 }; |
54 | 54 |
55 TEST_F(UpdateAttempterTest, ConstructWithUpdatedMarkerTest) { | 55 TEST_F(UpdateAttempterTest, RunAsRootConstructWithUpdatedMarkerTest) { |
56 extern const char* kUpdateCompletedMarker; | 56 extern const char* kUpdateCompletedMarker; |
57 const FilePath kMarker(kUpdateCompletedMarker); | 57 const FilePath kMarker(kUpdateCompletedMarker); |
58 EXPECT_EQ(0, file_util::WriteFile(kMarker, "", 0)); | 58 EXPECT_EQ(0, file_util::WriteFile(kMarker, "", 0)); |
59 UpdateAttempterUnderTest attempter; | 59 UpdateAttempterUnderTest attempter; |
60 EXPECT_EQ(UPDATE_STATUS_UPDATED_NEED_REBOOT, attempter.status()); | 60 EXPECT_EQ(UPDATE_STATUS_UPDATED_NEED_REBOOT, attempter.status()); |
61 EXPECT_TRUE(file_util::Delete(kMarker, false)); | 61 EXPECT_TRUE(file_util::Delete(kMarker, false)); |
62 } | 62 } |
63 | 63 |
64 TEST_F(UpdateAttempterTest, GetErrorCodeForActionTest) { | 64 TEST_F(UpdateAttempterTest, GetErrorCodeForActionTest) { |
65 extern ActionExitCode GetErrorCodeForAction(AbstractAction* action, | 65 extern ActionExitCode GetErrorCodeForAction(AbstractAction* action, |
(...skipping 81 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
147 EXPECT_EQ(attempter_.response_handler_action_.get(), | 147 EXPECT_EQ(attempter_.response_handler_action_.get(), |
148 attempter_.actions_[1].get()); | 148 attempter_.actions_[1].get()); |
149 DownloadAction* download_action = | 149 DownloadAction* download_action = |
150 dynamic_cast<DownloadAction*>(attempter_.actions_[5].get()); | 150 dynamic_cast<DownloadAction*>(attempter_.actions_[5].get()); |
151 ASSERT_TRUE(download_action != NULL); | 151 ASSERT_TRUE(download_action != NULL); |
152 EXPECT_EQ(&attempter_, download_action->delegate()); | 152 EXPECT_EQ(&attempter_, download_action->delegate()); |
153 EXPECT_EQ(UPDATE_STATUS_CHECKING_FOR_UPDATE, attempter_.status()); | 153 EXPECT_EQ(UPDATE_STATUS_CHECKING_FOR_UPDATE, attempter_.status()); |
154 } | 154 } |
155 | 155 |
156 } // namespace chromeos_update_engine | 156 } // namespace chromeos_update_engine |
OLD | NEW |