Struct MockMemory 

pub struct MockMemory { /* private fields */ }

In-memory mock that stores messages per session.

Each session gets its own Vec<Message>, ensuring test isolation when multiple sessions are used concurrently.

Example

use traitclaw_test_utils::memory::MockMemory;
use traitclaw_core::types::message::Message;
use traitclaw_core::Memory;

let mem = MockMemory::new();
mem.append("s1", Message::user("hi")).await.unwrap();
let msgs = mem.messages("s1").await.unwrap();
assert_eq!(msgs.len(), 1);

Implementations

impl MockMemory

pub fn new() -> Self

Create a new empty mock memory.

Trait Implementations

impl Default for MockMemory

fn default() -> Self

Returns the “default value” for a type.

impl Memory for MockMemory

fn messages<'life0, 'life1, 'async_trait>( &'life0 self, session_id: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<Vec<Message>>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Get conversation messages for a session.

fn append<'life0, 'life1, 'async_trait>( &'life0 self, session_id: &'life1 str, message: Message, ) -> Pin<Box<dyn Future<Output = Result<()>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Append a message to the conversation history.

fn get_context<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, _session_id: &'life1 str, _key: &'life2 str, ) -> Pin<Box<dyn Future<Output = Result<Option<Value>>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait,

Get a value from working memory.

fn set_context<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, _session_id: &'life1 str, _key: &'life2 str, _value: Value, ) -> Pin<Box<dyn Future<Output = Result<()>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait,

Set a value in working memory.

fn recall<'life0, 'life1, 'async_trait>( &'life0 self, _query: &'life1 str, _limit: usize, ) -> Pin<Box<dyn Future<Output = Result<Vec<MemoryEntry>>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Search for relevant memories.

fn store<'life0, 'async_trait>( &'life0 self, _entry: MemoryEntry, ) -> Pin<Box<dyn Future<Output = Result<()>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Store a new memory entry.

fn create_session<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<String, Error>> + Send + 'async_trait>>

where 'life0: 'async_trait, Self: 'async_trait,

Create a new session and return its ID.

fn list_sessions<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<Vec<String>, Error>> + Send + 'async_trait>>

where 'life0: 'async_trait, Self: 'async_trait,

List all known session IDs.

fn delete_session<'life0, 'life1, 'async_trait>( &'life0 self, _session_id: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, Self: 'async_trait,

Delete a session and all associated conversation history and working memory.