rsa digital signature tests

Author: WingZer0o

tests/digital_signatures.rs

@@ -1,6 +1,6 @@
 #[cfg(test)]
 mod digital_signatures {
-    use cas_lib::digital_signature::{cas_digital_signature_rsa::{ED25519DigitalSignature, SHAED25519DalekDigitalSignatureResult}, sha_256_ed25519::SHA256ED25519DigitalSignature, sha_512_ed25519::SHA512ED25519DigitalSignature};
+    use cas_lib::{asymmetric::cas_rsa::CASRSA, digital_signature::{cas_digital_signature_rsa::{ED25519DigitalSignature, RSADigitalSignature, RSADigitalSignatureResult, SHAED25519DalekDigitalSignatureResult}, sha_256_ed25519::SHA256ED25519DigitalSignature, sha_256_rsa::SHA256RSADigitalSignature, sha_512_ed25519::SHA512ED25519DigitalSignature, sha_512_rsa::SHA512RSADigitalSignature}};
 
     #[test]
     pub fn ed25519_sha_512_digital_signature_verify() {
@@ -33,4 +33,36 @@ mod digital_signatures {
         let verification = <SHA256ED25519DigitalSignature as ED25519DigitalSignature>::digital_signature_ed25519_verify_threadpool(result.public_key, data_to_sign, result.signature);
         assert_eq!(true, verification);
     }
+
+    #[test]
+    pub fn rsa_sha_512_digital_signature_verify() {
+        let data_to_sign = b"This is a test of a digital signature".to_vec();
+        let result: RSADigitalSignatureResult = SHA512RSADigitalSignature::digital_signature_rsa(2048, data_to_sign.clone());
+        let verification = SHA512RSADigitalSignature::verify_rsa(result.public_key, data_to_sign, result.signature);
+        assert_eq!(true, verification);
+    }
+
+    #[test]
+    pub fn rsa_sha_512_digital_signature_threadpool_verify() {
+        let data_to_sign = b"This is a test of a digital signature".to_vec();
+        let result: RSADigitalSignatureResult = SHA512RSADigitalSignature::digital_signature_rsa_threadpool(4096, data_to_sign.clone());
+        let verification = SHA512RSADigitalSignature::verify_rsa_threadpool(result.public_key, data_to_sign, result.signature);
+        assert_eq!(true, verification);
+    }
+
+    #[test]
+    pub fn rsa_sha_256_digital_signature_verify() {
+        let data_to_sign = b"This is a test of a digital signature".to_vec();
+        let result: RSADigitalSignatureResult = SHA256RSADigitalSignature::digital_signature_rsa(2048, data_to_sign.clone());
+        let verification = SHA256RSADigitalSignature::verify_rsa(result.public_key, data_to_sign, result.signature);
+        assert_eq!(true, verification);
+    }
+
+    #[test]
+    pub fn rsa_sha_256_digital_signature_threadpool_verify() {
+        let data_to_sign = b"This is a test of a digital signature".to_vec();
+        let result: RSADigitalSignatureResult = SHA256RSADigitalSignature::digital_signature_rsa_threadpool(4096, data_to_sign.clone());
+        let verification = SHA256RSADigitalSignature::verify_rsa_threadpool(result.public_key, data_to_sign, result.signature);
+        assert_eq!(true, verification);
+    }
 }