| // Copyright 2016 The Fuchsia Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "src/ledger/bin/storage/impl/page_db.h" |
| |
| #include <lib/async/cpp/task.h> |
| #include <lib/callback/set_when_called.h> |
| #include <lib/zx/time.h> |
| |
| #include <algorithm> |
| #include <memory> |
| #include <string> |
| #include <utility> |
| #include <vector> |
| |
| #include "gmock/gmock.h" |
| #include "gtest/gtest.h" |
| #include "peridot/lib/scoped_tmpfs/scoped_tmpfs.h" |
| #include "src/ledger/bin/encryption/fake/fake_encryption_service.h" |
| #include "src/ledger/bin/storage/impl/commit_impl.h" |
| #include "src/ledger/bin/storage/impl/commit_random_impl.h" |
| #include "src/ledger/bin/storage/impl/leveldb.h" |
| #include "src/ledger/bin/storage/impl/object_impl.h" |
| #include "src/ledger/bin/storage/impl/page_db_impl.h" |
| #include "src/ledger/bin/storage/impl/page_storage_impl.h" |
| #include "src/ledger/bin/storage/impl/storage_test_utils.h" |
| #include "src/ledger/bin/storage/public/constants.h" |
| #include "src/ledger/bin/storage/public/types.h" |
| #include "src/ledger/bin/testing/test_with_environment.h" |
| #include "src/lib/fxl/macros.h" |
| |
| namespace storage { |
| namespace { |
| |
| using ::coroutine::CoroutineHandler; |
| using ::testing::ElementsAre; |
| using ::testing::IsEmpty; |
| using ::testing::Pair; |
| using ::testing::UnorderedElementsAre; |
| |
| std::unique_ptr<LevelDb> GetLevelDb(async_dispatcher_t* dispatcher, |
| ledger::DetachedPath db_path) { |
| auto db = std::make_unique<LevelDb>(dispatcher, std::move(db_path)); |
| EXPECT_EQ(Status::OK, db->Init()); |
| return db; |
| } |
| |
| class PageDbTest : public ledger::TestWithEnvironment { |
| public: |
| PageDbTest() |
| : encryption_service_(dispatcher()), |
| base_path(tmpfs_.root_fd()), |
| page_storage_(&environment_, &encryption_service_, |
| GetLevelDb(dispatcher(), base_path.SubPath("storage")), |
| "page_id"), |
| page_db_(&environment_, |
| GetLevelDb(dispatcher(), base_path.SubPath("page_db"))) {} |
| |
| ~PageDbTest() override {} |
| |
| // Test: |
| void SetUp() override { |
| Status status; |
| bool called; |
| page_storage_.Init( |
| callback::Capture(callback::SetWhenCalled(&called), &status)); |
| RunLoopUntilIdle(); |
| ASSERT_TRUE(called); |
| ASSERT_EQ(Status::OK, status); |
| } |
| |
| protected: |
| scoped_tmpfs::ScopedTmpFS tmpfs_; |
| encryption::FakeEncryptionService encryption_service_; |
| ledger::DetachedPath base_path; |
| PageStorageImpl page_storage_; |
| PageDbImpl page_db_; |
| |
| FXL_DISALLOW_COPY_AND_ASSIGN(PageDbTest); |
| }; |
| |
| TEST_F(PageDbTest, HeadCommits) { |
| RunInCoroutine([&](CoroutineHandler* handler) { |
| std::vector<std::pair<zx::time_utc, CommitId>> heads; |
| EXPECT_EQ(Status::OK, page_db_.GetHeads(handler, &heads)); |
| EXPECT_TRUE(heads.empty()); |
| |
| CommitId cid = RandomCommitId(environment_.random()); |
| EXPECT_EQ(Status::OK, |
| page_db_.AddHead(handler, cid, |
| environment_.random()->Draw<zx::time_utc>())); |
| EXPECT_EQ(Status::OK, page_db_.GetHeads(handler, &heads)); |
| EXPECT_EQ(1u, heads.size()); |
| EXPECT_EQ(cid, heads[0].second); |
| |
| EXPECT_EQ(Status::OK, page_db_.RemoveHead(handler, cid)); |
| EXPECT_EQ(Status::OK, page_db_.GetHeads(handler, &heads)); |
| EXPECT_TRUE(heads.empty()); |
| }); |
| } |
| |
| TEST_F(PageDbTest, MergeCommits) { |
| RunInCoroutine([&](CoroutineHandler* handler) { |
| CommitId parent1 = RandomCommitId(environment_.random()); |
| CommitId parent2 = RandomCommitId(environment_.random()); |
| CommitId merge1 = RandomCommitId(environment_.random()); |
| CommitId merge2 = RandomCommitId(environment_.random()); |
| std::vector<CommitId> merges; |
| |
| // There are no merges |
| EXPECT_EQ(Status::OK, |
| page_db_.GetMerges(handler, parent1, parent2, &merges)); |
| EXPECT_THAT(merges, IsEmpty()); |
| |
| // Add two merges, check they are returned for both orders of the parents |
| std::unique_ptr<PageDbImpl::Batch> batch; |
| EXPECT_EQ(Status::OK, page_db_.StartBatch(handler, &batch)); |
| EXPECT_EQ(Status::OK, batch->AddMerge(handler, parent1, parent2, merge1)); |
| EXPECT_EQ(Status::OK, batch->AddMerge(handler, parent2, parent1, merge2)); |
| EXPECT_EQ(Status::OK, batch->Execute(handler)); |
| |
| EXPECT_EQ(Status::OK, |
| page_db_.GetMerges(handler, parent1, parent2, &merges)); |
| EXPECT_THAT(merges, UnorderedElementsAre(merge1, merge2)); |
| EXPECT_EQ(Status::OK, |
| page_db_.GetMerges(handler, parent2, parent1, &merges)); |
| EXPECT_THAT(merges, UnorderedElementsAre(merge1, merge2)); |
| }); |
| } |
| |
| TEST_F(PageDbTest, OrderHeadCommitsByTimestampThenId) { |
| RunInCoroutine([&](CoroutineHandler* handler) { |
| // Produce 10 random timestamps and 3 constants. |
| std::vector<zx::time_utc> timestamps(10); |
| std::generate(timestamps.begin(), timestamps.end(), [this] { |
| return environment_.random()->Draw<zx::time_utc>(); |
| }); |
| timestamps.insert(timestamps.end(), |
| {zx::time_utc::infinite_past(), zx::time_utc::infinite(), |
| zx::time_utc()}); |
| |
| // Generate 10 commits per timestamp. |
| std::vector<std::pair<zx::time_utc, CommitId>> commits; |
| for (auto ts : timestamps) { |
| for (size_t i = 0; i < 10; ++i) { |
| CommitId id = RandomCommitId(environment_.random()); |
| commits.emplace_back(ts, id); |
| } |
| } |
| |
| // Insert the commits in random order. |
| auto rng = environment_.random()->NewBitGenerator<uint64_t>(); |
| std::shuffle(commits.begin(), commits.end(), rng); |
| for (auto [ts, id] : commits) { |
| EXPECT_EQ(Status::OK, page_db_.AddHead(handler, id, ts)); |
| } |
| |
| // Check that GetHeads returns sorted commits. |
| std::vector<std::pair<zx::time_utc, CommitId>> heads; |
| EXPECT_EQ(Status::OK, page_db_.GetHeads(handler, &heads)); |
| std::sort(commits.begin(), commits.end()); |
| for (size_t i = 0; i < commits.size(); ++i) { |
| EXPECT_EQ(commits[i].second, heads[i].second); |
| } |
| }); |
| } |
| |
| TEST_F(PageDbTest, Commits) { |
| RunInCoroutine([&](CoroutineHandler* handler) { |
| std::vector<std::unique_ptr<const Commit>> parents; |
| parents.emplace_back( |
| std::make_unique<CommitRandomImpl>(environment_.random())); |
| |
| std::unique_ptr<const Commit> commit = CommitImpl::FromContentAndParents( |
| environment_.clock(), &page_storage_, |
| RandomObjectIdentifier(environment_.random()), std::move(parents)); |
| |
| std::string storage_bytes; |
| EXPECT_EQ(Status::INTERNAL_NOT_FOUND, |
| page_db_.GetCommitStorageBytes(handler, commit->GetId(), |
| &storage_bytes)); |
| |
| EXPECT_EQ(Status::OK, |
| page_db_.AddCommitStorageBytes(handler, commit->GetId(), |
| commit->GetRootIdentifier(), |
| commit->GetStorageBytes())); |
| EXPECT_EQ(Status::OK, page_db_.GetCommitStorageBytes( |
| handler, commit->GetId(), &storage_bytes)); |
| EXPECT_EQ(storage_bytes, commit->GetStorageBytes()); |
| |
| std::vector<CommitId> references; |
| EXPECT_EQ(Status::OK, |
| page_db_.GetInboundCommitReferences( |
| handler, commit->GetRootIdentifier(), &references)); |
| EXPECT_THAT(references, ElementsAre(commit->GetId())); |
| }); |
| } |
| |
| TEST_F(PageDbTest, ObjectStorage) { |
| RunInCoroutine([&](CoroutineHandler* handler) { |
| const ObjectIdentifier object_identifier = |
| RandomObjectIdentifier(environment_.random()); |
| const ObjectIdentifier child_identifier = |
| RandomObjectIdentifier(environment_.random()); |
| const std::string content = RandomString(environment_.random(), 32 * 1024); |
| std::unique_ptr<const Piece> piece; |
| PageDbObjectStatus object_status; |
| |
| EXPECT_EQ(Status::INTERNAL_NOT_FOUND, |
| page_db_.ReadObject(handler, object_identifier, &piece)); |
| ASSERT_EQ(Status::OK, |
| page_db_.WriteObject( |
| handler, |
| DataChunkPiece(object_identifier, |
| DataSource::DataChunk::Create(content)), |
| PageDbObjectStatus::TRANSIENT, |
| {{child_identifier.object_digest(), KeyPriority::LAZY}})); |
| ASSERT_EQ(Status::OK, page_db_.GetObjectStatus(handler, object_identifier, |
| &object_status)); |
| EXPECT_EQ(PageDbObjectStatus::TRANSIENT, object_status); |
| ASSERT_EQ(Status::OK, |
| page_db_.ReadObject(handler, object_identifier, &piece)); |
| EXPECT_EQ(content, piece->GetData()); |
| ObjectReferencesAndPriority references; |
| EXPECT_EQ(Status::OK, page_db_.GetInboundObjectReferences( |
| handler, child_identifier, &references)); |
| EXPECT_THAT(references, ElementsAre(Pair(object_identifier.object_digest(), |
| KeyPriority::LAZY))); |
| // Update the object to LOCAL. The new content and references should be |
| // ignored. |
| const std::string new_content = |
| RandomString(environment_.random(), 32 * 1024); |
| ASSERT_EQ(Status::OK, |
| page_db_.WriteObject( |
| handler, |
| DataChunkPiece(object_identifier, |
| DataSource::DataChunk::Create(new_content)), |
| PageDbObjectStatus::LOCAL, |
| {{child_identifier.object_digest(), KeyPriority::EAGER}})); |
| ASSERT_EQ(Status::OK, page_db_.GetObjectStatus(handler, object_identifier, |
| &object_status)); |
| EXPECT_EQ(PageDbObjectStatus::LOCAL, object_status); |
| EXPECT_EQ(content, piece->GetData()); |
| EXPECT_NE(new_content, piece->GetData()); |
| EXPECT_EQ(Status::OK, page_db_.GetInboundObjectReferences( |
| handler, child_identifier, &references)); |
| EXPECT_THAT(references, ElementsAre(Pair(object_identifier.object_digest(), |
| KeyPriority::LAZY))); |
| }); |
| } |
| |
| TEST_F(PageDbTest, LazyAndEagerReferences) { |
| RunInCoroutine([&](CoroutineHandler* handler) { |
| const auto object_identifier = |
| RandomObjectIdentifier(environment_.random()); |
| const ObjectIdentifier child_identifier = |
| RandomObjectIdentifier(environment_.random()); |
| |
| ASSERT_EQ(Status::OK, |
| page_db_.WriteObject( |
| handler, |
| DataChunkPiece(object_identifier, |
| DataSource::DataChunk::Create("")), |
| PageDbObjectStatus::LOCAL, |
| {{child_identifier.object_digest(), KeyPriority::LAZY}, |
| {child_identifier.object_digest(), KeyPriority::EAGER}})); |
| |
| ObjectReferencesAndPriority references; |
| EXPECT_EQ(Status::OK, page_db_.GetInboundObjectReferences( |
| handler, child_identifier, &references)); |
| EXPECT_THAT( |
| references, |
| UnorderedElementsAre( |
| Pair(object_identifier.object_digest(), KeyPriority::LAZY), |
| Pair(object_identifier.object_digest(), KeyPriority::EAGER))); |
| }); |
| } |
| |
| TEST_F(PageDbTest, UnsyncedCommits) { |
| RunInCoroutine([&](CoroutineHandler* handler) { |
| CommitId commit_id = RandomCommitId(environment_.random()); |
| std::vector<CommitId> commit_ids; |
| EXPECT_EQ(Status::OK, page_db_.GetUnsyncedCommitIds(handler, &commit_ids)); |
| EXPECT_TRUE(commit_ids.empty()); |
| |
| EXPECT_EQ(Status::OK, page_db_.MarkCommitIdUnsynced(handler, commit_id, 0)); |
| EXPECT_EQ(Status::OK, page_db_.GetUnsyncedCommitIds(handler, &commit_ids)); |
| EXPECT_EQ(1u, commit_ids.size()); |
| EXPECT_EQ(commit_id, commit_ids[0]); |
| bool is_synced; |
| EXPECT_EQ(Status::OK, |
| page_db_.IsCommitSynced(handler, commit_id, &is_synced)); |
| EXPECT_FALSE(is_synced); |
| |
| EXPECT_EQ(Status::OK, page_db_.MarkCommitIdSynced(handler, commit_id)); |
| EXPECT_EQ(Status::OK, page_db_.GetUnsyncedCommitIds(handler, &commit_ids)); |
| EXPECT_TRUE(commit_ids.empty()); |
| EXPECT_EQ(Status::OK, |
| page_db_.IsCommitSynced(handler, commit_id, &is_synced)); |
| EXPECT_TRUE(is_synced); |
| }); |
| } |
| |
| TEST_F(PageDbTest, OrderUnsyncedCommitsByTimestamp) { |
| RunInCoroutine([&](CoroutineHandler* handler) { |
| CommitId commit_ids[] = {RandomCommitId(environment_.random()), |
| RandomCommitId(environment_.random()), |
| RandomCommitId(environment_.random())}; |
| // Add three unsynced commits with timestamps 200, 300 and 100. |
| EXPECT_EQ(Status::OK, |
| page_db_.MarkCommitIdUnsynced(handler, commit_ids[0], 200)); |
| EXPECT_EQ(Status::OK, |
| page_db_.MarkCommitIdUnsynced(handler, commit_ids[1], 300)); |
| EXPECT_EQ(Status::OK, |
| page_db_.MarkCommitIdUnsynced(handler, commit_ids[2], 100)); |
| |
| // The result should be ordered by the given timestamps. |
| std::vector<CommitId> found_ids; |
| EXPECT_EQ(Status::OK, page_db_.GetUnsyncedCommitIds(handler, &found_ids)); |
| EXPECT_EQ(3u, found_ids.size()); |
| EXPECT_EQ(found_ids[0], commit_ids[2]); |
| EXPECT_EQ(found_ids[1], commit_ids[0]); |
| EXPECT_EQ(found_ids[2], commit_ids[1]); |
| }); |
| } |
| |
| TEST_F(PageDbTest, UnsyncedPieces) { |
| RunInCoroutine([&](CoroutineHandler* handler) { |
| auto object_identifier = RandomObjectIdentifier(environment_.random()); |
| std::vector<ObjectIdentifier> object_identifiers; |
| EXPECT_EQ(Status::OK, |
| page_db_.GetUnsyncedPieces(handler, &object_identifiers)); |
| EXPECT_TRUE(object_identifiers.empty()); |
| |
| EXPECT_EQ(Status::OK, page_db_.WriteObject( |
| handler, |
| DataChunkPiece(object_identifier, |
| DataSource::DataChunk::Create("")), |
| PageDbObjectStatus::LOCAL, {})); |
| EXPECT_EQ(Status::OK, page_db_.SetObjectStatus(handler, object_identifier, |
| PageDbObjectStatus::LOCAL)); |
| EXPECT_EQ(Status::OK, |
| page_db_.GetUnsyncedPieces(handler, &object_identifiers)); |
| EXPECT_EQ(1u, object_identifiers.size()); |
| EXPECT_EQ(object_identifier, object_identifiers[0]); |
| PageDbObjectStatus object_status; |
| EXPECT_EQ(Status::OK, page_db_.GetObjectStatus(handler, object_identifier, |
| &object_status)); |
| EXPECT_EQ(PageDbObjectStatus::LOCAL, object_status); |
| |
| EXPECT_EQ(Status::OK, page_db_.SetObjectStatus(handler, object_identifier, |
| PageDbObjectStatus::SYNCED)); |
| EXPECT_EQ(Status::OK, |
| page_db_.GetUnsyncedPieces(handler, &object_identifiers)); |
| EXPECT_TRUE(object_identifiers.empty()); |
| EXPECT_EQ(Status::OK, page_db_.GetObjectStatus(handler, object_identifier, |
| &object_status)); |
| EXPECT_EQ(PageDbObjectStatus::SYNCED, object_status); |
| }); |
| } |
| |
| TEST_F(PageDbTest, Batch) { |
| RunInCoroutine([&](CoroutineHandler* handler) { |
| std::unique_ptr<PageDb::Batch> batch; |
| ASSERT_EQ(Status::OK, page_db_.StartBatch(handler, &batch)); |
| ASSERT_TRUE(batch); |
| |
| auto object_identifier = RandomObjectIdentifier(environment_.random()); |
| auto eager_identifier = RandomObjectIdentifier(environment_.random()); |
| auto lazy_identifier = RandomObjectIdentifier(environment_.random()); |
| EXPECT_EQ(Status::OK, |
| batch->WriteObject( |
| handler, |
| DataChunkPiece(object_identifier, |
| DataSource::DataChunk::Create("")), |
| PageDbObjectStatus::LOCAL, |
| {{eager_identifier.object_digest(), KeyPriority::EAGER}, |
| {lazy_identifier.object_digest(), KeyPriority::LAZY}})); |
| |
| // Check that we don't have any unsynced piece nor reference prior to |
| // executing the batch. |
| std::vector<ObjectIdentifier> object_identifiers; |
| EXPECT_EQ(Status::OK, |
| page_db_.GetUnsyncedPieces(handler, &object_identifiers)); |
| EXPECT_THAT(object_identifiers, IsEmpty()); |
| ObjectReferencesAndPriority references; |
| EXPECT_EQ(Status::OK, page_db_.GetInboundObjectReferences( |
| handler, eager_identifier, &references)); |
| EXPECT_THAT(references, IsEmpty()); |
| EXPECT_EQ(Status::OK, page_db_.GetInboundObjectReferences( |
| handler, lazy_identifier, &references)); |
| EXPECT_THAT(references, IsEmpty()); |
| |
| // Execute the batch write. |
| EXPECT_EQ(Status::OK, batch->Execute(handler)); |
| |
| // Check unsynced status of written pieces. |
| EXPECT_EQ(Status::OK, |
| page_db_.GetUnsyncedPieces(handler, &object_identifiers)); |
| EXPECT_THAT(object_identifiers, ElementsAre(object_identifier)); |
| // Check the eager reference. |
| EXPECT_EQ(Status::OK, page_db_.GetInboundObjectReferences( |
| handler, eager_identifier, &references)); |
| EXPECT_THAT(references, ElementsAre(Pair(object_identifier.object_digest(), |
| KeyPriority::EAGER))); |
| // Check the lazy reference. |
| EXPECT_EQ(Status::OK, page_db_.GetInboundObjectReferences( |
| handler, lazy_identifier, &references)); |
| EXPECT_THAT(references, ElementsAre(Pair(object_identifier.object_digest(), |
| KeyPriority::LAZY))); |
| }); |
| } |
| |
| TEST_F(PageDbTest, PageDbObjectStatus) { |
| RunInCoroutine([&](CoroutineHandler* handler) { |
| PageDbObjectStatus initial_statuses[] = {PageDbObjectStatus::TRANSIENT, |
| PageDbObjectStatus::LOCAL, |
| PageDbObjectStatus::SYNCED}; |
| PageDbObjectStatus next_statuses[] = {PageDbObjectStatus::LOCAL, |
| PageDbObjectStatus::SYNCED}; |
| for (auto initial_status : initial_statuses) { |
| for (auto next_status : next_statuses) { |
| auto object_identifier = RandomObjectIdentifier(environment_.random()); |
| PageDbObjectStatus object_status; |
| ASSERT_EQ(Status::OK, page_db_.GetObjectStatus( |
| handler, object_identifier, &object_status)); |
| EXPECT_EQ(PageDbObjectStatus::UNKNOWN, object_status); |
| ASSERT_EQ(Status::OK, |
| page_db_.WriteObject( |
| handler, |
| DataChunkPiece(object_identifier, |
| DataSource::DataChunk::Create("")), |
| initial_status, {})); |
| ASSERT_EQ(Status::OK, page_db_.GetObjectStatus( |
| handler, object_identifier, &object_status)); |
| EXPECT_EQ(initial_status, object_status); |
| ASSERT_EQ(Status::OK, page_db_.SetObjectStatus( |
| handler, object_identifier, next_status)); |
| |
| PageDbObjectStatus expected_status = |
| std::max(initial_status, next_status); |
| ASSERT_EQ(Status::OK, page_db_.GetObjectStatus( |
| handler, object_identifier, &object_status)); |
| EXPECT_EQ(expected_status, object_status); |
| } |
| } |
| }); |
| } |
| |
| TEST_F(PageDbTest, SyncMetadata) { |
| RunInCoroutine([&](CoroutineHandler* handler) { |
| std::vector<std::pair<fxl::StringView, fxl::StringView>> keys_and_values = { |
| {"foo1", "foo2"}, {"bar1", " bar2 "}}; |
| for (const auto& key_and_value : keys_and_values) { |
| auto key = key_and_value.first; |
| auto value = key_and_value.second; |
| std::string returned_value; |
| EXPECT_EQ(Status::INTERNAL_NOT_FOUND, |
| page_db_.GetSyncMetadata(handler, key, &returned_value)); |
| |
| EXPECT_EQ(Status::OK, page_db_.SetSyncMetadata(handler, key, value)); |
| EXPECT_EQ(Status::OK, |
| page_db_.GetSyncMetadata(handler, key, &returned_value)); |
| EXPECT_EQ(value, returned_value); |
| } |
| }); |
| } |
| |
| TEST_F(PageDbTest, PageIsOnline) { |
| RunInCoroutine([&](CoroutineHandler* handler) { |
| bool page_is_online; |
| |
| // Check that the initial state is not online. |
| ASSERT_EQ(Status::OK, page_db_.IsPageOnline(handler, &page_is_online)); |
| EXPECT_FALSE(page_is_online); |
| |
| // Mark page as online and check it was updated. |
| EXPECT_EQ(Status::OK, page_db_.MarkPageOnline(handler)); |
| ASSERT_EQ(Status::OK, page_db_.IsPageOnline(handler, &page_is_online)); |
| EXPECT_TRUE(page_is_online); |
| }); |
| } |
| |
| // This test reproduces the crash of LE-451. The crash is due to a subtle |
| // ordering of coroutine execution that is exactly reproduced here. |
| TEST_F(PageDbTest, LE_451_ReproductionTest) { |
| auto id = RandomObjectIdentifier(environment_.random()); |
| RunInCoroutine([&](CoroutineHandler* handler) { |
| EXPECT_EQ( |
| Status::OK, |
| page_db_.WriteObject( |
| handler, DataChunkPiece(id, DataSource::DataChunk::Create("")), |
| PageDbObjectStatus::LOCAL, {})); |
| }); |
| CoroutineHandler* handler1 = nullptr; |
| CoroutineHandler* handler2 = nullptr; |
| environment_.coroutine_service()->StartCoroutine( |
| [&](CoroutineHandler* handler) { |
| handler1 = handler; |
| std::unique_ptr<PageDb::Batch> batch; |
| EXPECT_EQ(Status::OK, page_db_.StartBatch(handler, &batch)); |
| EXPECT_EQ(Status::OK, batch->SetObjectStatus( |
| handler, id, PageDbObjectStatus::SYNCED)); |
| if (handler->Yield() == coroutine::ContinuationStatus::INTERRUPTED) { |
| return; |
| } |
| EXPECT_EQ(Status::OK, batch->Execute(handler)); |
| handler1 = nullptr; |
| }); |
| environment_.coroutine_service()->StartCoroutine( |
| [&](CoroutineHandler* handler) { |
| handler2 = handler; |
| std::unique_ptr<PageDb::Batch> batch; |
| EXPECT_EQ(Status::OK, page_db_.StartBatch(handler, &batch)); |
| if (handler->Yield() == coroutine::ContinuationStatus::INTERRUPTED) { |
| return; |
| } |
| EXPECT_EQ(Status::OK, batch->SetObjectStatus( |
| handler, id, PageDbObjectStatus::LOCAL)); |
| EXPECT_EQ(Status::OK, batch->Execute(handler)); |
| handler2 = nullptr; |
| }); |
| ASSERT_TRUE(handler1); |
| ASSERT_TRUE(handler2); |
| |
| // Reach the 2 yield points. |
| RunLoopUntilIdle(); |
| |
| // Posting a task at this level ensures that the right interleaving between |
| // reading and writing object status happens. |
| async::PostTask(dispatcher(), |
| [&] { handler1->Resume(coroutine::ContinuationStatus::OK); }); |
| handler2->Resume(coroutine::ContinuationStatus::OK); |
| |
| // Finish the test. |
| RunLoopUntilIdle(); |
| |
| // Ensures both coroutines are terminated. |
| ASSERT_FALSE(handler1); |
| ASSERT_FALSE(handler2); |
| } |
| |
| } // namespace |
| } // namespace storage |