diff options
| author | Bartek Iwańczuk <biwanczuk@gmail.com> | 2023-12-13 11:00:45 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-13 11:00:45 +0100 |
| commit | bbf8f69cb979be0f36c38ae52b1588e648b3252e (patch) | |
| tree | c6b8f727a1c99eb848248c0891d08eb72dc73b49 /test_util/src | |
| parent | 86769b0d1cfb7341df31278913524492d5f9ab52 (diff) | |
test: integration tests for tarring/unfurling (#21544)
Diffstat (limited to 'test_util/src')
| -rw-r--r-- | test_util/src/lib.rs | 48 |
1 files changed, 48 insertions, 0 deletions
diff --git a/test_util/src/lib.rs b/test_util/src/lib.rs index 6700a49f1..ba54ef624 100644 --- a/test_util/src/lib.rs +++ b/test_util/src/lib.rs @@ -48,6 +48,7 @@ use pty::Pty; use regex::Regex; use rustls_tokio_stream::TlsStream; use serde::Serialize; +use serde_json::json; use std::collections::HashMap; use std::convert::Infallible; use std::env; @@ -120,6 +121,7 @@ const WS_CLOSE_PORT: u16 = 4244; const WS_PING_PORT: u16 = 4245; const H2_GRPC_PORT: u16 = 4246; const H2S_GRPC_PORT: u16 = 4247; +const REGISTRY_SERVER_PORT: u16 = 4250; pub const PERMISSION_VARIANTS: [&str; 5] = ["read", "write", "env", "net", "run"]; @@ -1815,6 +1817,8 @@ pub async fn run_all_servers() { let h2_only_server_fut = wrap_http_h2_only_server(); let h2_grpc_server_fut = h2_grpc_server(); + let registry_server_fut = registry_server(); + let server_fut = async { futures::join!( redirect_server_fut, @@ -1840,6 +1844,7 @@ pub async fn run_all_servers() { h1_only_server_fut, h2_only_server_fut, h2_grpc_server_fut, + registry_server_fut, ) } .boxed_local(); @@ -2698,6 +2703,49 @@ pub fn parse_max_mem(output: &str) -> Option<u64> { None } +async fn registry_server_handler( + req: Request<Body>, +) -> Result<Response<Body>, hyper::http::Error> { + let path = req.uri().path(); + + if path.starts_with("/scopes/") { + let body = serde_json::to_string_pretty(&json!({ + "id": "sdfwqer-sffg-qwerasdf", + "status": "success", + "error": null + })) + .unwrap(); + let res = Response::new(Body::from(body)); + return Ok(res); + } else if path.starts_with("/publish_status/") { + let body = serde_json::to_string_pretty(&json!({ + "id": "sdfwqer-qwer-qwerasdf", + "status": "success", + "error": null + })) + .unwrap(); + let res = Response::new(Body::from(body)); + return Ok(res); + } + + Response::builder() + .status(StatusCode::NOT_FOUND) + .body(Body::empty()) +} + +async fn registry_server() { + let registry_server_addr = + SocketAddr::from(([127, 0, 0, 1], REGISTRY_SERVER_PORT)); + let registry_server_svc = make_service_fn(|_| async { + Ok::<_, Infallible>(service_fn(registry_server_handler)) + }); + let registry_server = + Server::bind(®istry_server_addr).serve(registry_server_svc); + if let Err(e) = registry_server.await { + eprintln!("Registry server error: {:?}", e); + } +} + pub(crate) mod colors { use std::io::Write; |