test register vm node
This commit is contained in:
parent
c424d07cc3
commit
0359de54fe
@ -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();
|
||||
|
Loading…
Reference in New Issue
Block a user