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