104ebb410a
Roaster creation (basic + all fields), full brew chain through browser, entity deletion with confirm dialog, AI extraction via text prompt with mocked OpenRouter, and check-in wizard with saved cafe + existing roast.
111 lines
3.6 KiB
Rust
111 lines
3.6 KiB
Rust
use thirtyfour::prelude::*;
|
|
use wiremock::matchers::{method, path};
|
|
use wiremock::{Mock, ResponseTemplate};
|
|
|
|
use crate::helpers::auth::authenticate_browser;
|
|
use crate::helpers::browser::BrowserSession;
|
|
use crate::helpers::server_helpers::spawn_app_with_openrouter_mock;
|
|
use crate::helpers::wait::{wait_for_url_contains, wait_for_visible};
|
|
|
|
fn mock_openrouter_response(json_content: &str) -> ResponseTemplate {
|
|
let body = serde_json::json!({
|
|
"id": "gen-test",
|
|
"model": "test-model",
|
|
"choices": [{
|
|
"index": 0,
|
|
"message": {
|
|
"role": "assistant",
|
|
"content": json_content
|
|
},
|
|
"finish_reason": "stop"
|
|
}],
|
|
"usage": {
|
|
"prompt_tokens": 100,
|
|
"completion_tokens": 50,
|
|
"total_tokens": 150,
|
|
"cost": 0.001
|
|
}
|
|
});
|
|
ResponseTemplate::new(200).set_body_json(body)
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn extract_roaster_via_text_prompt_populates_form() {
|
|
let app = spawn_app_with_openrouter_mock().await;
|
|
let mock_server = app.mock_server.as_ref().unwrap();
|
|
|
|
// Mount mock that returns extracted roaster data
|
|
Mock::given(method("POST"))
|
|
.and(path("/api/v1/chat/completions"))
|
|
.respond_with(mock_openrouter_response(
|
|
r#"{"name": "Square Mile", "country": "United Kingdom", "city": "London", "homepage": "https://squaremilecoffee.com"}"#,
|
|
))
|
|
.mount(mock_server)
|
|
.await;
|
|
|
|
let session = BrowserSession::new(&app.address).await.unwrap();
|
|
authenticate_browser(&session, &app).await.unwrap();
|
|
|
|
// Navigate to /add (roaster tab is default)
|
|
session.goto("/add").await.unwrap();
|
|
|
|
// Wait for the extraction prompt to be visible
|
|
let prompt_input = wait_for_visible(
|
|
&session.driver,
|
|
"input[name='prompt'][form='roaster-extract-form']",
|
|
)
|
|
.await
|
|
.unwrap();
|
|
|
|
// Type a description and press Enter to submit the extraction form
|
|
prompt_input.send_keys("Square Mile Coffee").await.unwrap();
|
|
prompt_input.send_keys(Key::Enter).await.unwrap();
|
|
|
|
// Wait for extraction to complete — the name field should be populated
|
|
// Datastar patches signals which update form fields via data-bind
|
|
tokio::time::sleep(std::time::Duration::from_secs(2)).await;
|
|
|
|
// Verify the main form fields were populated by the signal patch
|
|
let name_input = session
|
|
.driver
|
|
.find(By::Css(
|
|
"form[action='/api/v1/roasters'] input[name='name']",
|
|
))
|
|
.await
|
|
.unwrap();
|
|
let name_value = name_input.value().await.unwrap().unwrap_or_default();
|
|
assert_eq!(name_value, "Square Mile", "Name field should be populated");
|
|
|
|
let country_input = session
|
|
.driver
|
|
.find(By::Css(
|
|
"form[action='/api/v1/roasters'] input[name='country']",
|
|
))
|
|
.await
|
|
.unwrap();
|
|
let country_value = country_input.value().await.unwrap().unwrap_or_default();
|
|
assert_eq!(
|
|
country_value, "United Kingdom",
|
|
"Country field should be populated"
|
|
);
|
|
|
|
// Submit the main form to save the roaster
|
|
let submit_button = session
|
|
.driver
|
|
.find(By::Css(
|
|
"form[action='/api/v1/roasters'] button[type='submit']",
|
|
))
|
|
.await
|
|
.unwrap();
|
|
submit_button.click().await.unwrap();
|
|
|
|
// Should redirect to the roaster detail page
|
|
wait_for_url_contains(&session.driver, "/roasters/")
|
|
.await
|
|
.unwrap();
|
|
|
|
let heading = session.driver.find(By::Css("h1")).await.unwrap();
|
|
assert_eq!(heading.text().await.unwrap(), "Square Mile");
|
|
|
|
session.quit().await;
|
|
}
|