106 lines
3.3 KiB
Rust
106 lines
3.3 KiB
Rust
#[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")
|
|
);
|
|
}
|