Clean up tests

rpm-rs · Jun 16, 2023 · 008e7f1 · 008e7f1
1 parent b484ef1
commit 008e7f1
Show file tree

Hide file tree

Showing 6 changed files with 171 additions and 229 deletions.
diff --git a/src/rpm/package.rs b/src/rpm/package.rs
@@ -9,11 +9,11 @@ use num_traits::FromPrimitive;
 
 use crate::{constants::*, errors::*, CompressionType};
 
-#[cfg(feature = "signature-meta")]
-use crate::{signature, Timestamp};
 #[cfg(feature = "signature-pgp")]
 use crate::signature::pgp::Verifier;
 #[cfg(feature = "signature-meta")]
+use crate::{signature, Timestamp};
+#[cfg(feature = "signature-meta")]
 use std::{fmt::Debug, io::Read};
 
 use super::headers::*;
@@ -117,7 +117,7 @@ impl RPMPackage {
     /// # Examples
     /// ```
     /// # fn main() -> Result<(), Box<dyn std::error::Error>> {
-    /// let mut package = rpm::RPMPackage::open("test_assets/monkeysphere-0.37-1.el7.noarch.rpm")?;
+    /// let mut package = rpm::RPMPackage::open("test_assets/ima_signed.rpm")?;
     /// let raw_secret_key = std::fs::read("./test_assets/secret_key.asc")?;
     /// let signer = rpm::signature::pgp::Signer::load_from_asc_bytes(&raw_secret_key)?;
     /// // It's recommended to use timestamp of last commit in your VCS
@@ -178,7 +178,7 @@ impl RPMPackage {
         Ok(())
     }
 
-    /// Return the key id (issuer) of the signature
+    /// Return the key id (issuer) of the signature as a hexadecimal string
     #[cfg(feature = "signature-pgp")]
     pub fn signature_key_id(&self) -> Result<Option<String>, RPMError> {
         let rsa_sig = &self
@@ -264,41 +264,6 @@ impl RPMPackage {
             verifier.verify(header_bytes.as_slice(), signature_header_only)?;
         }
 
-        // match verifier.algorithm() {
-        //     signature::AlgorithmType::RSA => {
-        //         if let Ok(signature_header_and_content) = rpm_v3_sig {
-        //             signature::echo_signature(
-        //                 "signature_header(header and content)",
-        //                 signature_header_and_content,
-        //             );
-        //             let header_and_content_cursor =
-        //                 io::Cursor::new(&header_bytes).chain(io::Cursor::new(&self.content));
-        //             verifier.verify(header_and_content_cursor, signature_header_and_content)?;
-        //         }
-
-        //         if let Ok(signature_header_only) = rsa_sig {
-        //             signature::echo_signature(
-        //                 "signature_header(header only)",
-        //                 signature_header_only,
-        //             );
-        //             verifier.verify(header_bytes.as_slice(), signature_header_only)?;
-        //         } else {
-        //             return Err(RPMError::VerificationError { source: (), key_ref: () })
-        //         }
-        //     }
-        //     signature::AlgorithmType::EdDSA => {
-        //         if let Ok(signature_header_only) = eddsa_sig {
-        //             signature::echo_signature(
-        //                 "signature_header(header only)",
-        //                 signature_header_only,
-        //             );
-        //             verifier.verify(header_bytes.as_slice(), signature_header_only)?;
-        //         } else {
-        //             return Err(RPMError::VerificationError { source: (), key_ref: () })
-        //         }
-        //     }
-        //}
-
         Ok(())
     }
 

diff --git a/src/rpm/signature/pgp.rs b/src/rpm/signature/pgp.rs
@@ -310,9 +310,9 @@ pub(crate) mod test {
 
     #[test]
     fn verify_pgp_crate() {
+        use chrono::{TimeZone, Utc};
         use pgp::types::{PublicKeyTrait, SecretKeyTrait};
         use pgp::Signature;
-        use chrono::{TimeZone, Utc};
 
         const RPM_SHA2_256: [u8; 32] =
             hex!("d92bfe276e311a67fe128768c5df4d06fd461e043afdf872ba4c679d860db81e");

diff --git a/src/rpm/signature/traits.rs b/src/rpm/signature/traits.rs
@@ -3,10 +3,10 @@
 //! Does not contain hashing! Hashes are fixed by the rpm
 //! "spec" to sha1, md5 (yes, that is correct), sha2_256.
 
-use std::io;
-use std::fmt::Debug;
 use crate::errors::*;
 use crate::Timestamp;
+use std::fmt::Debug;
+use std::io;
 
 #[derive(Clone, Copy, Debug)]
 pub enum AlgorithmType {

diff --git a/tests/common.rs b/tests/common.rs
@@ -39,15 +39,19 @@ pub fn load_asc_keys() -> (Vec<u8>, Vec<u8>) {
 }
 
 pub fn load_rsa_keys() -> (Vec<u8>, Vec<u8>) {
-    let signing_key_path =
+    (rsa_private_key(), rsa_public_key())
+}
+
+pub fn rsa_private_key() -> Vec<u8> {
+    let private_key =
         cargo_manifest_dir().join("test_assets/fixture_packages/signing_keys/secret_rsa4096.asc");
-    let signing_key = std::fs::read(signing_key_path).unwrap();
+    std::fs::read(private_key).unwrap()
+}
 
-    let verification_key_path =
+pub fn rsa_public_key() -> Vec<u8> {
+    let public_key =
         cargo_manifest_dir().join("test_assets/fixture_packages/signing_keys/public_rsa4096.asc");
-    let verification_key = std::fs::read(verification_key_path).unwrap();
-
-    (signing_key.to_vec(), verification_key.to_vec())
+    std::fs::read(public_key).unwrap()
 }
 
 pub fn eddsa_private_key() -> Vec<u8> {