Implement Debug for OpeningKey and SealingKey.

briansmith · Jan 24, 2019 · abbb616 · abbb616
1 parent c156075
commit abbb616
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/src/aead.rs b/src/aead.rs
@@ -38,6 +38,8 @@ pub struct OpeningKey {
     key: Key,
 }
 
+derive_debug_via_field!(OpeningKey, key);
+
 impl OpeningKey {
     /// Create a new opening key.
     ///
@@ -148,6 +150,8 @@ pub struct SealingKey {
     key: Key,
 }
 
+derive_debug_via_field!(SealingKey, key);
+
 impl SealingKey {
     /// Constructs a new sealing key from `key_bytes`.
     #[inline]
@@ -236,6 +240,8 @@ struct Key {
     algorithm: &'static Algorithm,
 }
 
+derive_debug_via_field!(Key, algorithm);
+
 #[allow(variant_size_differences)]
 enum KeyInner {
     AesGcm(aes_gcm::Key),

diff --git a/tests/aead_tests.rs b/tests/aead_tests.rs
@@ -300,3 +300,20 @@ fn aead_chacha20_poly1305_openssh() {
         },
     );
 }
+
+#[test]
+fn test_aead_key_debug() {
+    let key_bytes = [0; 32];
+
+    let key = aead::OpeningKey::new(&aead::AES_256_GCM, &key_bytes).unwrap();
+    assert_eq!(
+        "OpeningKey { key: Key { algorithm: AES_256_GCM } }",
+        format!("{:?}", key)
+    );
+
+    let key = aead::SealingKey::new(&aead::CHACHA20_POLY1305, &key_bytes).unwrap();
+    assert_eq!(
+        "SealingKey { key: Key { algorithm: CHACHA20_POLY1305 } }",
+        format!("{:?}", key)
+    );
+}