first pass of merging in warp (doesn't build)

2026-07-01 16:08:58 -05:00
parent 2f64909469
commit 4770ac06b5
3662 changed files with 414574 additions and 89772 deletions
@@ -0,0 +1,105 @@
+#[cfg(feature = "local_fs")]
+use chrono::{TimeZone, Utc};
+
+use super::*;
+#[cfg(feature = "local_fs")]
+use crate::server::server_api::ai::{
+    ArtifactDownloadCommonFields, FileArtifactResponseData, ScreenshotArtifactResponseData,
+};
+
+#[cfg(feature = "local_fs")]
+fn sample_file_download_response(filename: &str, filepath: &str) -> ArtifactDownloadResponse {
+    ArtifactDownloadResponse::File {
+        common: ArtifactDownloadCommonFields {
+            artifact_uid: "artifact-123".to_string(),
+            created_at: Utc.with_ymd_and_hms(2024, 1, 15, 10, 30, 0).unwrap(),
+        },
+        data: FileArtifactResponseData {
+            download_url: "https://storage.example.com/report.txt".to_string(),
+            expires_at: Utc.with_ymd_and_hms(2024, 1, 15, 11, 30, 0).unwrap(),
+            content_type: "text/plain".to_string(),
+            filepath: filepath.to_string(),
+            filename: filename.to_string(),
+            description: Some("daily summary".to_string()),
+            size_bytes: Some(42),
+        },
+    }
+}
+
+#[test]
+fn sanitized_basename_accepts_plain_filename() {
+    assert_eq!(
+        sanitized_basename("report.txt"),
+        Some("report.txt".to_string())
+    );
+}
+
+#[test]
+fn sanitized_basename_extracts_from_path() {
+    assert_eq!(
+        sanitized_basename("outputs/report.txt"),
+        Some("report.txt".to_string())
+    );
+}
+
+#[test]
+#[cfg(feature = "local_fs")]
+fn extension_for_content_type_recognizes_image_jpg_alias() {
+    assert_eq!(extension_for_content_type("image/jpg"), Some("jpg"));
+    assert_eq!(extension_for_content_type("image/jpeg"), Some("jpg"));
+}
+
+#[test]
+#[cfg(feature = "local_fs")]
+fn default_download_filename_prefers_server_filename() {
+    assert_eq!(
+        default_download_filename(&sample_file_download_response(
+            "report.txt",
+            "outputs/ignored.txt"
+        )),
+        "report.txt"
+    );
+}
+
+#[test]
+#[cfg(feature = "local_fs")]
+fn default_download_filename_uses_content_type_extension_when_filename_missing() {
+    assert_eq!(
+        default_download_filename(&sample_file_download_response("", "outputs/report.txt")),
+        "artifact-artifact-123.txt"
+    );
+}
+
+#[test]
+#[cfg(feature = "local_fs")]
+fn default_download_filename_omits_extension_when_content_type_unknown() {
+    let artifact = ArtifactDownloadResponse::Screenshot {
+        common: ArtifactDownloadCommonFields {
+            artifact_uid: "screenshot-123".to_string(),
+            created_at: Utc.with_ymd_and_hms(2024, 1, 15, 10, 30, 0).unwrap(),
+        },
+        data: ScreenshotArtifactResponseData {
+            download_url: "https://storage.example.com/screenshot".to_string(),
+            expires_at: Utc.with_ymd_and_hms(2024, 1, 15, 11, 30, 0).unwrap(),
+            content_type: "application/octet-stream".to_string(),
+            description: Some("dashboard screenshot".to_string()),
+        },
+    };
+
+    assert_eq!(
+        default_download_filename(&artifact),
+        "artifact-screenshot-123"
+    );
+}
+
+#[test]
+#[cfg(feature = "local_fs")]
+fn download_destination_uses_explicit_path() {
+    assert_eq!(
+        download_destination(
+            &sample_file_download_response("report.txt", "outputs/report.txt"),
+            Some(PathBuf::from("downloads/report.txt"))
+        ),
+        PathBuf::from("downloads/report.txt")
+    );
+}