otel-arrow-dfe-engine 0.61.0

Async pipeline engine
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0

//! Test-only reference capabilities.
//!
//! Reference capabilities used by integration tests (and downstream
//! tests via the public `testing` surface) to exercise the
//! extension/capability wiring. Real production capabilities live
//! under [`crate::capability`] and are re-exported through
//! [`local::capability`](crate::local::capability) /
//! [`shared::capability`](crate::shared::capability).

pub mod no_op_stateful;
pub mod no_op_stateless;

#[cfg(any(test, feature = "test-utils"))]
use crate::capability::registry::{
    Capabilities, CapabilityRegistry, ConsumedTracker, resolve_bindings,
};
#[cfg(any(test, feature = "test-utils"))]
use otel_arrow_dfe_config::{CapabilityId, ExtensionId};
#[cfg(any(test, feature = "test-utils"))]
use std::collections::{HashMap, HashSet};

/// Resolves capability bindings for downstream component tests.
///
/// Production code resolves bindings as part of pipeline construction. This
/// helper exposes the same path to tests that need a real [`Capabilities`]
/// instance rather than [`Capabilities::empty`].
#[cfg(any(test, feature = "test-utils"))]
pub fn resolve_bindings_for_test(
    bindings: &HashMap<CapabilityId, ExtensionId>,
    registry: &CapabilityRegistry,
    known_extensions: &HashSet<ExtensionId>,
) -> Result<Capabilities, crate::error::Error> {
    let mut tracker = ConsumedTracker::new();
    resolve_bindings(bindings, registry, known_extensions, &mut tracker)
}