rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2026-10-20 19:02:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: strengthen OTLP HTTP route coverage

Browse Source 
Signed-off-by: Lei, HUANG <ratuthomm@gmail.com>
This commit is contained in:
Lei, HUANG
2026-09-09 17:01:13 +08:00
parent bb6217fd2b
commit 2542b74cc4
1 changed files with 32 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/frontend/src/server.rs
+32 -24
View File
@@ -761,13 +761,16 @@ mod tests {
.await
.unwrap(),
);
let mut services = Services::new_with_http_routes(
opts.clone(),
ExternalMetricHttpConsumers::default(),
otlp_http_routes,
instance,
Default::default(),
);
let mut services = match otlp_http_routes {
OtlpHttpRoutes::Configured => Services::new(opts.clone(), instance, Default::default()),
OtlpHttpRoutes::MetricsOnly => Services::new_with_http_routes(
opts.clone(),
ExternalMetricHttpConsumers::default(),
OtlpHttpRoutes::MetricsOnly,
instance,
Default::default(),
),
};
let wiring = services.metric_batcher_wiring.clone();
let request_memory_limiter = services.server_memory_limiter.clone();
let (http_server, _) = services
@@ -802,8 +805,8 @@ mod tests {
"/v1/otlp/v1/traces",
"/v1/otlp/v1/logs",
] {
assert_ne!(
StatusCode::NOT_FOUND,
assert_eq!(
StatusCode::BAD_REQUEST,
route_status(&app, path).await,
"{path}"
);
@@ -838,22 +841,27 @@ mod tests {
#[tokio::test]
async fn test_metrics_only_exposes_metrics_but_not_traces_or_logs() {
let mut opts = FrontendOptions::default();
opts.otlp.enable = false;
let (app, _) = services_http_app(opts, OtlpHttpRoutes::MetricsOnly).await;
for otlp_enabled in [false, true] {
let mut opts = FrontendOptions::default();
opts.otlp.enable = otlp_enabled;
let (app, _) = services_http_app(opts, OtlpHttpRoutes::MetricsOnly).await;
assert_ne!(
StatusCode::NOT_FOUND,
route_status(&app, "/v1/otlp/v1/metrics").await
);
assert_eq!(
StatusCode::NOT_FOUND,
route_status(&app, "/v1/otlp/v1/traces").await
);
assert_eq!(
StatusCode::NOT_FOUND,
route_status(&app, "/v1/otlp/v1/logs").await
);
assert_eq!(
StatusCode::BAD_REQUEST,
route_status(&app, "/v1/otlp/v1/metrics").await,
"otlp_enabled={otlp_enabled}"
);
assert_eq!(
StatusCode::NOT_FOUND,
route_status(&app, "/v1/otlp/v1/traces").await,
"otlp_enabled={otlp_enabled}"
);
assert_eq!(
StatusCode::NOT_FOUND,
route_status(&app, "/v1/otlp/v1/logs").await,
"otlp_enabled={otlp_enabled}"
);
}
}
#[test]