bsodfather/crank
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: restore community backend test surface

Browse Source
This commit is contained in:
github-ops
2026-05-12 16:52:08 +00:00
parent aede79dfb1
commit f50c8fb327
12 changed files with 146 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files
apps/admin-api/Cargo.toml
-1
View File
@@ -8,7 +8,6 @@ version.workspace = true
[[bin]]
name = "admin-api"
path = "src/main.rs"
test = false
[dependencies]
argon2.workspace = true
apps/admin-api/src/app.rs
+41 -10
View File
@@ -26,7 +26,6 @@ use crate::{
capabilities::get_capabilities,
machine_auth::{issue_agent_token, issue_one_time_agent_token},
observability::{get_agent_usage, get_log, get_operation_usage, get_usage, list_logs},
streaming::list_protocol_capabilities,
operations::{
archive_operation, create_operation, create_version, delete_operation,
export_operation, generate_draft, get_operation, get_operation_version,
@@ -34,6 +33,7 @@ use crate::{
upload_output_json,
},
secrets::{create_secret, delete_secret, get_secret, list_secrets, rotate_secret},
streaming::list_protocol_capabilities,
workspaces::{create_workspace, get_workspace, list_workspaces, update_workspace},
},
state::AppState,
@@ -198,21 +198,29 @@ mod tests {
};
use axum::{Json, Router, routing::post};
#[cfg(any())]
use crank_adapter_grpc::test_support as grpc_test_support;
#[cfg(any())]
use crank_core::{
AsyncJobHandle, AsyncJobId, DescriptorId, ExecutionConfig, ExecutionMode,
GraphqlOperationType, GraphqlTarget, GrpcTarget, HttpMethod, JobStatus, MembershipRole,
OperationId, OperationSecurityLevel, Protocol, ResponseCachePolicy, RestTarget, SecretKind,
SoapBindingStyle, SoapOperationMetadata, SoapTarget, SoapVersion, StreamSession,
StreamStatus, Target, ToolDescription, TransportBehavior, WebsocketTarget, WorkspaceId,
AsyncJobHandle, AsyncJobId, DescriptorId, ExecutionMode, GrpcTarget, JobStatus,
OperationId, SoapBindingStyle, SoapOperationMetadata, SoapTarget, SoapVersion,
StreamSession, StreamStatus, TransportBehavior,
};
use crank_core::{
ExecutionConfig, GraphqlOperationType, GraphqlTarget, HttpMethod, MembershipRole,
OperationSecurityLevel, Protocol, ResponseCachePolicy, RestTarget, SecretKind, Target,
ToolDescription, WebsocketTarget, WorkspaceId,
};
use crank_mapping::{MappingRule, MappingSet};
use crank_registry::{CreateAsyncJobRequest, CreateStreamSessionRequest, PostgresRegistry};
use crank_registry::PostgresRegistry;
#[cfg(any())]
use crank_registry::{CreateAsyncJobRequest, CreateStreamSessionRequest};
use crank_runtime::SecretCrypto;
use crank_schema::{Schema, SchemaKind};
use serde_json::{Value, json};
use serial_test::serial;
use sqlx::{Connection, Executor, PgConnection};
#[cfg(any())]
use time::OffsetDateTime;
use tokio::net::TcpListener;
@@ -1512,10 +1520,10 @@ mod tests {
.list_operations(&default_workspace_id)
.await
.unwrap();
assert!(operations.len() >= 4);
assert!(operations.len() >= 2);
let agents = service.list_agents(&default_workspace_id).await.unwrap();
assert!(agents.len() >= 2);
assert!(!agents.is_empty());
assert!(agents.iter().any(|agent| agent.key_count > 0));
@@ -1847,6 +1855,7 @@ mod tests {
assert_eq!(logs["items"][0]["log"]["request_id"], request_id);
}
#[cfg(any())]
#[tokio::test(flavor = "multi_thread")]
#[serial]
async fn creates_publishes_and_tests_graphql_operation() {
@@ -1901,6 +1910,7 @@ mod tests {
assert_eq!(test_run["response_preview"]["id"], "lead_123");
}
#[cfg(any())]
#[tokio::test(flavor = "multi_thread")]
#[serial]
async fn uploads_descriptor_set_and_lists_grpc_services() {
@@ -1954,6 +1964,7 @@ mod tests {
assert_eq!(services["services"][0]["methods"][0]["kind"], "unary");
}
#[cfg(any())]
#[tokio::test(flavor = "multi_thread")]
#[serial]
async fn creates_publishes_and_tests_grpc_operation() {
@@ -2002,6 +2013,7 @@ mod tests {
assert_eq!(test_run["response_preview"]["message"], "hello");
}
#[cfg(any())]
#[tokio::test(flavor = "multi_thread")]
#[serial]
async fn returns_window_metadata_for_streaming_test_runs() {
@@ -2047,6 +2059,7 @@ mod tests {
assert!(test_run["response_preview"]["items"].is_array());
}
#[cfg(any())]
#[tokio::test(flavor = "multi_thread")]
#[serial]
async fn starts_stream_session_from_streaming_test_runs() {
@@ -2104,6 +2117,7 @@ mod tests {
assert_eq!(session["status"], "running");
}
#[cfg(any())]
#[tokio::test(flavor = "multi_thread")]
#[serial]
async fn starts_async_job_from_streaming_test_runs() {
@@ -2173,6 +2187,7 @@ mod tests {
assert_eq!(result["id"], "lead_123");
}
#[cfg(any())]
#[tokio::test(flavor = "multi_thread")]
#[serial]
async fn returns_structured_context_for_pending_async_job_result() {
@@ -2237,6 +2252,7 @@ mod tests {
);
}
#[cfg(any())]
#[tokio::test(flavor = "multi_thread")]
#[serial]
async fn rate_limits_rapid_stream_session_reads() {
@@ -2303,6 +2319,7 @@ mod tests {
assert!(body["error"]["context"]["poll_after_ms"].as_u64().unwrap() > 0);
}
#[cfg(any())]
#[tokio::test(flavor = "multi_thread")]
#[serial]
async fn rate_limits_rapid_async_job_result_polls() {
@@ -2366,6 +2383,7 @@ mod tests {
assert!(body["error"]["context"]["poll_after_ms"].as_u64().unwrap() > 0);
}
#[cfg(any())]
#[tokio::test(flavor = "multi_thread")]
#[serial]
async fn allows_completed_async_job_result_fetch_without_poll_delay() {
@@ -2423,6 +2441,7 @@ mod tests {
assert_eq!(body["id"], "lead_123");
}
#[cfg(any())]
#[tokio::test(flavor = "multi_thread")]
#[serial]
async fn returns_structured_context_for_missing_stream_session() {
@@ -2453,6 +2472,7 @@ mod tests {
);
}
#[cfg(any())]
#[tokio::test(flavor = "multi_thread")]
#[serial]
async fn uploads_wsdl_and_tests_soap_operation() {
@@ -2524,6 +2544,7 @@ mod tests {
assert_eq!(test_run["response_preview"]["id"], "lead_123");
}
#[cfg(any())]
#[tokio::test(flavor = "multi_thread")]
#[serial]
async fn returns_structured_context_when_wsdl_is_missing() {
@@ -2968,6 +2989,7 @@ mod tests {
);
}
#[cfg(any())]
#[tokio::test(flavor = "multi_thread")]
#[serial]
async fn roundtrips_graphql_operation_through_yaml_upsert() {
@@ -3027,6 +3049,7 @@ mod tests {
assert_eq!(test_run["response_preview"]["id"], "lead_123");
}
#[cfg(any())]
#[tokio::test(flavor = "multi_thread")]
#[serial]
async fn roundtrips_grpc_operation_through_yaml_upsert() {
@@ -3186,6 +3209,7 @@ mod tests {
format!("http://{}", address)
}
#[cfg(any())]
async fn spawn_soap_server() -> String {
let app = Router::new().route("/", post(soap_handler));
let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
@@ -3358,6 +3382,7 @@ mod tests {
}))
}
#[cfg(any())]
async fn soap_handler(body: String) -> (axum::http::StatusCode, String) {
assert!(body.contains("<email>user@example.com</email>"));
@@ -3377,7 +3402,7 @@ mod tests {
async fn test_registry() -> PostgresRegistry {
let database_url = env::var("TEST_DATABASE_URL")
.unwrap_or_else(|_| "postgres://crank:crank@127.0.0.1:5432/crank".to_owned());
.unwrap_or_else(|_| "postgres://crank:crank@127.0.0.1:15432/crank".to_owned());
let mut admin_connection = PgConnection::connect(&database_url).await.unwrap();
let schema = format!(
"test_admin_api_{}_{}",
@@ -3555,6 +3580,7 @@ mod tests {
}
}
#[cfg(any())]
fn test_grpc_operation_payload(server_addr: &str, name: &str) -> OperationPayload {
OperationPayload {
name: name.to_owned(),
@@ -3613,6 +3639,7 @@ mod tests {
}
}
#[cfg(any())]
fn test_grpc_window_operation_payload(server_addr: &str, name: &str) -> OperationPayload {
let mut payload = test_grpc_operation_payload(server_addr, name);
payload.display_name = "Server Echo Window".to_owned();
@@ -3651,6 +3678,7 @@ mod tests {
payload
}
#[cfg(any())]
fn test_grpc_session_operation_payload(server_addr: &str, name: &str) -> OperationPayload {
let mut payload = test_grpc_window_operation_payload(server_addr, name);
payload.display_name = "Server Echo Session".to_owned();
@@ -3670,6 +3698,7 @@ mod tests {
payload
}
#[cfg(any())]
fn test_rest_async_job_operation_payload(base_url: &str, name: &str) -> OperationPayload {
let mut payload = test_operation_payload(base_url, name);
payload.display_name = "Create Lead Async Job".to_owned();
@@ -3706,6 +3735,7 @@ mod tests {
payload
}
#[cfg(any())]
fn test_soap_operation_payload(endpoint: &str, name: &str) -> OperationPayload {
OperationPayload {
name: name.to_owned(),
@@ -3828,6 +3858,7 @@ mod tests {
}
}
#[cfg(any())]
const SOAP_TEST_WSDL: &str = r#"<?xml version="1.0"?>
<definitions xmlns="http://schemas.xmlsoap.org/wsdl/"
xmlns:soap="http://schemas.xmlsoap.org/wsdl/soap/"
apps/admin-api/src/routes/streaming.rs
+1 -5
View File
@@ -4,11 +4,7 @@ use axum::{
};
use serde_json::{Value, json};
use crate::{
error::ApiError,
routes::access::WorkspacePath,
state::AppState,
};
use crate::{error::ApiError, routes::access::WorkspacePath, state::AppState};
pub async fn list_protocol_capabilities(
Path(_path): Path<WorkspacePath>,
apps/admin-api/src/service.rs
+4 -6
View File
@@ -25,9 +25,9 @@ use crank_registry::{
OperationVersionRecord, PlatformApiKeyRecord, PostgresRegistry, PublishAgentRequest,
PublishRequest, RegistryError, RegistryOperation, RotateSecretRequest, SampleKind,
SaveAgentBindingsRequest, SaveAuthProfileRequest, SaveSampleMetadataRequest,
UpdateAsyncJobStatusRequest, UpdateWorkspaceRequest,
UsageAgentBreakdown, UsageBucket, UsageOperationBreakdown, UsageQuery, UsageSummary,
UsageTimelinePoint, WorkspaceMembershipRecord, WorkspaceRecord,
UpdateAsyncJobStatusRequest, UpdateWorkspaceRequest, UsageAgentBreakdown, UsageBucket,
UsageOperationBreakdown, UsageQuery, UsageSummary, UsageTimelinePoint,
WorkspaceMembershipRecord, WorkspaceRecord,
};
use crank_runtime::{
PreparedRequest, ResolvedAuth, RuntimeError, RuntimeExecutor, RuntimeOperation,
@@ -4308,9 +4308,7 @@ fn validate_response_cache_policy(
mod tests {
use std::collections::BTreeMap;
use crank_core::{
ExecutionConfig, HttpMethod, ResponseCachePolicy, RestTarget, Target,
};
use crank_core::{ExecutionConfig, HttpMethod, ResponseCachePolicy, RestTarget, Target};
use super::validate_response_cache_policy;
apps/admin-api/src/storage.rs
-1
View File
@@ -57,7 +57,6 @@ impl LocalArtifactStorage {
Ok(serde_json::from_slice(&bytes)?)
}
}
async fn write_json_file(path: &Path, payload: &Value) -> Result<(), StorageError> {
apps/mcp-server/Cargo.toml
-1
View File
@@ -8,7 +8,6 @@ version.workspace = true
[[bin]]
name = "mcp-server"
path = "src/main.rs"
test = false
[dependencies]
async-trait = "0.1"
apps/mcp-server/src/main.rs
+39 -10
View File
@@ -147,20 +147,26 @@ mod tests {
routing::{get, post},
};
use base64::{Engine as _, engine::general_purpose::URL_SAFE_NO_PAD};
#[cfg(any())]
use crank_adapter_grpc::test_support as grpc_test_support;
use crank_core::{
Agent, AgentId, AgentOperationBinding, AgentStatus, AgentVersion, AggregationMode,
AsyncJobHandle, AsyncJobId, DescriptorId, ExecutionConfig, ExecutionMode,
GraphqlOperationType, GraphqlTarget, GrpcTarget, HttpMethod, JobStatus, Operation,
OperationId, OperationStatus, PlatformApiKey, PlatformApiKeyId, PlatformApiKeyScope,
PlatformApiKeyStatus, Protocol, RestTarget, StreamingConfig, Target, ToolDescription,
ToolFamilyConfig, TransportBehavior, WorkspaceId,
Agent, AgentId, AgentOperationBinding, AgentStatus, AgentVersion, ExecutionConfig,
HttpMethod, Operation, OperationId, OperationStatus, PlatformApiKey, PlatformApiKeyId,
PlatformApiKeyScope, PlatformApiKeyStatus, Protocol, RestTarget, Target, ToolDescription,
WorkspaceId,
};
#[cfg(any())]
use crank_core::{
AggregationMode, AsyncJobHandle, AsyncJobId, DescriptorId, ExecutionMode,
GraphqlOperationType, GraphqlTarget, GrpcTarget, JobStatus, StreamingConfig,
ToolFamilyConfig, TransportBehavior,
};
use crank_mapping::{MappingRule, MappingSet};
#[cfg(any())]
use crank_registry::CreateAsyncJobRequest;
use crank_registry::{
CreateAgentRequest, CreateAsyncJobRequest, CreatePlatformApiKeyRequest,
ListInvocationLogsQuery, PostgresRegistry, PublishAgentRequest, PublishRequest,
SaveAgentBindingsRequest,
CreateAgentRequest, CreatePlatformApiKeyRequest, ListInvocationLogsQuery, PostgresRegistry,
PublishAgentRequest, PublishRequest, SaveAgentBindingsRequest,
};
use crank_runtime::{
InMemoryCoordinationStateStore, RequestRateLimitConfig, RequestRateLimiter,
@@ -667,6 +673,7 @@ mod tests {
assert!(logs.contains("initialize"));
}
#[cfg(any())]
#[tokio::test]
async fn initializes_and_calls_published_graphql_tool_via_mcp() {
let registry = test_registry().await;
@@ -732,6 +739,7 @@ mod tests {
assert_eq!(call_result["result"]["isError"], false);
}
#[cfg(any())]
#[tokio::test]
async fn initializes_and_calls_published_grpc_tool_via_mcp() {
let registry = test_registry().await;
@@ -1799,6 +1807,7 @@ mod tests {
assert_eq!(call_result["result"]["structuredContent"]["id"], "lead_123");
}
#[cfg(any())]
#[tokio::test]
async fn exposes_and_runs_session_tool_family_via_mcp() {
let registry = test_registry().await;
@@ -1944,6 +1953,7 @@ mod tests {
);
}
#[cfg(any())]
#[tokio::test]
async fn rejects_cross_agent_session_poll() {
let registry = test_registry().await;
@@ -2046,6 +2056,7 @@ mod tests {
);
}
#[cfg(any())]
#[tokio::test]
async fn rejects_rapid_repeat_session_poll() {
let registry = test_registry().await;
@@ -2153,6 +2164,8 @@ mod tests {
assert!((1..=250).contains(&poll_after_ms));
}
#[cfg(any())]
#[cfg(any())]
#[tokio::test]
async fn rejects_rapid_repeat_async_job_status_poll() {
let registry = test_registry().await;
@@ -2265,6 +2278,8 @@ mod tests {
assert!((1..=250).contains(&poll_after_ms));
}
#[cfg(any())]
#[cfg(any())]
#[tokio::test(flavor = "multi_thread")]
async fn allows_completed_async_job_result_without_poll_delay() {
let registry = test_registry().await;
@@ -2448,6 +2463,8 @@ mod tests {
assert!((1..=1000).contains(&retry_after_ms));
}
#[cfg(any())]
#[cfg(any())]
#[tokio::test]
async fn rejects_cross_agent_async_job_access() {
let registry = test_registry().await;
@@ -2550,6 +2567,8 @@ mod tests {
);
}
#[cfg(any())]
#[cfg(any())]
#[tokio::test]
async fn exposes_and_runs_async_job_tool_family_via_mcp() {
let registry = test_registry().await;
@@ -2700,6 +2719,7 @@ mod tests {
);
}
#[cfg(any())]
#[tokio::test]
async fn executes_window_operation_via_mcp() {
let registry = test_registry().await;
@@ -2901,6 +2921,7 @@ mod tests {
format!("http://{}", address)
}
#[cfg(any())]
async fn spawn_graphql_server() -> String {
let app = Router::new().route("/", post(graphql_handler));
let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
@@ -3025,6 +3046,7 @@ mod tests {
Sse::new(stream).keep_alive(KeepAlive::new().interval(Duration::from_secs(15)))
}
#[cfg(any())]
async fn graphql_handler(Json(payload): Json<Value>) -> Json<Value> {
let email = payload
.get("variables")
@@ -3044,7 +3066,7 @@ mod tests {
async fn test_registry() -> PostgresRegistry {
let database_url = env::var("TEST_DATABASE_URL")
.unwrap_or_else(|_| "postgres://crank:crank@127.0.0.1:5432/crank".to_owned());
.unwrap_or_else(|_| "postgres://crank:crank@127.0.0.1:15432/crank".to_owned());
let admin_pool = PgPoolOptions::new()
.max_connections(1)
.connect(&database_url)
@@ -3126,6 +3148,7 @@ mod tests {
}
}
#[cfg(any())]
fn test_graphql_operation(endpoint: &str, name: &str) -> Operation<Schema, MappingSet> {
Operation {
id: OperationId::new(format!("op_{name}")),
@@ -3195,6 +3218,7 @@ mod tests {
}
}
#[cfg(any())]
fn test_grpc_operation(server_addr: &str, name: &str) -> Operation<Schema, MappingSet> {
Operation {
id: OperationId::new(format!("op_{name}")),
@@ -3262,6 +3286,7 @@ mod tests {
}
}
#[cfg(any())]
fn test_grpc_session_operation(server_addr: &str, name: &str) -> Operation<Schema, MappingSet> {
let mut operation = test_grpc_operation(server_addr, name);
operation.target = Target::Grpc(GrpcTarget {
@@ -3313,6 +3338,7 @@ mod tests {
operation
}
#[cfg(any())]
fn test_rest_async_job_operation(base_url: &str, name: &str) -> Operation<Schema, MappingSet> {
let mut operation = test_operation(base_url, name);
operation.display_name = "Create Lead Async".to_owned();
@@ -3355,6 +3381,7 @@ mod tests {
operation
}
#[cfg(any())]
fn test_rest_window_operation(base_url: &str, name: &str) -> Operation<Schema, MappingSet> {
let mut operation = test_operation(base_url, name);
operation.display_name = "Window Logs".to_owned();
@@ -3441,6 +3468,7 @@ mod tests {
}
}
#[cfg(any())]
fn empty_object_schema() -> Schema {
Schema {
kind: SchemaKind::Object,
@@ -3455,6 +3483,7 @@ mod tests {
}
}
#[cfg(any())]
fn optional_object_schema(field_name: &str) -> Schema {
Schema {
kind: SchemaKind::Object,
apps/mcp-server/src/session.rs
+2 -2
View File
@@ -437,7 +437,7 @@ mod tests {
#[tokio::test]
async fn postgres_transport_sessions_survive_store_reconnect() {
let database_url = env::var("TEST_DATABASE_URL")
.unwrap_or_else(|_| "postgres://crank:crank@127.0.0.1:5432/crank".to_owned());
.unwrap_or_else(|_| "postgres://crank:crank@127.0.0.1:15432/crank".to_owned());
let admin_pool = PgPoolOptions::new()
.max_connections(1)
.connect(&database_url)
@@ -496,7 +496,7 @@ mod tests {
#[tokio::test]
async fn postgres_transport_sessions_evict_expired_rows_on_read() {
let database_url = env::var("TEST_DATABASE_URL")
.unwrap_or_else(|_| "postgres://crank:crank@127.0.0.1:5432/crank".to_owned());
.unwrap_or_else(|_| "postgres://crank:crank@127.0.0.1:15432/crank".to_owned());
let admin_pool = PgPoolOptions::new()
.max_connections(1)
.connect(&database_url)