vendor/der/tests/set_of.rs

   1 //! `SetOf` tests.
   2
   3 #![cfg(feature = "alloc")]
   4
   5 use der::{asn1::SetOfVec, DerOrd};
   6 use proptest::{prelude::*, string::*};
   7 use std::collections::BTreeSet;
   8
   9 proptest! {
  10     #[test]
  11     fn sort_equiv(bytes in bytes_regex(".{0,64}").unwrap()) {
  12         let mut uniq = BTreeSet::new();
  13
  14         // Ensure there are no duplicates
  15         if bytes.iter().copied().all(move |x| uniq.insert(x)) {
  16             let mut expected = bytes.clone();
  17             expected.sort_by(|a, b| a.der_cmp(b).unwrap());
  18
  19             let set = SetOfVec::try_from(bytes).unwrap();
  20             prop_assert_eq!(expected.as_slice(), set.as_slice());
  21         }
  22     }
  23 }
  24
  25 /// Set ordering tests.
  26 #[cfg(all(feature = "derive", feature = "oid"))]
  27 mod ordering {
  28     use der::{
  29         asn1::{AnyRef, ObjectIdentifier, SetOf, SetOfVec},
  30         Decode, Sequence, ValueOrd,
  31     };
  32     use hex_literal::hex;
  33
  34     /// X.501 `AttributeTypeAndValue`
  35     #[derive(Copy, Clone, Debug, Eq, PartialEq, Sequence, ValueOrd)]
  36     pub struct AttributeTypeAndValue<'a> {
  37         pub oid: ObjectIdentifier,
  38         pub value: AnyRef<'a>,
  39     }
  40
  41     const OUT_OF_ORDER_RDN_EXAMPLE: &[u8] =
  42         &hex!("311F301106035504030C0A4A4F484E20534D495448300A060355040A0C03313233");
  43
  44     /// For compatibility reasons, we allow non-canonical DER with out-of-order
  45     /// sets in order to match the behavior of other implementations.
  46     #[test]
  47     fn allow_out_of_order_setof() {
  48         assert!(SetOf::<AttributeTypeAndValue<'_>, 2>::from_der(OUT_OF_ORDER_RDN_EXAMPLE).is_ok());
  49     }
  50
  51     /// Same as above, with `SetOfVec` instead of `SetOf`.
  52     #[test]
  53     fn allow_out_of_order_setofvec() {
  54         assert!(SetOfVec::<AttributeTypeAndValue<'_>>::from_der(OUT_OF_ORDER_RDN_EXAMPLE).is_ok());
  55     }
  56
  57     /// Test to ensure ordering is handled correctly.
  58     #[test]
  59     fn ordering_regression() {
  60         let der_bytes = hex!("3139301906035504030C12546573742055736572393031353734333830301C060A0992268993F22C640101130E3437303031303030303134373333");
  61         let set = SetOf::<AttributeTypeAndValue<'_>, 3>::from_der(&der_bytes).unwrap();
  62         let attr1 = set.get(0).unwrap();
  63         assert_eq!(ObjectIdentifier::new("2.5.4.3").unwrap(), attr1.oid);
  64     }
  65 }