forge_core/testing/context/
mod.rs1use crate::function::AuthContext;
10use std::collections::HashMap;
11use uuid::Uuid;
12
13mod cron;
14mod daemon;
15mod job;
16mod mcp_tool;
17mod mutation;
18mod query;
19mod webhook;
20mod workflow;
21
22pub use cron::{TestCronContext, TestCronContextBuilder};
23pub use daemon::{TestDaemonContext, TestDaemonContextBuilder};
24pub use job::{TestJobContext, TestJobContextBuilder, TestProgressUpdate};
25pub use mcp_tool::{TestMcpToolContext, TestMcpToolContextBuilder};
26pub use mutation::{TestMutationContext, TestMutationContextBuilder};
27pub use query::{TestQueryContext, TestQueryContextBuilder};
28pub use webhook::{TestWebhookContext, TestWebhookContextBuilder};
29pub use workflow::{TestWorkflowContext, TestWorkflowContextBuilder};
30
31pub(crate) fn build_test_auth(
34 user_id: Option<Uuid>,
35 roles: Vec<String>,
36 claims: HashMap<String, serde_json::Value>,
37) -> AuthContext {
38 if let Some(user_id) = user_id {
39 AuthContext::authenticated(user_id, roles, claims)
40 } else if claims.contains_key("sub") {
41 AuthContext::authenticated_without_uuid(roles, claims)
42 } else {
43 AuthContext::unauthenticated()
44 }
45}