Update tests for coverage

nealfennimore · May 30, 2023 · 7fbcb35 · 7fbcb35
1 parent 365e84d
commit 7fbcb35
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/src/aes/__tests__/aes_gcm.test.ts b/src/aes/__tests__/aes_gcm.test.ts
@@ -132,11 +132,10 @@ describe("AES_GCM", () => {
             const wrappedKey = await kek.wrapKey("raw", dek.self, {
                 iv,
             });
-            const unwrappedKey = (await AES.AES_GCM.unwrapKey(
+            const unwrappedKey = (await kek.unwrapKey(
                 "raw",
                 wrappedKey,
                 { name: AES.Alg.Mode.AES_GCM },
-                kek.self,
                 { iv }
             )) as AES.AesGcmCryptoKey;
 

diff --git a/src/rsa/__tests__/rsa_oaep.test.ts b/src/rsa/__tests__/rsa_oaep.test.ts
@@ -97,20 +97,25 @@ describe("RSA_OAEP", () => {
 
         it("should import and export key", async () => {
             const keyPair = await RSA_OAEP.generateKey();
-            const jwk = await keyPair.publicKey.exportKey("jwk");
+            const pubJwk = await keyPair.publicKey.exportKey("jwk");
+            const privJwk = await keyPair.privateKey.exportKey("jwk");
             const pubKey = (await RSA_OAEP.importKey(
                 "jwk",
-                jwk,
+                pubJwk,
                 { hash: "SHA-512" },
                 false,
                 ["encrypt"]
             )) as RsaOaepProxiedPubCryptoKey;
             const text = encode("a message");
             const ciphertext = await pubKey.encrypt({ label }, text);
-            const plaintext = await keyPair.privateKey.decrypt(
-                { label },
-                ciphertext
-            );
+            const privKey = (await RSA_OAEP.importKey(
+                "jwk",
+                privJwk,
+                { hash: "SHA-512" },
+                false,
+                ["decrypt"]
+            )) as RSA.RsaOaepProxiedPrivCryptoKey;
+            const plaintext = await privKey.decrypt({ label }, ciphertext);
             expect(decode(plaintext)).toEqual(decode(text));
         });
         it("should encrypt and decrypt", async () => {