| OLD | NEW |
| 1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2009 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 <sys/stat.h> | 5 #include <sys/stat.h> |
| 6 #include <sys/types.h> | 6 #include <sys/types.h> |
| 7 #include <unistd.h> | 7 #include <unistd.h> |
| 8 #include <string> | 8 #include <string> |
| 9 #include <vector> | 9 #include <vector> |
| 10 #include <gtest/gtest.h> | 10 #include <gtest/gtest.h> |
| 11 #include "update_engine/set_bootable_flag_action.h" | 11 #include "update_engine/set_bootable_flag_action.h" |
| 12 #include "update_engine/test_utils.h" | 12 #include "update_engine/test_utils.h" |
| 13 #include "update_engine/utils.h" | 13 #include "update_engine/utils.h" |
| 14 | 14 |
| 15 using std::string; | 15 using std::string; |
| 16 using std::vector; | 16 using std::vector; |
| 17 | 17 |
| 18 namespace chromeos_update_engine { | 18 namespace chromeos_update_engine { |
| 19 | 19 |
| 20 class SetBootableFlagActionProcessorDelegate : public ActionProcessorDelegate { | 20 class SetBootableFlagActionTest : public ::testing::Test {}; |
| 21 public: | |
| 22 SetBootableFlagActionProcessorDelegate() | |
| 23 : success_(false), success_set_(false) {} | |
| 24 void ActionCompleted(ActionProcessor* processor, | |
| 25 AbstractAction* action, | |
| 26 bool success) { | |
| 27 if (action->Type() == SetBootableFlagAction::StaticType()) { | |
| 28 success_ = success; | |
| 29 success_set_ = true; | |
| 30 } | |
| 31 } | |
| 32 bool success_; | |
| 33 bool success_set_; | |
| 34 }; | |
| 35 | 21 |
| 36 class SetBootableFlagActionTest : public ::testing::Test { | 22 // These disabled tests are a reminder this needs to change. |
| 37 public: | 23 TEST_F(SetBootableFlagActionTest, DISABLED_SimpleTest) { |
| 38 SetBootableFlagActionTest(); | 24 // TODO(adlr): find a way to test this object. |
| 39 protected: | |
| 40 enum TestFlags { | |
| 41 EXPECT_SUCCESS = 0x01, | |
| 42 WRITE_FILE = 0x02, | |
| 43 SKIP_INPUT_OBJECT = 0x04 | |
| 44 }; | |
| 45 static const char* kTestDir; | |
| 46 virtual void SetUp() { | |
| 47 EXPECT_EQ(0, mkdir(kTestDir, 0755)); | |
| 48 } | |
| 49 virtual void TearDown() { | |
| 50 EXPECT_EQ(0, System(string("rm -rf ") + kTestDir)); | |
| 51 } | |
| 52 vector<char> DoTest(vector<char> mbr_in, | |
| 53 const string& filename, | |
| 54 uint32_t test_flags); | |
| 55 // first partition bootable, no others bootable | |
| 56 const vector<char> sample_mbr_; | |
| 57 }; | |
| 58 | |
| 59 const char* SetBootableFlagActionTest::kTestDir = | |
| 60 "SetBootableFlagActionTestDir"; | |
| 61 | |
| 62 SetBootableFlagActionTest::SetBootableFlagActionTest() | |
| 63 : sample_mbr_(GenerateSampleMbr()) {} | |
| 64 | |
| 65 vector<char> SetBootableFlagActionTest::DoTest(vector<char> mbr_in, | |
| 66 const string& filename, | |
| 67 uint32_t flags) { | |
| 68 CHECK(!filename.empty()); | |
| 69 const string root_filename(filename.begin(), --filename.end()); | |
| 70 if (flags & WRITE_FILE) | |
| 71 EXPECT_TRUE(WriteFileVector(root_filename, mbr_in)); | |
| 72 | |
| 73 ActionProcessor processor; | |
| 74 SetBootableFlagActionProcessorDelegate delegate; | |
| 75 processor.set_delegate(&delegate); | |
| 76 | |
| 77 ObjectFeederAction<string> feeder_action; | |
| 78 feeder_action.set_obj(filename); | |
| 79 SetBootableFlagAction set_bootable_action; | |
| 80 if (!(flags & SKIP_INPUT_OBJECT)) | |
| 81 BondActions(&feeder_action, &set_bootable_action); | |
| 82 ObjectCollectorAction<string> collector_action; | |
| 83 BondActions(&set_bootable_action, &collector_action); | |
| 84 processor.EnqueueAction(&feeder_action); | |
| 85 processor.EnqueueAction(&set_bootable_action); | |
| 86 processor.EnqueueAction(&collector_action); | |
| 87 processor.StartProcessing(); | |
| 88 EXPECT_TRUE(!processor.IsRunning()) | |
| 89 << "Update test to handle non-asynch actions"; | |
| 90 | |
| 91 EXPECT_TRUE(delegate.success_set_); | |
| 92 EXPECT_EQ(flags & EXPECT_SUCCESS, delegate.success_); | |
| 93 | |
| 94 vector<char> new_mbr; | |
| 95 if (flags & WRITE_FILE) | |
| 96 utils::ReadFile(root_filename, &new_mbr); | |
| 97 | |
| 98 unlink(root_filename.c_str()); | |
| 99 return new_mbr; | |
| 100 } | 25 } |
| 101 | 26 |
| 102 TEST_F(SetBootableFlagActionTest, SimpleTest) { | 27 TEST_F(SetBootableFlagActionTest, DISABLED_BadDeviceTest) { |
| 103 for (int i = 0; i < 4; i++) { | |
| 104 vector<char> expected_new_mbr = sample_mbr_; | |
| 105 for (int j = 0; j < 4; j++) | |
| 106 expected_new_mbr[446 + 16 * j] = '\0'; // mark non-bootable | |
| 107 expected_new_mbr[446 + 16 * i] = 0x80; // mark bootable | |
| 108 | |
| 109 string filename(string(kTestDir) + "/SetBootableFlagActionTest.devX"); | |
| 110 filename[filename.size() - 1] = '1' + i; | |
| 111 | |
| 112 vector<char> actual_new_mbr = DoTest(expected_new_mbr, filename, | |
| 113 EXPECT_SUCCESS | WRITE_FILE); | |
| 114 | |
| 115 ExpectVectorsEq(expected_new_mbr, actual_new_mbr); | |
| 116 } | |
| 117 } | 28 } |
| 118 | 29 |
| 119 TEST_F(SetBootableFlagActionTest, BadDeviceTest) { | 30 TEST_F(SetBootableFlagActionTest, DISABLED_NoInputObjectTest) { |
| 120 vector<char> actual_new_mbr = DoTest(sample_mbr_, | |
| 121 string(kTestDir) + | |
| 122 "SetBootableFlagActionTest.dev5", | |
| 123 WRITE_FILE); | |
| 124 ExpectVectorsEq(sample_mbr_, actual_new_mbr); // no change | |
| 125 | |
| 126 actual_new_mbr = DoTest(sample_mbr_, | |
| 127 string(kTestDir) + "SetBootableFlagActionTest.dev13", | |
| 128 WRITE_FILE); | |
| 129 ExpectVectorsEq(sample_mbr_, actual_new_mbr); // no change | |
| 130 | |
| 131 actual_new_mbr = DoTest(sample_mbr_, | |
| 132 "/some/nonexistent/file3", | |
| 133 0); | |
| 134 EXPECT_TRUE(actual_new_mbr.empty()); | |
| 135 | |
| 136 vector<char> bad_mbr = sample_mbr_; | |
| 137 bad_mbr[510] = 'x'; // makes signature invalid | |
| 138 | |
| 139 actual_new_mbr = DoTest(bad_mbr, | |
| 140 string(kTestDir) + "SetBootableFlagActionTest.dev2", | |
| 141 WRITE_FILE); | |
| 142 ExpectVectorsEq(bad_mbr, actual_new_mbr); // no change | |
| 143 } | |
| 144 | |
| 145 TEST_F(SetBootableFlagActionTest, NoInputObjectTest) { | |
| 146 vector<char> actual_new_mbr = DoTest(sample_mbr_, | |
| 147 string(kTestDir) + | |
| 148 "SetBootableFlagActionTest.dev5", | |
| 149 WRITE_FILE | SKIP_INPUT_OBJECT); | |
| 150 ExpectVectorsEq(sample_mbr_, actual_new_mbr); // no change | |
| 151 } | 31 } |
| 152 | 32 |
| 153 } // namespace chromeos_update_engine | 33 } // namespace chromeos_update_engine |
| OLD | NEW |