From 84bb9595276b4ac52eb60b39b01de025bb35d2b9 Mon Sep 17 00:00:00 2001
From: Hatter Jiang <jht5945@gmail.com>
Date: Wed, 13 Sep 2023 00:57:31 +0800
Subject: [PATCH] feat: add test_parse_wrap_key

---
 src/wrap_key.rs | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/wrap_key.rs b/src/wrap_key.rs
index 4f6e885..2906a89 100644
--- a/src/wrap_key.rs
+++ b/src/wrap_key.rs
@@ -41,4 +41,20 @@ impl WrapKey {
             encrypted_data,
         })
     }
+}
+
+#[test]
+fn test_parse_wrap_key() {
+    let wk = "WK:eyJlUHViS2V5IjoiTUZrd0V3WUhLb1pJemowQ0FRWUlLb1pJemowREFRY0RRZ0FFcExLeXBzU2hKdl\
+    ZSeC0xaGZaRjJwOWN6SUZwanBfRExJTE1GOVo4VTlZcUZEVFpNZE5CQ3R5NFJsWG1JaEhaSUxVT1pMWW90bjR0QmF6WndnVk\
+    c3alEiLCJlbmMiOiJhZXMyNTYtZ2NtLXAyNTYifQ.bil863KUslf7nzHs.VR24eaonTSZnHs8hWp4QP-5RjFcZH3i7V79DiZ\
+    dHuCnxyywfw4daWuJzYgouxCBE";
+
+    let wrap_key = WrapKey::parse(wk).unwrap();
+    assert_eq!("aes256-gcm-p256", wrap_key.header.enc);
+    assert_eq!("MFkwEwYHKoZIzj0CAQYIKoZIzj0DAQcDQgAEpLKypsShJvVRx-1hfZF2p9czIFpjp_DLILMF9Z8U9YqFDTZMd\
+    NBCty4RlXmIhHZILUOZLYotn4tBazZwgVG7jQ", wrap_key.header.e_pub_key);
+    assert_eq!("6e297ceb7294b257fb9f31ec", hex::encode(&wrap_key.nonce));
+    assert_eq!("551db879aa274d26671ecf215a9e103fee518c57191f78bb57bf43899747b829f1cb2c1fc3875a5ae273\
+    620a2ec42044", hex::encode(&wrap_key.encrypted_data));
 }
\ No newline at end of file