Merge "test_v3: write multiple ops" into main am: 03b2213e38

Original change: https://android-review.googlesource.com/c/platform/system/core/+/2816448

Change-Id: I0faf42a9092d9a7642fd9174d4cc440e53ade4d2
Signed-off-by: Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
This commit is contained in:
Daniel Zheng 2023-11-07 06:01:26 +00:00 committed by Automerger Merge Worker
commit e42383472a

View file

@ -339,5 +339,87 @@ TEST_F(CowTestV3, ConsecutiveXorOp) {
ASSERT_EQ(i, 5);
}
TEST_F(CowTestV3, AllOps) {
CowOptions options;
options.op_count_max = 100;
auto writer = CreateCowWriter(3, options, GetCowFd());
std::string data;
data.resize(options.block_size * 5);
for (int i = 0; i < data.size(); i++) {
data[i] = char(rand() % 256);
}
ASSERT_TRUE(writer->AddZeroBlocks(10, 5));
ASSERT_TRUE(writer->AddCopy(15, 3, 5));
ASSERT_TRUE(writer->AddRawBlocks(18, data.data(), data.size()));
ASSERT_TRUE(writer->AddXorBlocks(50, data.data(), data.size(), 24, 10));
ASSERT_TRUE(writer->Finalize());
CowReader reader;
ASSERT_TRUE(reader.Parse(cow_->fd));
const auto& header = reader.header_v3();
ASSERT_EQ(header.prefix.magic, kCowMagicNumber);
ASSERT_EQ(header.prefix.major_version, 3);
ASSERT_EQ(header.prefix.minor_version, kCowVersionMinor);
ASSERT_EQ(header.block_size, options.block_size);
ASSERT_EQ(header.buffer_size, BUFFER_REGION_DEFAULT_SIZE);
ASSERT_EQ(header.op_count, 20);
ASSERT_EQ(header.op_count_max, 100);
auto iter = reader.GetOpIter();
ASSERT_NE(iter, nullptr);
ASSERT_FALSE(iter->AtEnd());
size_t i = 0;
while (i < 5) {
auto op = iter->Get();
ASSERT_EQ(op->type, kCowZeroOp);
ASSERT_EQ(op->new_block, 10 + i);
iter->Next();
i++;
}
i = 0;
while (i < 5) {
auto op = iter->Get();
ASSERT_EQ(op->type, kCowCopyOp);
ASSERT_EQ(op->new_block, 15 + i);
ASSERT_EQ(GetCowOpSourceInfoData(*op), 3 + i);
iter->Next();
i++;
}
i = 0;
std::string sink(data.size(), '\0');
while (i < 5) {
auto op = iter->Get();
ASSERT_EQ(op->type, kCowReplaceOp);
ASSERT_EQ(op->data_length, options.block_size);
ASSERT_EQ(op->new_block, 18 + i);
ASSERT_TRUE(
ReadData(reader, op, sink.data() + (i * options.block_size), options.block_size));
iter->Next();
i++;
}
ASSERT_EQ(sink, data);
i = 0;
std::fill(sink.begin(), sink.end(), '\0');
while (i < 5) {
auto op = iter->Get();
ASSERT_EQ(op->type, kCowXorOp);
ASSERT_EQ(op->data_length, 4096);
ASSERT_EQ(op->new_block, 50 + i);
ASSERT_EQ(GetCowOpSourceInfoData(*op), 98314 + (i * options.block_size)); // 4096 * 24 + 10
ASSERT_TRUE(
ReadData(reader, op, sink.data() + (i * options.block_size), options.block_size));
iter->Next();
i++;
}
ASSERT_EQ(sink, data);
}
} // namespace snapshot
} // namespace android