Struct UsageTracker 

pub struct UsageTracker { /* private fields */ }

Thread-safe usage tracker for the test runner.

Implementations

impl UsageTracker

pub fn new() -> Self

Creates a new empty usage tracker.

pub fn record_llm_call( &self, endpoint_name: &str, endpoint: &ResolvedEndpoint, input_tokens: u64, output_tokens: u64, )

Records a completed call, adding usage and cost.

Panics

Panics if the mutex is poisoned.

pub fn record_flat_call(&self, endpoint_name: &str, endpoint: &ResolvedEndpoint)

Records a flat-cost call (MCP tool, agent task).

Panics

Panics if the mutex is poisoned.

pub fn current_test_snapshot(&self) -> UsageSnapshot

Reads current usage without locking for the full snapshot.

Panics

Panics if the mutex is poisoned.

pub fn global_snapshot(&self) -> UsageSnapshot

Reads the global aggregated snapshot.

Panics

Panics if the mutex is poisoned.

pub fn per_test_snapshots(&self) -> Vec<(String, UsageSnapshot)>

Reads per-test snapshots.

Panics

Panics if the mutex is poisoned.

pub fn reset_per_test(&self)

Resets the per-test accumulator. Call at the start of each test.

Panics

Panics if the mutex is poisoned.

pub fn commit_test(&self, test_name: &str)

Commits the current test’s usage to the global accumulator and stores it as a per-test snapshot.

Panics

Panics if the mutex is poisoned.

Trait Implementations

impl Default for UsageTracker

fn default() -> Self

Returns the “default value” for a type.