test: add cors test

GreptimeTeam · Jan 24, 2025 · cf504f3 · cf504f3
1 parent dd800d8
commit cf504f3
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/src/servers/src/http.rs b/src/servers/src/http.rs
@@ -747,7 +747,6 @@ impl HttpServer {
                                 Method::PUT,
                                 Method::DELETE,
                                 Method::HEAD,
-                                Method::OPTIONS,
                             ])
                             .allow_origin(Any)
                             .allow_headers(Any),

diff --git a/tests-integration/Cargo.toml b/tests-integration/Cargo.toml
@@ -86,6 +86,7 @@ zstd.workspace = true
 datafusion.workspace = true
 datafusion-expr.workspace = true
 hex.workspace = true
+http.workspace = true
 itertools.workspace = true
 opentelemetry-proto.workspace = true
 partition.workspace = true

diff --git a/tests-integration/tests/http.rs b/tests-integration/tests/http.rs
@@ -148,6 +148,24 @@ pub async fn test_http_auth(store_type: StorageType) {
     guard.remove_all().await;
 }
 
+#[tokio::test]
+pub async fn test_cors() {
+    let (app, mut guard) = setup_test_http_app_with_frontend(StorageType::File, "test_cors").await;
+    let client = TestClient::new(app).await;
+
+    let res = client.get("/health").send().await;
+
+    assert_eq!(res.status(), StatusCode::OK);
+    assert_eq!(
+        res.headers()
+            .get(http::header::ACCESS_CONTROL_ALLOW_ORIGIN)
+            .expect("expect cors header origin"),
+        "*"
+    );
+
+    guard.remove_all().await;
+}
+
 pub async fn test_sql_api(store_type: StorageType) {
     let (app, mut guard) = setup_test_http_app_with_frontend(store_type, "sql_api").await;
     let client = TestClient::new(app).await;