testnet/brain
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor test environment to use real gRPC clients and servers

Browse Source
This commit is contained in:
Noor 2025-04-25 13:19:25 +05:30
parent bea41c939d
commit 48820189e3
Signed by: noormohammedb
GPG Key ID: D83EFB8B3B967146
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/common/prepare_test_env.rs
View File

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