add test for bad connection string

plane/plane-tests/tests/proxy.rs

@@ -1,4 +1,5 @@
 use common::{proxy_mock::MockProxy, test_env::TestEnvironment};
+use plane::{protocol::RouteInfoResponse, types::BearerToken};
 use plane_test_macro::plane_test;
 use reqwest::StatusCode;
 
@@ -16,3 +17,27 @@ async fn proxy_no_bearer_token(env: TestEnvironment) {
 
     assert_eq!(response.status(), StatusCode::UNAUTHORIZED);
 }
+
+#[plane_test]
+async fn proxy_bad_bearer_token(env: TestEnvironment) {
+    let mut proxy = MockProxy::new().await;
+    let url = format!("http://{}/abc123/", proxy.addr());
+    let handle = tokio::spawn(async { reqwest::get(url).await.expect("Failed to send request") });
+
+    let route_info_request = proxy.recv_route_info_request().await;
+    assert_eq!(
+        route_info_request.token,
+        BearerToken::from("abc123".to_string())
+    );
+
+    proxy
+        .send_route_info_response(RouteInfoResponse {
+            token: BearerToken::from("abc123".to_string()),
+            route_info: None,
+        })
+        .await;
+
+    let response = handle.await.unwrap();
+
+    assert_eq!(response.status(), StatusCode::UNAUTHORIZED);
+}

plane/src/proxy/proxy_server.rs

@@ -76,8 +76,10 @@ impl Service<Request<Incoming>> for ProxyState {
             let route_info = inner.route_map.lookup(&bearer_token).await;
 
             let Some(route_info) = route_info else {
-                // TODO
-                panic!("Route info not found for bearer token");
+                return Ok(Response::builder()
+                    .status(StatusCode::UNAUTHORIZED)
+                    .body(simple_empty_body())
+                    .unwrap());
             };
 
             request.set_upstream_address(route_info.address.0);