Initial public release of Warp.

Repo-Sync-Origin: warpdotdev/warp-internal@12af1d983b
2026-04-28 08:43:33 -05:00
commit 0dbd3d567a
4982 changed files with 1431549 additions and 0 deletions
@@ -0,0 +1,89 @@
+use crate::ai::agent::task::TaskId;
+use crate::ai::agent::{
+    AIAgentActionResult, AIAgentActionResultType, TransferShellCommandControlToUserResult,
+};
+use crate::terminal::model::block::BlockId;
+use warp_core::command::ExitCode;
+use warp_multi_agent_api as api;
+
+#[test]
+fn transfer_control_snapshot_result_converts_to_tool_call_result_input() {
+    let block_id = BlockId::default();
+    let input =
+        api::request::input::user_inputs::user_input::Input::try_from(AIAgentActionResult {
+            id: "tool_call".to_string().into(),
+            task_id: TaskId::new("task".to_string()),
+            result: AIAgentActionResultType::TransferShellCommandControlToUser(
+                TransferShellCommandControlToUserResult::Snapshot {
+                    block_id: block_id.clone(),
+                    grid_contents: "snapshot".to_string(),
+                    cursor: "<|cursor|>".to_string(),
+                    is_alt_screen_active: false,
+                    is_preempted: false,
+                },
+            ),
+        })
+        .unwrap();
+
+    match input {
+        api::request::input::user_inputs::user_input::Input::ToolCallResult(result) => {
+            assert_eq!(result.tool_call_id, "tool_call");
+            match result.result {
+                Some(api::request::input::tool_call_result::Result::TransferShellCommandControlToUser(
+                    api_result,
+                )) => match api_result.result {
+                    Some(
+                        api::transfer_shell_command_control_to_user_result::Result::LongRunningCommandSnapshot(snapshot),
+                    ) => {
+                        assert_eq!(snapshot.command_id, block_id.to_string());
+                        assert_eq!(snapshot.output, "snapshot");
+                        assert_eq!(snapshot.cursor, "<|cursor|>");
+                    }
+                    other => panic!("Expected snapshot result, got {other:?}"),
+                },
+                other => panic!("Expected transfer-control tool call result, got {other:?}"),
+            }
+        }
+        other => panic!("Expected tool-call-result input, got {other:?}"),
+    }
+}
+
+#[test]
+fn transfer_control_finished_result_converts_to_tool_call_result_input() {
+    let block_id = BlockId::default();
+    let input =
+        api::request::input::user_inputs::user_input::Input::try_from(AIAgentActionResult {
+            id: "tool_call".to_string().into(),
+            task_id: TaskId::new("task".to_string()),
+            result: AIAgentActionResultType::TransferShellCommandControlToUser(
+                TransferShellCommandControlToUserResult::CommandFinished {
+                    block_id: block_id.clone(),
+                    output: "done".to_string(),
+                    exit_code: ExitCode::from(17),
+                },
+            ),
+        })
+        .unwrap();
+
+    match input {
+        api::request::input::user_inputs::user_input::Input::ToolCallResult(result) => {
+            assert_eq!(result.tool_call_id, "tool_call");
+            match result.result {
+                Some(api::request::input::tool_call_result::Result::TransferShellCommandControlToUser(
+                    api_result,
+                )) => match api_result.result {
+                    Some(
+                        api::transfer_shell_command_control_to_user_result::Result::CommandFinished(finished),
+                    ) => {
+                        assert_eq!(finished.command_id, block_id.to_string());
+                        assert_eq!(finished.output, "done");
+                        assert_eq!(finished.exit_code, 17);
+                    }
+                    other => panic!("Expected command-finished result, got {other:?}"),
+                },
+                other => panic!("Expected transfer-control tool call result, got {other:?}"),
+            }
+        }
+        other => panic!("Expected tool-call-result input, got {other:?}"),
+    }
+}