From 93588a650a1b77c7277502de861424bfdaafed33 Mon Sep 17 00:00:00 2001
From: Justus Winter <justus@sequoia-pgp.org>
Date: Wed, 6 Oct 2021 14:31:57 +0200
Subject: openpgp: Generalize test to all curves.

---
 openpgp/src/packet/signature.rs | 29 ++++++++++++++++++++---------
 1 file changed, 20 insertions(+), 9 deletions(-)

(limited to 'openpgp/src/packet')

diff --git a/openpgp/src/packet/signature.rs b/openpgp/src/packet/signature.rs
index 0b8cde02..b5adf702 100644
--- a/openpgp/src/packet/signature.rs
+++ b/openpgp/src/packet/signature.rs
@@ -3367,17 +3367,28 @@ mod test {
 
     #[test]
     fn sign_message() {
-        use crate::types::Curve;
+        use crate::types::Curve::*;
 
-        let key: Key<key::SecretParts, key::PrimaryRole>
-            = Key4::generate_ecc(true, Curve::Ed25519)
-            .unwrap().into();
-        let msg = b"Hello, World";
-        let mut pair = key.into_keypair().unwrap();
-        let mut sig = SignatureBuilder::new(SignatureType::Binary)
-            .sign_message(&mut pair, msg).unwrap();
+        for curve in vec![
+            Ed25519,
+            NistP256,
+            NistP384,
+            NistP521,
+        ] {
+            if ! curve.is_supported() {
+                eprintln!("Skipping unsupported {:?}", curve);
+                continue;
+            }
 
-        sig.verify_message(pair.public(), msg).unwrap();
+            let key: Key<key::SecretParts, key::PrimaryRole>
+                = Key4::generate_ecc(true, curve).unwrap().into();
+            let msg = b"Hello, World";
+            let mut pair = key.into_keypair().unwrap();
+            let mut sig = SignatureBuilder::new(SignatureType::Binary)
+                .sign_message(&mut pair, msg).unwrap();
+
+            sig.verify_message(pair.public(), msg).unwrap();
+        }
     }
 
     #[test]
-- 
cgit v1.2.3