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 |             self.price_per_unit | ||||||
|         ); |         ); | ||||||
|         //let _: Vec<Self> = db.insert(NEW_VM_REQ).relation(self).await?;
 |         //let _: Vec<Self> = db.insert(NEW_VM_REQ).relation(self).await?;
 | ||||||
|         let mut query_resp = db.query(query).await?; |         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); |  | ||||||
|             } |  | ||||||
|         } |  | ||||||
| 
 |  | ||||||
|         Ok(()) |         Ok(()) | ||||||
|     } |     } | ||||||
| } | } | ||||||
| @ -287,7 +276,6 @@ impl WrappedMeasurement { | |||||||
|         if let Some(args) = args { |         if let Some(args) = args { | ||||||
|             return Ok(Self::Args(vm_id.to_string(), args)); |             return Ok(Self::Args(vm_id.to_string(), args)); | ||||||
|         } |         } | ||||||
|         log::trace!("listening for table: {table}"); |  | ||||||
| 
 | 
 | ||||||
|         tokio::time::timeout(Duration::from_secs(10), async { |         tokio::time::timeout(Duration::from_secs(10), async { | ||||||
|             loop { |             loop { | ||||||
|  | |||||||
| @ -20,17 +20,11 @@ async fn test_vm_creation() { | |||||||
| 
 | 
 | ||||||
|     let key = Key::new(); |     let key = Key::new(); | ||||||
| 
 | 
 | ||||||
|     let new_vm_resp = create_new_vm(&db, &key, &daemon_key, &brain_channel).await; |     let _ = 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
 |  | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| #[tokio::test] | #[tokio::test] | ||||||
| async fn test_timeout_vm_creation() { | async fn test_vm_creation_timeout() { | ||||||
|     prepare_test_db().await.unwrap(); |     prepare_test_db().await.unwrap(); | ||||||
|     // env_logger::builder().filter_level(log::LevelFilter::Error).init();
 |     // env_logger::builder().filter_level(log::LevelFilter::Error).init();
 | ||||||
| 
 | 
 | ||||||
|  | |||||||
		Loading…
	
		Reference in New Issue
	
	Block a user