1 files changed, 27 insertions, 0 deletions
diff --git a/src/wireguard/types/keys.rs b/src/wireguard/types.rs
index 282c4ae..51898a0 100644
--- a/src/wireguard/types/keys.rs
+++ b/src/wireguard/types.rs
@@ -1,6 +1,33 @@
 use clear_on_drop::clear::Clear;
 use std::time::Instant;
 
+#[cfg(test)]
+pub fn dummy_keypair(initiator: bool) -> KeyPair {
+    let k1 = Key {
+        key: [0x53u8; 32],
+        id: 0x646e6573,
+    };
+    let k2 = Key {
+        key: [0x52u8; 32],
+        id: 0x76636572,
+    };
+    if initiator {
+        KeyPair {
+            birth: Instant::now(),
+            initiator: true,
+            send: k1,
+            recv: k2,
+        }
+    } else {
+        KeyPair {
+            birth: Instant::now(),
+            initiator: false,
+            send: k2,
+            recv: k1,
+        }
+    }
+}
+
 #[derive(Debug, Clone)]
 pub struct Key {
     pub key: [u8; 32],