theater 0.3.14

A WebAssembly actor system for AI agents
Documentation 
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95

use std::time::Duration;

use theater::{
    ActorHandle, ActorOperation, StartActorResult, ActorStore, HandlerConfig, InterfacesConfig, ManifestConfig, MessageServerConfig,
    TheaterId, ActorMetrics, OperationMetrics, ResourceMetrics, ShutdownController
};
use theater::messages::{ActorMessage, ActorSend, TheaterCommand};
use tokio::sync::mpsc;

/// Helper to create a test manifest
fn create_test_manifest() -> ManifestConfig {
    ManifestConfig {
        name: "test-actor".to_string(),
        version: "1.0.0".to_string(),
        component: "test_component.wasm".to_string(),
        description: None,
        long_description: None,
        save_chain: None,
        permission_policy: Default::default(),
        init_state: None,
        handlers: vec![HandlerConfig::MessageServer(MessageServerConfig {})],
    }
}

/// Helper to create test metrics
fn create_test_metrics() -> ActorMetrics {
    ActorMetrics {
        operation_metrics: OperationMetrics {
            total_operations: 10,
            failed_operations: 0,
            total_processing_time: Duration::from_millis(1000),
            max_processing_time: Duration::from_millis(200),
            min_processing_time: Some(Duration::from_millis(10)),
        },
        resource_metrics: ResourceMetrics {
            memory_usage: 1024,
            operation_queue_size: 5,
            peak_memory_usage: 2048,
            peak_queue_size: 10,
        },
        last_update: Some(std::time::SystemTime::now()),
        uptime_secs: 60,
        start_time: std::time::SystemTime::now(),
    }
}

/// A basic test setup for ActorRuntime
/// Note: This test is marked as ignored because it requires more complex mocking of the
/// ActorComponent creation process which we'll implement in the future
#[tokio::test]
#[ignore]
async fn test_actor_runtime_basic() {
    // Create basic test components
    let actor_id = TheaterId::generate();
    let _config = create_test_manifest();

    let (theater_tx, mut theater_rx) = mpsc::channel(10);
    let (actor_tx, _actor_rx) = mpsc::channel::<ActorMessage>(10);
    let (op_tx, _op_rx) = mpsc::channel::<ActorOperation>(10);
    let mut shutdown_controller = ShutdownController::new();
    let _shutdown_receiver = shutdown_controller.subscribe();
    let (_result_tx, _result_rx) = mpsc::channel::<StartActorResult>(1);

    // Set up a monitor for TheaterCommands
    tokio::spawn(async move {
        while let Some(cmd) = theater_rx.recv().await {
            match cmd {
                TheaterCommand::NewEvent { actor_id, event } => {
                    println!("New event from actor {}: {:?}", actor_id, event);
                }
                _ => println!("Other theater command: {:?}", cmd),
            }
        }
    });

    // TODO: Implement actual test with ActorRuntime
    // This would require more complex mocking of the component creation process

    // For now, just verify the basics
    let actor_handle = ActorHandle::new(op_tx.clone());
    let _actor_store = ActorStore::new(actor_id.clone(), theater_tx.clone(), actor_handle.clone());

    // Send a test message
    let test_message = ActorMessage::Send(ActorSend {
        data: b"test message".to_vec(),
    });

    actor_tx.send(test_message).await.unwrap();

    // Signal shutdown
    shutdown_controller.signal_shutdown();

    // Wait a bit for shutdown to propagate
    tokio::time::sleep(Duration::from_millis(100)).await;
}