diff --git a/tests/grpcs_test.rs b/tests/grpcs_test.rs index ef93e13..917bf22 100644 --- a/tests/grpcs_test.rs +++ b/tests/grpcs_test.rs @@ -1,7 +1,8 @@ use detee_shared::common_proto::Empty; use detee_shared::general_proto::ReportNodeReq; 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::{ 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 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] async fn test_general_balance() { // env_logger::builder().filter_level(log::LevelFilter::Trace).init();