From 48820189e380753616d3de2a79484a3ec603b3dd Mon Sep 17 00:00:00 2001
From: Noor <noormohammedb@protonmail.com>
Date: Fri, 25 Apr 2025 13:19:25 +0530
Subject: [PATCH] refactor test environment to use real gRPC clients and
 servers

---
 tests/common/prepare_test_env.rs | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/tests/common/prepare_test_env.rs b/tests/common/prepare_test_env.rs
index 7813d3b..c4fa642 100644
--- a/tests/common/prepare_test_env.rs
+++ b/tests/common/prepare_test_env.rs
@@ -2,7 +2,7 @@ use detee_shared::{
     general_proto::brain_general_cli_server::BrainGeneralCliServer,
     vm_proto::brain_vm_cli_server::BrainVmCliServer,
 };
-use surreal_brain::grpc::{BrainGeneralCliMock, BrainVmCliMock};
+use surreal_brain::grpc::{BrainGeneralCliForReal, BrainVmCliForReal};
 use tokio::sync::OnceCell;
 use tonic::transport::Channel;
 
@@ -24,16 +24,16 @@ pub async fn prepare_test_db() {
     surreal_brain::db::migration0(&old_brain_data).await.unwrap();
 }
 
-pub async fn fake_grpc_server() {
+pub async fn mock_grpc_server() {
     tonic::transport::Server::builder()
-        .add_service(BrainGeneralCliServer::new(BrainGeneralCliMock {}))
-        .add_service(BrainVmCliServer::new(BrainVmCliMock {}))
+        .add_service(BrainGeneralCliServer::new(BrainGeneralCliForReal {}))
+        .add_service(BrainVmCliServer::new(BrainVmCliForReal {}))
         .serve(GRPC_ADDR.parse().unwrap())
         .await
         .unwrap();
 }
 
-pub async fn fake_grpc_client() -> Channel {
+pub async fn mock_grpc_client() -> Channel {
     let url = format!("http://{GRPC_ADDR}");
     Channel::from_shared(url).unwrap().connect().await.unwrap()
 }
@@ -44,11 +44,11 @@ pub async fn prepare_test_setup() {
             prepare_test_db().await;
 
             tokio::spawn(async {
-                fake_grpc_server().await;
+                mock_grpc_server().await;
             });
 
             tokio::time::sleep(std::time::Duration::from_secs(1)).await;
-            fake_grpc_client().await
+            mock_grpc_client().await
         })
         .await;
 }