use kitsune_client_instrumented::WtChatter;
use std::{fmt::Debug, sync::Arc};
use wind_tunnel_runner::prelude::UserValuesConstraint;
#[derive(Debug, Default)]
pub struct KitsuneAgentContext {
pub(crate) chatter: Option<Arc<WtChatter>>,
}
impl UserValuesConstraint for KitsuneAgentContext {}
impl KitsuneAgentContext {
pub fn chatter(&self) -> Arc<WtChatter> {
self.chatter.clone().expect(
"chatter is not set, did you forget to call `create_chatter` in your agent setup?",
)
}
pub fn chatter_id(&self) -> String {
self.chatter
.clone()
.expect(
"chatter is not set, did you forget to call `create_chatter` in your agent setup?",
)
.id()
.to_string()
}
}