test register vm node

This commit is contained in:
Noor 2025-04-28 21:07:32 +05:30
parent c424d07cc3
commit 0359de54fe
Signed by: noormohammedb
GPG Key ID: D83EFB8B3B967146

@ -1,7 +1,8 @@
use detee_shared::common_proto::Empty; use detee_shared::common_proto::Empty;
use detee_shared::general_proto::ReportNodeReq; use detee_shared::general_proto::ReportNodeReq;
use detee_shared::vm_proto::brain_vm_cli_client::BrainVmCliClient; use detee_shared::vm_proto::brain_vm_cli_client::BrainVmCliClient;
use detee_shared::vm_proto::ListVmContractsReq; use detee_shared::vm_proto::brain_vm_daemon_client::BrainVmDaemonClient;
use detee_shared::vm_proto::{ListVmContractsReq, RegisterVmNodeReq};
use detee_shared::{ use detee_shared::{
common_proto::Pubkey, general_proto::brain_general_cli_client::BrainGeneralCliClient, common_proto::Pubkey, general_proto::brain_general_cli_client::BrainGeneralCliClient,
}; };
@ -12,6 +13,47 @@ use common::prepare_test_env::{
use common::test_utils::Key; use common::test_utils::Key;
use futures::StreamExt; use futures::StreamExt;
#[tokio::test]
async fn test_reg_vm_node() {
prepare_test_db().await;
let addr = run_service_in_background().await;
let mut client = BrainVmDaemonClient::connect(format!("http://{}", addr)).await.unwrap();
let operator_wallet = Key::new().pubkey;
let key = Key::new();
let node_pubkey = key.pubkey.clone();
let req = RegisterVmNodeReq {
node_pubkey,
operator_wallet,
main_ip: String::from("185.243.218.213"),
city: String::from("Oslo"),
country: String::from("Norway"),
region: String::from("EU"),
price: 1200,
};
let mut grpc_stream =
client.register_vm_node(key.sign_request(req).unwrap()).await.unwrap().into_inner();
let mut vm_contracts = Vec::new();
while let Some(stream_update) = grpc_stream.next().await {
match stream_update {
Ok(vm_c) => {
vm_contracts.push(vm_c);
}
Err(e) => {
panic!("Received error instead of vm_contracts: {e:?}");
}
}
}
assert!(vm_contracts.is_empty())
}
#[tokio::test] #[tokio::test]
async fn test_general_balance() { async fn test_general_balance() {
// env_logger::builder().filter_level(log::LevelFilter::Trace).init(); // env_logger::builder().filter_level(log::LevelFilter::Trace).init();