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::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();
|
||||||
|
Loading…
Reference in New Issue
Block a user