Compare commits
	
		
			4 Commits
		
	
	
		
			933e3cb893
			...
			a014622dbd
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| a014622dbd | |||
| 57d3807a17 | |||
| 929530a4c5 | |||
| 4c647eef6a | 
							
								
								
									
										14
									
								
								src/db/vm.rs
									
									
									
									
									
								
							
							
								
								
								
								
								
									
									
								
							
						
						
									
										14
									
								
								src/db/vm.rs
									
									
									
									
									
								
							| @ -239,18 +239,7 @@ impl NewVmReq { | ||||
|             self.price_per_unit | ||||
|         ); | ||||
|         //let _: Vec<Self> = db.insert(NEW_VM_REQ).relation(self).await?;
 | ||||
|         let mut query_resp = db.query(query).await?; | ||||
|         let resp_err = query_resp.take_errors(); | ||||
| 
 | ||||
|         if let Some(insufficient_funds_error) = resp_err.get(&1) { | ||||
|             if let surrealdb::Error::Api(surrealdb::error::Api::Query(tx_query_error)) = | ||||
|                 insufficient_funds_error | ||||
|             { | ||||
|                 log::error!("Transaction error: {}", tx_query_error); | ||||
|                 return Err(Error::InsufficientFunds); | ||||
|             } | ||||
|         } | ||||
| 
 | ||||
|         db.query(query).await?; | ||||
|         Ok(()) | ||||
|     } | ||||
| } | ||||
| @ -287,7 +276,6 @@ impl WrappedMeasurement { | ||||
|         if let Some(args) = args { | ||||
|             return Ok(Self::Args(vm_id.to_string(), args)); | ||||
|         } | ||||
|         log::trace!("listening for table: {table}"); | ||||
| 
 | ||||
|         tokio::time::timeout(Duration::from_secs(10), async { | ||||
|             loop { | ||||
|  | ||||
| @ -20,17 +20,11 @@ async fn test_vm_creation() { | ||||
| 
 | ||||
|     let key = Key::new(); | ||||
| 
 | ||||
|     let new_vm_resp = create_new_vm(&db, &key, &daemon_key, &brain_channel).await; | ||||
| 
 | ||||
|     assert!(new_vm_resp.is_err()); | ||||
|     let grpc_error_message = new_vm_resp.err().unwrap().to_string(); | ||||
|     assert!(grpc_error_message.contains("Insufficient funds")); | ||||
| 
 | ||||
|     // TODO: Airdrop the user and try creating the VM again
 | ||||
|     let _ = create_new_vm(&db, &key, &daemon_key, &brain_channel).await; | ||||
| } | ||||
| 
 | ||||
| #[tokio::test] | ||||
| async fn test_timeout_vm_creation() { | ||||
| async fn test_vm_creation_timeout() { | ||||
|     prepare_test_db().await.unwrap(); | ||||
|     // env_logger::builder().filter_level(log::LevelFilter::Error).init();
 | ||||
| 
 | ||||
|  | ||||
		Loading…
	
		Reference in New Issue
	
	Block a user