v0.3.13-pre.008

crates/ksp-worker-raw-transaction-ingest-lib/unit_tests/runtime_resources.rs

@@ -1,5 +1,5 @@
 // file: crates/ksp-worker-raw-transaction-ingest-lib/unit_tests/runtime_resources.rs
-// version: 19
+// version: 20
 
 fn grpc_endpoint(cluster: &str) -> std::option::Option<ksp_onchain_transport_lib::YellowstoneGrpcEndpointSettings> {
     return grpc_endpoint_with_identity(cluster, "yellowstone-fixture", "fixture-provider");
@@ -2992,3 +2992,45 @@ fn v0_3_13_pre_006_runtime_resources_reject_duplicate_http_polling_identity_even
     assert_eq!(resources.source_count(), 1);
     return;
 }
+
+#[test]
+fn v0_3_13_pre_008_global_hydration_registry_coalesces_cross_source_key_to_one_leader() {
+    let settings = match pre_004_settings() {
+        std::option::Option::Some(value) => value,
+        std::option::Option::None => return,
+    };
+    let network = match ksp_store_lib::RawNetworkId::new("devnet") {
+        std::result::Result::Ok(value) => value,
+        std::result::Result::Err(_) => return,
+    };
+    let key = super::RawTransactionIngestHydrationKey {
+        commitment: ksp_onchain_transport_lib::SolanaCommitment::Confirmed.as_str(),
+        network,
+        signature: ksp_store_lib::RawTransactionSignature::new([61_u8; 64]),
+    };
+    let registry = super::RawTransactionIngestGlobalHydrationRegistry::new(&settings);
+    let (first_leader, _first_receiver) = match registry.subscribe_or_lead(&key) {
+        std::result::Result::Ok(value) => value,
+        std::result::Result::Err(_) => return,
+    };
+    let (second_leader, second_receiver) = match registry.subscribe_or_lead(&key) {
+        std::result::Result::Ok(value) => value,
+        std::result::Result::Err(_) => return,
+    };
+    assert!(first_leader);
+    assert!(!second_leader);
+    registry.publish_and_remove(&key, super::RawTransactionIngestSharedHydrationResult::Failed(crate::ERROR_CODE_RAW_TRANSACTION_INGEST_SOURCE_FAILED));
+    let shared = second_receiver.borrow().clone();
+    match shared {
+        std::option::Option::Some(super::RawTransactionIngestSharedHydrationResult::Failed(code)) => {
+            assert_eq!(code, crate::ERROR_CODE_RAW_TRANSACTION_INGEST_SOURCE_FAILED);
+        },
+        _ => return,
+    }
+    let (third_leader, _third_receiver) = match registry.subscribe_or_lead(&key) {
+        std::result::Result::Ok(value) => value,
+        std::result::Result::Err(_) => return,
+    };
+    assert!(third_leader);
+    return;
+}