Update public backend API surface
Private source commit: ba3f7ce2b6d9
This commit is contained in:
parent
784463622c
commit
26fdcb9d84
34 changed files with 5473 additions and 72 deletions
183
crates/clusterflux-client/tests/client_contract.rs
Normal file
183
crates/clusterflux-client/tests/client_contract.rs
Normal file
|
|
@ -0,0 +1,183 @@
|
|||
use std::net::TcpListener;
|
||||
|
||||
use clusterflux_client::{
|
||||
ApiErrorCategory, ApiErrorCode, ArtifactDownloadPoll, ArtifactId, ClientError,
|
||||
ClusterfluxClient, MockTransport, ProjectId, SessionCredential, TenantId, UserId,
|
||||
CLIENT_API_VERSION,
|
||||
};
|
||||
use clusterflux_control::CONTROL_API_PATH;
|
||||
use clusterflux_coordinator::service::CoordinatorService;
|
||||
use serde_json::{json, Value};
|
||||
|
||||
#[tokio::test]
|
||||
async fn mock_transport_exercises_typed_envelope_and_session_plumbing() {
|
||||
let transport = MockTransport::from_json_responses([json!({
|
||||
"type": "projects",
|
||||
"projects": [{
|
||||
"id": "project-one",
|
||||
"tenant": "tenant-one",
|
||||
"name": "Project one"
|
||||
}],
|
||||
"actor": "user-one"
|
||||
})
|
||||
.to_string()]);
|
||||
let client = ClusterfluxClient::with_transport(transport.clone())
|
||||
.with_session_credential(&SessionCredential::from_secret("test-session-secret"));
|
||||
|
||||
let projects = client.list_projects().await.unwrap();
|
||||
assert_eq!(projects.len(), 1);
|
||||
assert_eq!(projects[0].id, ProjectId::from("project-one"));
|
||||
|
||||
let requests = transport.requests();
|
||||
assert_eq!(requests.len(), 1);
|
||||
assert_eq!(requests[0].api_path, CONTROL_API_PATH);
|
||||
let envelope: Value = serde_json::from_slice(&requests[0].body).unwrap();
|
||||
assert_eq!(envelope["protocol_version"], CLIENT_API_VERSION);
|
||||
assert_eq!(envelope["request_id"], "client-1");
|
||||
assert_eq!(envelope["operation"], "authenticated");
|
||||
assert_eq!(envelope["payload"]["type"], "authenticated");
|
||||
assert_eq!(envelope["payload"]["request"]["type"], "list_projects");
|
||||
assert_eq!(envelope["payload"]["session_secret"], "test-session-secret");
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn structured_errors_retain_machine_fields_and_originating_request_id() {
|
||||
let transport = MockTransport::from_json_responses([json!({
|
||||
"type": "error",
|
||||
"code": "account_suspended",
|
||||
"category": "authorization",
|
||||
"message": "account access is suspended",
|
||||
"retryable": false,
|
||||
"request_id": "client-1"
|
||||
})
|
||||
.to_string()]);
|
||||
let client = ClusterfluxClient::with_transport(transport)
|
||||
.with_session_credential(&SessionCredential::from_secret("test-session-secret"));
|
||||
|
||||
let ClientError::Api(error) = client.account_status().await.unwrap_err() else {
|
||||
panic!("expected typed API error");
|
||||
};
|
||||
assert_eq!(error.code, ApiErrorCode::AccountSuspended);
|
||||
assert_eq!(error.category, ApiErrorCategory::Authorization);
|
||||
assert_eq!(error.request_id, "client-1");
|
||||
assert!(!error.retryable);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn a_mismatched_error_request_id_is_rejected_as_a_protocol_error() {
|
||||
let transport = MockTransport::from_json_responses([json!({
|
||||
"type": "error",
|
||||
"code": "validation_error",
|
||||
"category": "validation",
|
||||
"message": "bad request",
|
||||
"retryable": false,
|
||||
"request_id": "another-request"
|
||||
})
|
||||
.to_string()]);
|
||||
let client = ClusterfluxClient::with_transport(transport)
|
||||
.with_session_credential(&SessionCredential::from_secret("test-session-secret"));
|
||||
|
||||
let ClientError::Protocol(message) = client.list_projects().await.unwrap_err() else {
|
||||
panic!("expected protocol error");
|
||||
};
|
||||
assert!(message.contains("does not match client-1"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn session_credentials_are_redacted_from_debug_output() {
|
||||
let credential = SessionCredential::from_secret("must-not-appear");
|
||||
assert_eq!(format!("{credential:?}"), "SessionCredential([REDACTED])");
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn artifact_download_is_a_typed_bounded_chunk_stream() {
|
||||
let link = json!({
|
||||
"artifact": "artifact-one",
|
||||
"artifact_digest": "sha256:aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa",
|
||||
"artifact_size_bytes": 5,
|
||||
"source": { "RetainedNode": "node-one" },
|
||||
"url_path": "/artifacts/tenant-one/project-one/process-one/artifact-one",
|
||||
"scoped_token_digest": "sha256:bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb",
|
||||
"expires_at_epoch_seconds": 1000,
|
||||
"tenant": "tenant-one",
|
||||
"project": "project-one",
|
||||
"process": "process-one",
|
||||
"actor": { "User": "user-one" },
|
||||
"max_bytes": 1024,
|
||||
"policy_context_digest": "sha256:cccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccc"
|
||||
});
|
||||
let transport = MockTransport::from_json_responses([
|
||||
json!({ "type": "artifact_download_link", "link": link.clone() }).to_string(),
|
||||
json!({
|
||||
"type": "artifact_download_stream",
|
||||
"link": link.clone(),
|
||||
"streamed_bytes": 0,
|
||||
"charged_download_bytes": 0,
|
||||
"content_bytes_available": false,
|
||||
"content_eof": false
|
||||
})
|
||||
.to_string(),
|
||||
json!({
|
||||
"type": "artifact_download_stream",
|
||||
"link": link,
|
||||
"streamed_bytes": 5,
|
||||
"charged_download_bytes": 5,
|
||||
"content_bytes_available": true,
|
||||
"content_offset": 0,
|
||||
"content_eof": true,
|
||||
"content_base64": "aGVsbG8="
|
||||
})
|
||||
.to_string(),
|
||||
]);
|
||||
let client = ClusterfluxClient::with_transport(transport)
|
||||
.with_session_credential(&SessionCredential::from_secret("test-session-secret"));
|
||||
let mut download = client
|
||||
.begin_artifact_download(ArtifactId::from("artifact-one"), 1024, 60, 64)
|
||||
.await
|
||||
.unwrap();
|
||||
assert_eq!(
|
||||
download.next_chunk().await.unwrap(),
|
||||
ArtifactDownloadPoll::Pending
|
||||
);
|
||||
assert_eq!(
|
||||
download.next_chunk().await.unwrap(),
|
||||
ArtifactDownloadPoll::Chunk {
|
||||
offset: 0,
|
||||
bytes: b"hello".to_vec(),
|
||||
eof: true,
|
||||
}
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn typed_client_runs_against_the_real_strict_control_endpoint() {
|
||||
let listener = TcpListener::bind("127.0.0.1:0").unwrap();
|
||||
let address = listener.local_addr().unwrap();
|
||||
let server = std::thread::spawn(move || {
|
||||
let mut service = CoordinatorService::new(7);
|
||||
service
|
||||
.issue_cli_session(
|
||||
TenantId::from("tenant-one"),
|
||||
ProjectId::from("project-one"),
|
||||
UserId::from("user-one"),
|
||||
"real-endpoint-session",
|
||||
None,
|
||||
)
|
||||
.unwrap();
|
||||
let (stream, _) = listener.accept().unwrap();
|
||||
service.handle_stream(stream).unwrap();
|
||||
});
|
||||
|
||||
let client = ClusterfluxClient::connect(format!("clusterflux+tcp://{address}"))
|
||||
.unwrap()
|
||||
.with_session_credential(&SessionCredential::from_secret("real-endpoint-session"));
|
||||
let projects = client.list_projects().await.unwrap();
|
||||
assert_eq!(projects.len(), 1);
|
||||
assert_eq!(projects[0].id, ProjectId::from("project-one"));
|
||||
let status = client.account_status().await.unwrap();
|
||||
assert!(status.authenticated);
|
||||
assert_eq!(status.actor, UserId::from("user-one"));
|
||||
|
||||
drop(client);
|
||||
server.join().unwrap();
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue