LNVPS/api
1
0
Fork 0
Code Issues Pull Requests Activity

chore: add more tests

Browse Source
This commit is contained in:
kieran
2025-02-28 12:48:48 +00:00
parent 5e2088f09c
commit f105b90849
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/provisioner/network.rs
View File

@ -103,9 +103,15 @@ mod tests {
db.clone(), db.clone(),
); );
let first = IpAddr::from_str("10.0.0.1").unwrap();
let second = IpAddr::from_str("10.0.0.2").unwrap();
let ip = mgr.pick_ip_for_region(1).await.expect("No ip found in db"); let ip = mgr.pick_ip_for_region(1).await.expect("No ip found in db");
assert_eq!(1, ip.region_id); assert_eq!(1, ip.region_id);
assert_eq!(IpAddr::from_str("10.0.0.1").unwrap(), ip.ip); assert_eq!(first, ip.ip);
let ip = mgr.pick_ip_for_region(1).await.expect("No ip found in db");
assert_eq!(1, ip.region_id);
assert_eq!(first, ip.ip);
db.insert_vm_ip_assignment(&VmIpAssignment { db.insert_vm_ip_assignment(&VmIpAssignment {
id: 0, id: 0,
vm_id: 0, vm_id: 0,
@ -116,7 +122,7 @@ mod tests {
.await .await
.expect("Could not insert vm ip"); .expect("Could not insert vm ip");
let ip = mgr.pick_ip_for_region(1).await.expect("No ip found in db"); let ip = mgr.pick_ip_for_region(1).await.expect("No ip found in db");
assert_eq!(IpAddr::from_str("10.0.0.2").unwrap(), ip.ip); assert_eq!(second, ip.ip);
} }
#[tokio::test] #[tokio::test]