fix: add agent_server_logs module import to lib.rs

server/packages/extracted-agent-schemas/tests/schema_roundtrip.rs

@@ -0,0 +1,77 @@
+use sandbox_agent_extracted_agent_schemas::{amp, claude, codex};
+
+#[test]
+fn test_claude_bash_input() {
+    let input = claude::BashInput {
+        command: "ls -la".to_string(),
+        timeout: Some(5000.0),
+        working_directory: None,
+    };
+
+    let json = serde_json::to_string(&input).unwrap();
+    assert!(json.contains("ls -la"));
+
+    let parsed: claude::BashInput = serde_json::from_str(&json).unwrap();
+    assert_eq!(parsed.command, "ls -la");
+}
+
+#[test]
+fn test_codex_server_notification() {
+    let notification = codex::ServerNotification::ItemCompleted(
+        codex::ItemCompletedNotification {
+            item: codex::ThreadItem::AgentMessage {
+                id: "msg-123".to_string(),
+                text: "Hello from Codex".to_string(),
+            },
+            thread_id: "thread-123".to_string(),
+            turn_id: "turn-456".to_string(),
+        },
+    );
+
+    let json = serde_json::to_string(&notification).unwrap();
+    assert!(json.contains("item/completed"));
+    assert!(json.contains("Hello from Codex"));
+    assert!(json.contains("agentMessage"));
+}
+
+#[test]
+fn test_codex_thread_item_variants() {
+    let user_msg = codex::ThreadItem::UserMessage {
+        content: vec![codex::UserInput::Text {
+            text: "Hello".to_string(),
+            text_elements: vec![],
+        }],
+        id: "user-1".to_string(),
+    };
+    let json = serde_json::to_string(&user_msg).unwrap();
+    assert!(json.contains("userMessage"));
+    assert!(json.contains("Hello"));
+
+    let cmd = codex::ThreadItem::CommandExecution {
+        aggregated_output: Some("output".to_string()),
+        command: "ls -la".to_string(),
+        command_actions: vec![],
+        cwd: "/tmp".to_string(),
+        duration_ms: Some(100),
+        exit_code: Some(0),
+        id: "cmd-1".to_string(),
+        process_id: None,
+        status: codex::CommandExecutionStatus::Completed,
+    };
+    let json = serde_json::to_string(&cmd).unwrap();
+    assert!(json.contains("commandExecution"));
+    assert!(json.contains("ls -la"));
+}
+
+#[test]
+fn test_amp_message() {
+    let msg = amp::Message {
+        role: amp::MessageRole::User,
+        content: "Hello".to_string(),
+        tool_calls: vec![],
+    };
+
+    let json = serde_json::to_string(&msg).unwrap();
+    assert!(json.contains("user"));
+    assert!(json.contains("Hello"));
+}