v0.3.15-pre.009

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: 35
+// version: 36
 
 fn grpc_endpoint(cluster: &str) -> std::option::Option<ksp_onchain_transport_lib::YellowstoneGrpcEndpointSettings> {
     return grpc_endpoint_with_identity(cluster, "yellowstone-fixture", "fixture-provider");
@@ -359,6 +359,25 @@ fn transaction_request(
     return std::option::Option::Some(request);
 }
 
+fn block_request(
+    commitment: std::option::Option<ksp_onchain_transport_lib::SolanaCommitment>,
+) -> std::option::Option<ksp_onchain_transport_lib::YellowstoneSubscribeRequest> {
+    let name = match ksp_onchain_transport_lib::YellowstoneSubscribeFilterName::new("block-fixture") {
+        std::result::Result::Ok(value) => value,
+        std::result::Result::Err(_) => return std::option::Option::None,
+    };
+    let mut filter = ksp_onchain_transport_lib::YellowstoneSubscribeBlockFilter::new();
+    filter.set_include_transactions(std::option::Option::Some(false));
+    filter.set_include_accounts(std::option::Option::Some(false));
+    filter.set_include_entries(std::option::Option::Some(false));
+    let mut request = ksp_onchain_transport_lib::YellowstoneSubscribeRequest::new();
+    if request.insert_block_filter(name, filter).is_err() {
+        return std::option::Option::None;
+    }
+    request.set_commitment(commitment);
+    return std::option::Option::Some(request);
+}
+
 struct SignalViewFixture {
     created_at: std::option::Option<ksp_onchain_transport_lib::YellowstoneUpdateTimestamp>,
     family: super::RawTransactionIngestSourceFamily,
@@ -836,6 +855,60 @@ async fn pre_002_source_accepts_matching_confirmed_transaction_and_get_transacti
     return;
 }
 
+#[tokio::test(flavor = "current_thread")]
+async fn pre_009_yellowstone_block_mode_uses_get_block_without_transaction_hydration() {
+    let endpoint = match grpc_endpoint("devnet") {
+        std::option::Option::Some(value) => value,
+        std::option::Option::None => return,
+    };
+    let channel = match ksp_onchain_transport_lib::YellowstoneGrpcChannel::prepare(&endpoint) {
+        std::result::Result::Ok(value) => value,
+        std::result::Result::Err(_) => return,
+    };
+    let request = match block_request(std::option::Option::Some(ksp_onchain_transport_lib::SolanaCommitment::Confirmed)) {
+        std::option::Option::Some(value) => value,
+        std::option::Option::None => return,
+    };
+    let pool = match http_pool("devnet", "hydration", "get_block") {
+        std::option::Option::Some(value) => value,
+        std::option::Option::None => return,
+    };
+    let source = crate::RawTransactionIngestYellowstoneSource::new(channel, request, pool, ksp_onchain_transport_lib::HttpRoleName::new("hydration"));
+    let source = match source {
+        std::result::Result::Ok(value) => value,
+        std::result::Result::Err(_) => return,
+    };
+    assert_eq!(source.mode, super::RawTransactionIngestYellowstoneMode::BlockHydration);
+    assert!(!source.uses_transaction_hydration());
+}
+
+#[tokio::test(flavor = "current_thread")]
+async fn pre_009_yellowstone_block_mode_rejects_get_transaction_only_http_role() {
+    let endpoint = match grpc_endpoint("devnet") {
+        std::option::Option::Some(value) => value,
+        std::option::Option::None => return,
+    };
+    let channel = match ksp_onchain_transport_lib::YellowstoneGrpcChannel::prepare(&endpoint) {
+        std::result::Result::Ok(value) => value,
+        std::result::Result::Err(_) => return,
+    };
+    let request = match block_request(std::option::Option::Some(ksp_onchain_transport_lib::SolanaCommitment::Confirmed)) {
+        std::option::Option::Some(value) => value,
+        std::option::Option::None => return,
+    };
+    let pool = match http_pool("devnet", "hydration", "get_transaction") {
+        std::option::Option::Some(value) => value,
+        std::option::Option::None => return,
+    };
+    let source = crate::RawTransactionIngestYellowstoneSource::new(channel, request, pool, ksp_onchain_transport_lib::HttpRoleName::new("hydration"));
+    let error = match source {
+        std::result::Result::Ok(_) => return,
+        std::result::Result::Err(error) => error,
+    };
+    assert_eq!(error.code(), crate::ERROR_CODE_RAW_TRANSACTION_INGEST_RUNTIME_INVALID);
+    assert!(error.context().iter().any(|context| return context.value() == "runtime_resources.hydration_role_unsupported"));
+}
+
 #[tokio::test(flavor = "current_thread")]
 async fn pre_002_source_rejects_processed_and_implicit_commitment() {
     for commitment in [std::option::Option::Some(ksp_onchain_transport_lib::SolanaCommitment::Processed), std::option::Option::None] {