candidates_pending_availability added

qdrvm · Oct 30, 2024 · b62d9a6 · b62d9a6
1 parent 75b7a35
commit b62d9a6
Show file tree

Hide file tree

Showing 5 changed files with 45 additions and 0 deletions.
diff --git a/core/runtime/runtime_api/impl/parachain_host.cpp b/core/runtime/runtime_api/impl/parachain_host.cpp
@@ -149,6 +149,18 @@ namespace kagome::runtime {
     return *ref;
   }
 
+  outcome::result<std::vector<std::optional<CommittedCandidateReceipt>>>
+  ParachainHostImpl::candidates_pending_availability(
+      const primitives::BlockHash &block, ParachainId id) {
+    OUTCOME_TRY(ref,
+                candidates_pending_availability_.call(
+                    *executor_,
+                    block,
+                    "ParachainHost_candidates_pending_availability",
+                    id));
+    return *ref;
+  }
+
   outcome::result<std::vector<CandidateEvent>>
   ParachainHostImpl::candidate_events(const primitives::BlockHash &block) {
     OUTCOME_TRY(ref,
@@ -213,6 +225,7 @@ namespace kagome::runtime {
     availability_cores_.erase(blocks);
     session_index_for_child_.erase(blocks);
     candidate_pending_availability_.erase(blocks);
+    candidates_pending_availability_.erase(blocks);
     candidate_events_.erase(blocks);
     session_info_.erase(blocks);
     dmq_contents_.erase(blocks);

diff --git a/core/runtime/runtime_api/impl/parachain_host.hpp b/core/runtime/runtime_api/impl/parachain_host.hpp
@@ -68,6 +68,10 @@ namespace kagome::runtime {
     candidate_pending_availability(const primitives::BlockHash &block,
                                    ParachainId id) override;
 
+    outcome::result<std::vector<std::optional<CommittedCandidateReceipt>>>
+    candidates_pending_availability(const primitives::BlockHash &block,
+                                    ParachainId id) override;
+
     outcome::result<std::vector<CandidateEvent>> candidate_events(
         const primitives::BlockHash &block) override;
 
@@ -148,6 +152,9 @@ namespace kagome::runtime {
     };
     RuntimeApiLruBlockArg<ParachainId, std::optional<CommittedCandidateReceipt>>
         candidate_pending_availability_{10};
+    RuntimeApiLruBlockArg<ParachainId,
+                          std::vector<std::optional<CommittedCandidateReceipt>>>
+        candidates_pending_availability_{10};
     RuntimeApiLruBlock<std::vector<CandidateEvent>> candidate_events_{10};
     RuntimeApiLruBlockArg<SessionIndex, std::optional<SessionInfo>>
         session_info_{10};

diff --git a/core/runtime/runtime_api/parachain_host.hpp b/core/runtime/runtime_api/parachain_host.hpp
@@ -168,6 +168,16 @@ namespace kagome::runtime {
     candidate_pending_availability(const primitives::BlockHash &block,
                                    ParachainId id) = 0;
 
+    /**
+     * @brief Get the receipts of all candidates pending availability.
+     * @param id parachain id
+     * @return vector of CommittedCandidateReceipt
+     */
+    virtual outcome::result<
+        std::vector<std::optional<CommittedCandidateReceipt>>>
+    candidates_pending_availability(const primitives::BlockHash &block,
+                                    ParachainId id) = 0;
+
     ///
     /**
      * @brief Get a vector of events concerning candidates that occurred within

diff --git a/test/core/runtime/binaryen/parachain_test.cpp b/test/core/runtime/binaryen/parachain_test.cpp
@@ -167,6 +167,16 @@ TEST_F(ParachainHostTest, DISABLED_CandidatePendingAvailabilityTest) {
   ASSERT_TRUE(api_->candidate_pending_availability("block_hash"_hash256, id));
 }
 
+/**
+ * @given initialized parachain host api
+ * @when candidates_pending_availability() is invoked
+ * @then successful result is returned
+ */
+TEST_F(ParachainHostTest, DISABLED_CandidatesPendingAvailabilityTest) {
+  auto id = createParachainId();
+  ASSERT_TRUE(api_->candidates_pending_availability("block_hash"_hash256, id));
+}
+
 /**
  * @given initialized parachain host api
  * @when candidate_events() is invoked

diff --git a/test/mock/core/runtime/parachain_host_mock.hpp b/test/mock/core/runtime/parachain_host_mock.hpp
@@ -80,6 +80,11 @@ namespace kagome::runtime {
                 (const primitives::BlockHash &, ParachainId),
                 (override));
 
+    MOCK_METHOD(outcome::result<std::vector<std::optional<CommittedCandidateReceipt>>>,
+                candidates_pending_availability,
+                (const primitives::BlockHash &, ParachainId),
+                (override));
+
     MOCK_METHOD(outcome::result<std::vector<CandidateEvent>>,
                 candidate_events,
                 (const primitives::BlockHash &),