fix web test (#872)

easytier-web/src/client_manager/mod.rs

@@ -124,6 +124,7 @@ mod tests {
         },
         web_client::WebClient,
     };
+    use sqlx::Executor;
 
     use crate::{client_manager::ClientManager, db::Db};
 
@@ -133,6 +134,12 @@ mod tests {
         let mut mgr = ClientManager::new(Db::memory_db().await);
         mgr.serve(Box::new(listener)).await.unwrap();
 
+        mgr.db()
+            .inner()
+            .execute("INSERT INTO users (username, password) VALUES ('test', 'test')")
+            .await
+            .unwrap();
+
         let connector = UdpTunnelConnector::new("udp://127.0.0.1:54333".parse().unwrap());
         let _c = WebClient::new(connector, "test", "test");
 

easytier-web/src/client_manager/session.rs

@@ -69,13 +69,9 @@ struct SessionRpcService {
     data: SharedSessionData,
 }
 
-#[async_trait::async_trait]
-impl WebServerService for SessionRpcService {
-    type Controller = BaseController;
-
-    async fn heartbeat(
+impl SessionRpcService {
+    async fn handle_heartbeat(
         &self,
-        _: BaseController,
         req: HeartbeatRequest,
     ) -> rpc_types::error::Result<HeartbeatResponse> {
         let mut data = self.data.write().await;
@@ -133,6 +129,25 @@ impl WebServerService for SessionRpcService {
     }
 }
 
+#[async_trait::async_trait]
+impl WebServerService for SessionRpcService {
+    type Controller = BaseController;
+
+    async fn heartbeat(
+        &self,
+        _: BaseController,
+        req: HeartbeatRequest,
+    ) -> rpc_types::error::Result<HeartbeatResponse> {
+        let ret = self.handle_heartbeat(req).await;
+        if ret.is_err() {
+            tracing::warn!("Failed to handle heartbeat: {:?}", ret);
+            // sleep for a while to avoid client busy loop
+            tokio::time::sleep(std::time::Duration::from_secs(2)).await;
+        }
+        ret
+    }
+}
+
 pub struct Session {
     rpc_mgr: BidirectRpcManager,