theater 0.3.5

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

use anyhow::Result;
use serde::{Deserialize, Serialize};
use std::time::Duration;
use theater::actor_executor::ActorOperation;
use theater::actor_handle::ActorHandle;
use theater::actor_runtime::{ActorRuntime, StartActorResult};
use theater::config::{HandlerConfig, ManifestConfig, MessageServerConfig};
use theater::id::TheaterId;
use theater::messages::{ActorMessage, TheaterCommand};
use theater::shutdown::ShutdownController;
use tokio::sync::{mpsc, oneshot};
use tokio::time::timeout;

// Create a basic test manifest config for a simple actor
fn create_test_manifest() -> ManifestConfig {
    let mut config = ManifestConfig {
        name: "test-actor".to_string(),
        package: "test-package-path".to_string(),
        ..Default::default()
    };
    
    // Add a message server handler
    config.handlers.push(HandlerConfig::MessageServer(MessageServerConfig {
        port: None, // Use ephemeral port
    }));
    
    config
}

// This test is more complex and requires mocking WASM components
// For now, we'll define the structure but mark it as ignored
#[tokio::test]
#[ignore]
async fn test_actor_lifecycle() {
    // Create essential channels and IDs
    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(10);
    let (op_tx, op_rx) = mpsc::channel(10);
    let (shutdown_controller, shutdown_receiver) = ShutdownController::new();
    let (result_tx, mut result_rx) = mpsc::channel(1);
    
    // TODO: Properly mock the WASM component initialization
    // This will require significant mocking of the actor component
    
    // Monitor theater commands
    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),
            }
        }
    });
    
    // This part would test the full lifecycle once we have proper mocks
    // 1. Start the actor
    // 2. Send messages to the actor
    // 3. Verify state changes
    // 4. Shutdown the actor
}

// A simpler test that just verifies message routing
#[tokio::test]
async fn test_actor_message_routing() {
    let sender_id = TheaterId::generate();
    let recipient_id = TheaterId::generate();
    
    let (tx, mut rx) = mpsc::channel::<ActorMessage>(10);
    
    // Create and send a test message
    let test_payload = b"test message payload".to_vec();
    let message = ActorMessage {
        sender: sender_id.clone(),
        recipient: recipient_id.clone(),
        payload: test_payload.clone(),
    };
    
    tx.send(message).await.unwrap();
    
    // Receive and verify
    let received = rx.recv().await.unwrap();
    assert_eq!(received.sender, sender_id);
    assert_eq!(received.recipient, recipient_id);
    assert_eq!(received.payload, test_payload);
}