From 105c0e44cd1202145d3db7047dd8615a5962cc39 Mon Sep 17 00:00:00 2001
From: KOSASIH <kosasihg88@gmail.com>
Date: Sun, 8 Dec 2024 20:40:04 +0700
Subject: [PATCH] Create security_tests.rs

---
 src/security/security_tests.rs | 36 ++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 src/security/security_tests.rs

diff --git a/src/security/security_tests.rs b/src/security/security_tests.rs
new file mode 100644
index 0000000..a1b9ee8
--- /dev/null
+++ b/src/security/security_tests.rs
@@ -0,0 +1,36 @@
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use ed25519_dalek::{Keypair, Signature, Signer};
+    use std::collections::HashMap;
+
+    #[test]
+    fn test_multi_sig_wallet() {
+        let mut wallet = MultiSigWallet::new(2, vec![
+            Keypair::generate(&mut rand::thread_rng()).public,
+            Keypair::generate(&mut rand::thread_rng()).public,
+        ]);
+
+        let message = b"Hello, world!";
+        let keypair = Keypair::generate(&mut rand::thread_rng());
+        let signature: Signature = keypair.sign(message);
+
+        wallet.add_signature(keypair.public, signature);
+        assert!(!wallet.verify(message));
+
+        let keypair2 = Keypair::generate(&mut rand::thread_rng());
+        let signature2: Signature = keypair2.sign(message);
+
+        wallet.add_signature(keypair2.public, signature2);
+        assert!(wallet.verify(message));
+    }
+
+    #[test]
+    fn test_encryption() {
+        let key = rand::thread_rng().gen::<[u8; 32]>();
+        let data = b"Hello, world!";
+        let encrypted_data = encrypt(data, &key);
+        let decrypted_data = decrypt(&encrypted_data, &key);
+        assert_eq!(data, &decrypted_data[..]);
+    }
+}