Function test_proxy_forwards_upstream_errors

src/app/tests.rs:189–221

()

#[tokio::test]
async fn test_proxy_forwards_upstream_errors() {
    let mock_server = MockServer::start().await;

    Mock::given(method("POST"))
        .and(path("/v1/chat/completions"))
        .respond_with(ResponseTemplate::new(400).set_body_json(serde_json::json!({
            "error": {"message": "Invalid model", "type": "invalid_request_error"}
        })))
        .mount(&mock_server)
        .await;

    let app = make_app(&mock_server.uri());
    let body = r#"{"model":"nonexistent"}"#;
    let req = Request::builder()
        .method("POST")
        .uri("/v1/chat/completions")
        .header("authorization", "Bearer vk-test-1")
        .header("content-type", "application/json")
        .body(Body::from(body))
        .unwrap();

    let resp = app.oneshot(req).await.unwrap();
    assert_eq!(resp.status(), StatusCode::BAD_REQUEST);

    let body = resp.into_body().collect().await.unwrap().to_bytes();
    let json: serde_json::Value = serde_json::from_slice(&body).unwrap();
    assert!(
        json["error"]["message"]
            .as_str()
            .unwrap()
            .contains("Invalid model")
    );
}

#[tokio::test]
async fn test_real_key_injected_correctly() {