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,
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;
}