From 7647564c1e51db6838efa5b8ef0b8caa6c77ee8a Mon Sep 17 00:00:00 2001
From: onyinyang <onyinyang@torproject.org>
Date: Wed, 2 Aug 2023 17:20:08 -0400
Subject: [PATCH] Add test for k-invites

---
 crates/lox-library/src/tests.rs | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/crates/lox-library/src/tests.rs b/crates/lox-library/src/tests.rs
index 08482aa..428fcf1 100644
--- a/crates/lox-library/src/tests.rs
+++ b/crates/lox-library/src/tests.rs
@@ -394,6 +394,21 @@ fn test_open_invite() {
     assert!(bridgeline == bucket.0[0]);
 }
 
+
+#[test]
+fn test_k_invites() {
+    let mut th = TestHarness::new();
+    for i in 0..25 {
+        let _ = th.open_invite();
+        if (i+1) % OPENINV_K != 0 {
+        assert!(th.bdb.current_k == (i+1)%OPENINV_K, "the current_k should be (i+1)%OPENINV_K");
+        } else {
+        assert!(th.bdb.current_k == OPENINV_K, "the current_k should be OPENINV_K");
+
+        }
+    }
+}
+
 #[test]
 fn test_trust_promotion() {
     let mut th = TestHarness::new();