test update

This commit is contained in:
Brandon Thomas committed 2026-04-21 20:15:04 -04:00
1 parent bb7ee7e9bb
commit 580975ae76
2 files changed
+12 -10

No files matched your search

@@ -51,10 +51,11 @@ mod tests {
use artcraft_client::credentials::storyteller_credential_set::StorytellerCredentialSet;
use artcraft_client::utils::api_host::ApiHost;
// Media file tokens from test_data::web::image_urls comments.
const JUNO_AT_LAKE_TOKEN: &str = "m_m1bz02z1kkzanxy6rb4vk1kvq9de9g";
const FOREST_BACKDROP_TOKEN: &str = "m_f5yy0fdmkwrf2hxaq4yw4sd95tvvrf";
const WHITE_HOUSE_SUNSET_TOKEN: &str = "m_xmzdeqhjpxs5418kvepcrj5a3vkfb1";
use test_data::web::image_media_tokens::{
FOREST_BACKDROP_PRODUCTION_MEDIA_TOKEN,
JUNO_AT_LAKE_PRODUCTION_MEDIA_TOKEN,
WHITE_HOUSE_SUNSET_PRODUCTION_MEDIA_TOKEN,
};
// TODO: Replace with real character tokens from the test account.
const STEAMPUNK_CLOWN_CHAR_TOKEN: &str = "character_placeholder_steampunk_clown";
@@ -168,8 +169,8 @@ mod tests {
// Tokens from test_data::web::image_urls comments.
let response = run_pipeline(GenerateVideoRequestBuilder {
prompt: Some("The dog walks from the lake to the forest.".to_string()),
start_frame: Some(ImageRef::MediaFileToken(MediaFileToken::new(JUNO_AT_LAKE_TOKEN.to_string()))),
end_frame: Some(ImageRef::MediaFileToken(MediaFileToken::new(FOREST_BACKDROP_TOKEN.to_string()))),
start_frame: Some(ImageRef::MediaFileToken(MediaFileToken::new(JUNO_AT_LAKE_PRODUCTION_MEDIA_TOKEN.to_string()))),
end_frame: Some(ImageRef::MediaFileToken(MediaFileToken::new(FOREST_BACKDROP_PRODUCTION_MEDIA_TOKEN.to_string()))),
aspect_ratio: Some(CommonAspectRatio::WideSixteenByNine),
..artcraft_builder()
}).await;
@@ -185,9 +186,9 @@ mod tests {
let response = run_pipeline(GenerateVideoRequestBuilder {
prompt: Some("The dog in @2 runs through the scenery in @1 towards the building in @3.".to_string()),
reference_images: Some(ImageListRef::MediaFileTokens(vec![
MediaFileToken::new(FOREST_BACKDROP_TOKEN.to_string()),
MediaFileToken::new(JUNO_AT_LAKE_TOKEN.to_string()),
MediaFileToken::new(WHITE_HOUSE_SUNSET_TOKEN.to_string()),
MediaFileToken::new(FOREST_BACKDROP_PRODUCTION_MEDIA_TOKEN.to_string()),
MediaFileToken::new(JUNO_AT_LAKE_PRODUCTION_MEDIA_TOKEN.to_string()),
MediaFileToken::new(WHITE_HOUSE_SUNSET_PRODUCTION_MEDIA_TOKEN.to_string()),
])),
aspect_ratio: Some(CommonAspectRatio::WideSixteenByNine),
..artcraft_builder()
@@ -203,7 +204,7 @@ mod tests {
let response = run_pipeline(GenerateVideoRequestBuilder {
prompt: Some("Change @video1 to a nighttime scene with moonlight.".to_string()),
reference_videos: Some(VideoListRef::MediaFileTokens(vec![
MediaFileToken::new(JUNO_AT_LAKE_TOKEN.to_string()), // Using an image token as placeholder
MediaFileToken::new(JUNO_AT_LAKE_PRODUCTION_MEDIA_TOKEN.to_string()), // Using an image token as placeholder
])),
aspect_ratio: Some(CommonAspectRatio::WideSixteenByNine),
..artcraft_builder()
+1
View File
@@ -1 +1,2 @@
pub mod image_media_tokens;
pub mod image_urls;