tests: Verify converting generic ParameterSet to specific ones

Signed-off-by: Jakub Jelen <[email protected]>

Author: Jakuje

cryptoki/tests/ml_dsa.rs

@@ -7,7 +7,7 @@ use cryptoki::context::Function;
 use cryptoki::error::{Error, RvError};
 use cryptoki::mechanism::mldsa::{HashSignAdditionalContext, HedgeType, SignAdditionalContext};
 use cryptoki::mechanism::{Mechanism, MechanismType};
-use cryptoki::object::{Attribute, MlDsaParameterSetType};
+use cryptoki::object::{Attribute, AttributeType, MlDsaParameterSetType};
 use cryptoki::session::UserType;
 use cryptoki::types::AuthPin;
 use serial_test::serial;
@@ -100,6 +100,17 @@ fn ml_dsa() -> TestResult {
         Error::Pkcs11(_, Function::VerifySignature)
     ));
 
+    // Test convrting ParameterSet attributes back to algorithm specific values
+    let param_attribute = session
+        .get_attributes(public, &[AttributeType::ParameterSet])?
+        .remove(0);
+    let param: MlDsaParameterSetType = if let Attribute::ParameterSet(num) = param_attribute {
+        num.into()
+    } else {
+        panic!("Expected ParameterSet attribute.");
+    };
+    assert_eq!(param, MlDsaParameterSetType::ML_DSA_65);
+
     // delete keys
     session.destroy_object(public)?;
     session.destroy_object(private)?;

cryptoki/tests/ml_kem.rs

@@ -95,6 +95,17 @@ fn ml_kem() -> TestResult {
     let ptext = session.decrypt(&Mechanism::AesEcb, secret, &ctext)?;
     assert_eq!(data, ptext);
 
+    // Test convrting ParameterSet attributes back to algorithm specific values
+    let param_attribute = session
+        .get_attributes(public, &[AttributeType::ParameterSet])?
+        .remove(0);
+    let param: MlKemParameterSetType = if let Attribute::ParameterSet(num) = param_attribute {
+        num.into()
+    } else {
+        panic!("Expected ParameterSet attribute.");
+    };
+    assert_eq!(param, MlKemParameterSetType::ML_KEM_768);
+
     // delete keys
     session.destroy_object(public)?;
     session.destroy_object(private)?;