Douwe Osinga
d2195289cf
Co-authored-by: Douwe Osinga <douwe@squareup.com> Co-authored-by: Jack Amadeo <jackamadeo@squareup.com> Co-authored-by: Jack Amadeo <jackamadeo@block.xyz>
42 lines
1.3 KiB
Rust
42 lines
1.3 KiB
Rust
use axum::http::StatusCode;
|
|
use axum::Router;
|
|
use axum::{body::Body, http::Request};
|
|
use etcetera::AppStrategy;
|
|
use serde_json::json;
|
|
use std::sync::Arc;
|
|
use tower::ServiceExt;
|
|
|
|
async fn create_test_app() -> Router {
|
|
let agent = Arc::new(goose::agents::Agent::default());
|
|
let state = goose_server::AppState::new(agent, "test".to_string());
|
|
|
|
// Add scheduler setup like in the existing tests
|
|
let sched_storage_path = etcetera::choose_app_strategy(goose::config::APP_STRATEGY.clone())
|
|
.unwrap()
|
|
.data_dir()
|
|
.join("schedules.json");
|
|
let sched = goose::scheduler_factory::SchedulerFactory::create_legacy(sched_storage_path)
|
|
.await
|
|
.unwrap();
|
|
state.set_scheduler(sched).await;
|
|
|
|
goose_server::routes::config_management::routes(state)
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn test_pricing_endpoint_basic() {
|
|
// Basic test to ensure pricing endpoint responds correctly
|
|
let app = create_test_app().await;
|
|
|
|
let request = Request::builder()
|
|
.uri("/config/pricing")
|
|
.method("POST")
|
|
.header("content-type", "application/json")
|
|
.header("x-secret-key", "test")
|
|
.body(Body::from(json!({"configured_only": true}).to_string()))
|
|
.unwrap();
|
|
|
|
let response = app.oneshot(request).await.unwrap();
|
|
assert_eq!(response.status(), StatusCode::OK);
|
|
}
|