diff options
author | Sen Jiang <senj@google.com> | 2018-07-02 17:34:56 -0700 |
---|---|---|
committer | Sen Jiang <senj@google.com> | 2018-07-03 18:24:09 -0700 |
commit | 0779a151e585ac465de4ea03e710b24f60037724 (patch) | |
tree | 8aee0e5625b8316a95628831f535d272b3f50fee /payload_generator/payload_file_unittest.cc | |
parent | 0a582fbf50cd420198d29ee64505dd6785a68184 (diff) |
Use ScopedTempFile in unit test.
Replace these 3 lines of code:
string path;
ASSERT_TRUE(utils::MakeTempFile("name-XXXXXX", &path, nullptr));
ScopedPathUnlinker path_unlinker(path);
with one liner:
test_utils::ScopedTempFile file("name-XXXXXX");
Bug: None
Test: unit test
Change-Id: Ic5be7dc8339842270023055bcc3a97e526953f04
Diffstat (limited to 'payload_generator/payload_file_unittest.cc')
-rw-r--r-- | payload_generator/payload_file_unittest.cc | 17 |
1 files changed, 5 insertions, 12 deletions
diff --git a/payload_generator/payload_file_unittest.cc b/payload_generator/payload_file_unittest.cc index e8e7e143..45faebb9 100644 --- a/payload_generator/payload_file_unittest.cc +++ b/payload_generator/payload_file_unittest.cc @@ -36,23 +36,16 @@ class PayloadFileTest : public ::testing::Test { }; TEST_F(PayloadFileTest, ReorderBlobsTest) { - string orig_blobs; - EXPECT_TRUE(utils::MakeTempFile("ReorderBlobsTest.orig.XXXXXX", &orig_blobs, - nullptr)); - ScopedPathUnlinker orig_blobs_unlinker(orig_blobs); + test_utils::ScopedTempFile orig_blobs("ReorderBlobsTest.orig.XXXXXX"); // The operations have three blob and one gap (the whitespace): // Rootfs operation 1: [8, 3] bcd // Rootfs operation 2: [7, 1] a // Kernel operation 1: [0, 6] kernel string orig_data = "kernel abcd"; - EXPECT_TRUE( - utils::WriteFile(orig_blobs.c_str(), orig_data.data(), orig_data.size())); + EXPECT_TRUE(test_utils::WriteFileString(orig_blobs.path(), orig_data)); - string new_blobs; - EXPECT_TRUE( - utils::MakeTempFile("ReorderBlobsTest.new.XXXXXX", &new_blobs, nullptr)); - ScopedPathUnlinker new_blobs_unlinker(new_blobs); + test_utils::ScopedTempFile new_blobs("ReorderBlobsTest.new.XXXXXX"); payload_.part_vec_.resize(2); @@ -71,12 +64,12 @@ TEST_F(PayloadFileTest, ReorderBlobsTest) { aop.op.set_data_length(6); payload_.part_vec_[1].aops = {aop}; - EXPECT_TRUE(payload_.ReorderDataBlobs(orig_blobs, new_blobs)); + EXPECT_TRUE(payload_.ReorderDataBlobs(orig_blobs.path(), new_blobs.path())); const vector<AnnotatedOperation>& part0_aops = payload_.part_vec_[0].aops; const vector<AnnotatedOperation>& part1_aops = payload_.part_vec_[1].aops; string new_data; - EXPECT_TRUE(utils::ReadFile(new_blobs, &new_data)); + EXPECT_TRUE(utils::ReadFile(new_blobs.path(), &new_data)); // Kernel blobs should appear at the end. EXPECT_EQ("bcdakernel", new_data); |