fraiseql-server 2.7.0

//! Tests for `routes/functions/` module.
#![allow(clippy::unwrap_used)] // Reason: test code, panics acceptable
#![allow(missing_docs)] // Reason: test code

use std::sync::Arc;

use async_trait::async_trait;
use axum::{
    body::Body,
    http::{Method, Request, StatusCode},
};
use fraiseql_functions::{
    FunctionStore, InMemoryFunctionStore,
    runtime::SendFunctionRuntime,
    types::{EventPayload, FunctionModule, FunctionResult, LogEntry, ResourceLimits, RuntimeType},
};
use tower::ServiceExt as _;

use super::{FunctionsRouteState, functions_router};

/// Build a test router with a given store and noop runtime.
fn make_test_state(store: Arc<dyn FunctionStore>) -> FunctionsRouteState {
    struct NoopRuntime;
    // async_trait: dyn-dispatch required; remove when RTN + Send is stable (RFC 3425)
    // Reason: SendFunctionRuntime is itself an async_trait; impls must match the macro signature.
    #[async_trait]
    impl SendFunctionRuntime for NoopRuntime {
        async fn invoke_raw(
            &self,
            _module: &FunctionModule,
            _event: EventPayload,
            _limits: ResourceLimits,
        ) -> fraiseql_error::Result<FunctionResult> {
            Ok(FunctionResult {
                value:             Some(serde_json::json!({"ok": true})),
                logs:              Vec::<LogEntry>::new(),
                duration:          std::time::Duration::from_millis(1),
                memory_peak_bytes: 0,
            })
        }

        fn supported_extensions(&self) -> &[&str] {
            &[]
        }

        fn supports_hot_reload(&self) -> bool {
            false
        }

        // Reason: trait signature returns `&str` (elided to `&'a self → &'a str`);
        // narrowing to `&'static str` would change the impl signature and fail E0195.
        #[allow(clippy::unnecessary_literal_bound)]
        fn name(&self) -> &str {
            "noop"
        }
    }

    FunctionsRouteState {
        store,
        runtime: Arc::new(NoopRuntime),
    }
}

#[tokio::test]
async fn test_invoke_returns_404_for_unknown_function() {
    let store = Arc::new(InMemoryFunctionStore::new());
    let state = make_test_state(store);
    let app = functions_router(state);

    let req = Request::builder()
        .method(Method::POST)
        .uri("/functions/v1/ghost")
        .body(Body::empty())
        .unwrap();

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

#[tokio::test]
async fn test_invoke_returns_200_for_known_function() {
    let store = Arc::new(InMemoryFunctionStore::new());
    store
        .store_function("echo", RuntimeType::Wasm, bytes::Bytes::from_static(b"fake"))
        .await
        .unwrap();

    let state = make_test_state(store);
    let app = functions_router(state);

    let req = Request::builder()
        .method(Method::POST)
        .uri("/functions/v1/echo")
        .body(Body::from(r#"{"hello":"world"}"#))
        .unwrap();

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