Unverified Commit 56321e9f authored by Jimmy's avatar Jimmy Committed by GitHub
Browse files

[Router]fix: fix get_load missing api_key (#10385)

parent 12d6cf18
...@@ -54,6 +54,7 @@ mod test_pd_routing { ...@@ -54,6 +54,7 @@ mod test_pd_routing {
.worker_type(WorkerType::Prefill { .worker_type(WorkerType::Prefill {
bootstrap_port: Some(9000), bootstrap_port: Some(9000),
}) })
.api_key("test_api_key")
.build(), .build(),
); );
assert_eq!(prefill_worker.url(), "http://prefill:8080"); assert_eq!(prefill_worker.url(), "http://prefill:8080");
...@@ -68,6 +69,7 @@ mod test_pd_routing { ...@@ -68,6 +69,7 @@ mod test_pd_routing {
let decode_worker: Box<dyn Worker> = Box::new( let decode_worker: Box<dyn Worker> = Box::new(
BasicWorkerBuilder::new("http://decode:8080") BasicWorkerBuilder::new("http://decode:8080")
.worker_type(WorkerType::Decode) .worker_type(WorkerType::Decode)
.api_key("test_api_key")
.build(), .build(),
); );
assert_eq!(decode_worker.url(), "http://decode:8080"); assert_eq!(decode_worker.url(), "http://decode:8080");
...@@ -80,6 +82,7 @@ mod test_pd_routing { ...@@ -80,6 +82,7 @@ mod test_pd_routing {
let regular_worker: Box<dyn Worker> = Box::new( let regular_worker: Box<dyn Worker> = Box::new(
BasicWorkerBuilder::new("http://regular:8080") BasicWorkerBuilder::new("http://regular:8080")
.worker_type(WorkerType::Regular) .worker_type(WorkerType::Regular)
.api_key("test_api_key")
.build(), .build(),
); );
assert_eq!(regular_worker.url(), "http://regular:8080"); assert_eq!(regular_worker.url(), "http://regular:8080");
...@@ -297,6 +300,7 @@ mod test_pd_routing { ...@@ -297,6 +300,7 @@ mod test_pd_routing {
.worker_type(WorkerType::Prefill { .worker_type(WorkerType::Prefill {
bootstrap_port: Some(9000), bootstrap_port: Some(9000),
}) })
.api_key("test_api_key")
.build(), .build(),
); );
...@@ -700,6 +704,7 @@ mod test_pd_routing { ...@@ -700,6 +704,7 @@ mod test_pd_routing {
.worker_type(WorkerType::Prefill { .worker_type(WorkerType::Prefill {
bootstrap_port: Some(9000), bootstrap_port: Some(9000),
}) })
.api_key("test_api_key")
.build(), .build(),
); );
...@@ -836,6 +841,7 @@ mod test_pd_routing { ...@@ -836,6 +841,7 @@ mod test_pd_routing {
.worker_type(WorkerType::Prefill { .worker_type(WorkerType::Prefill {
bootstrap_port: Some(9000), bootstrap_port: Some(9000),
}) })
.api_key("test_api_key")
.build(), .build(),
); );
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment