From e9add9fd7a53b5198e5b9378d559a542908b8af5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Garillot?= <francois@garillot.net>
Date: Fri, 12 Jun 2020 18:16:02 -0400
Subject: [PATCH] add tests showing X25519KeyPair == SignalKeyPair

---
 src/keypair.rs | 62 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)

diff --git a/src/keypair.rs b/src/keypair.rs
index 4fb404f..a7fb3e7 100644
--- a/src/keypair.rs
+++ b/src/keypair.rs
@@ -306,4 +306,66 @@ impl KeyPair for SignalKeyPair {
 mod tests {
     use super::*;
     use proptest::prelude::*;
+
+    proptest! {
+        #[test]
+        fn test_x25519_check(kp in X25519KeyPair::uniform_keypair_strategy()) {
+            let pk = kp.public();
+            prop_assert!(X25519KeyPair::check_public_key(pk.clone()).is_ok());
+        }
+
+        #[test]
+        fn test_x25519_pub_from_priv(kp in X25519KeyPair::uniform_keypair_strategy()) {
+            let pk = kp.public();
+            let sk = kp.private();
+            prop_assert_eq!(&X25519KeyPair::public_from_private(sk), pk);
+        }
+
+        #[test]
+        fn test_signal_check(kp in SignalKeyPair::uniform_keypair_strategy()) {
+            let pk = kp.public();
+            prop_assert!(SignalKeyPair::check_public_key(pk.clone()).is_ok());
+        }
+
+        #[test]
+        fn test_signal_pub_from_priv(kp in SignalKeyPair::uniform_keypair_strategy()) {
+            let pk = kp.public();
+            let sk = kp.private();
+            prop_assert_eq!(&SignalKeyPair::public_from_private(sk), pk);
+        }
+
+        #[test]
+        fn test_signal_x25519_roundtrips(kp_signal in SignalKeyPair::uniform_keypair_strategy(),
+                                         kp_x25519 in X25519KeyPair::uniform_keypair_strategy()) {
+            let kp_signal_bytes: &[u8] = &kp_signal.to_arr();
+            let kp_x25519_bytes: &[u8] = &kp_x25519.to_arr();
+
+            let reinterpret_signal = X25519KeyPair::from_bytes(kp_signal_bytes).unwrap();
+            let reinterpret_x25519 = SignalKeyPair::from_bytes(kp_x25519_bytes).unwrap();
+
+            prop_assert_eq!(kp_signal_bytes, &reinterpret_signal.to_arr()[..]);
+            prop_assert_eq!(kp_x25519_bytes, &reinterpret_x25519.to_arr()[..]);
+        }
+
+        #[test]
+        fn test_signal_as_x25519(kp_signal in SignalKeyPair::uniform_keypair_strategy()) {
+            let kp: X25519KeyPair = X25519KeyPair::from_bytes(&kp_signal.to_arr()).unwrap();
+            let pk = kp.public();
+            prop_assert!(X25519KeyPair::check_public_key(pk.clone()).is_ok());
+            let sk = kp.private();
+            prop_assert_eq!(&X25519KeyPair::public_from_private(sk), pk);
+
+        }
+
+        #[test]
+        fn test_x25519_as_signal(kp_x25519 in X25519KeyPair::uniform_keypair_strategy()) {
+            let kp: SignalKeyPair = SignalKeyPair::from_bytes(&kp_x25519.to_arr()).unwrap();
+            let pk = kp.public();
+            prop_assert!(SignalKeyPair::check_public_key(pk.clone()).is_ok());
+            let sk = kp.private();
+            prop_assert_eq!(&SignalKeyPair::public_from_private(sk), pk);
+
+        }
+
+    }
 }