test(networking): add illustrative tests for various node fullness le…

…vels
maidsafe · Apr 2, 2024 · 7cd0861 · 7cd0861
1 parent c490399
commit 7cd0861
Showing 1 changed file with 55 additions and 2 deletions.
diff --git a/sn_networking/src/record_store.rs b/sn_networking/src/record_store.rs
@@ -775,11 +775,64 @@ mod tests {
     }
 
     #[test]
-    fn test_calculate_cost_for_records() {
-        let sut = calculate_cost_for_records(2049, 2050, 2048, 1);
+    fn test_calculate_max_cost_for_records() {
+        let sut = calculate_cost_for_records(2049, 2049, 2048, 1);
         assert_eq!(sut, TOTAL_SUPPLY / CLOSE_GROUP_SIZE as u64);
     }
 
+    #[test]
+    fn test_calculate_50_percent_cost_for_records() {
+        let percent = MAX_RECORDS_COUNT * 50 / 100;
+        let sut = calculate_cost_for_records(percent, percent, 2048, 1);
+        // at this point we should be at max cost
+        assert_eq!(sut, 10240);
+    }
+    #[test]
+    fn test_calculate_60_percent_cost_for_records() {
+        let percent = MAX_RECORDS_COUNT * 60 / 100;
+        let sut = calculate_cost_for_records(percent, percent, 2048, 1);
+        // at this point we should be at max cost
+        assert_eq!(sut, 12280);
+    }
+
+    #[test]
+    fn test_calculate_65_percent_cost_for_records() {
+        let percent = MAX_RECORDS_COUNT * 65 / 100;
+        let sut = calculate_cost_for_records(percent, percent, 2048, 1);
+        // at this point we should be at max cost
+        assert_eq!(sut, 2023120);
+    }
+
+    #[test]
+    fn test_calculate_70_percent_cost_for_records() {
+        let percent = MAX_RECORDS_COUNT * 70 / 100;
+        let sut = calculate_cost_for_records(percent, percent, 2048, 1);
+        // at this point we should be at max cost
+        assert_eq!(sut, 316248770);
+    }
+
+    #[test]
+    fn test_calculate_80_percent_cost_for_records() {
+        let percent = MAX_RECORDS_COUNT * 80 / 100;
+        let sut = calculate_cost_for_records(percent, percent, 2048, 1);
+        // at this point we should be at max cost
+        assert_eq!(sut, 7978447975680);
+    }
+
+    #[test]
+    fn test_calculate_90_percent_cost_for_records() {
+        let percent = MAX_RECORDS_COUNT * 90 / 100;
+        let sut = calculate_cost_for_records(percent, percent, 2048, 1);
+        // at this point we should be at max cost
+        assert_eq!(sut, 198121748221132800);
+    }
+
+    #[test]
+    fn test_calculate_min_cost_for_records() {
+        let sut = calculate_cost_for_records(0, 0, 2048, 1);
+        assert_eq!(sut, 10);
+    }
+
     #[test]
     fn put_get_remove_record() {
         fn prop(r: ArbitraryRecord) {