add: AES-KW + {AES-CTR, AES-CBC} tests

Author: kentakayama

tests/test_recipient.py

@@ -782,6 +782,108 @@ def test_recipients_open_with_invalid_rsk(self, rpk1):
             pytest.fail("decode() should fail.")
         assert "Failed to open." in str(err.value)
 
+    def test_recipients_a128ctr_aeskw(self, rsk1, rsk2):
+        a128kw_key = COSEKey.from_symmetric_key(alg="A128KW")
+        a128ctr_key = COSEKey.from_symmetric_key(alg="A128CTR", kid="A128CTR key")
+
+        # The sender side:
+        r = Recipient.new(unprotected={"alg": "A128KW"}, sender_key=a128kw_key)
+        sender = COSE.new(alg_auto_inclusion=True)
+        encoded = sender.encode_and_encrypt(
+            b"Hello world!",
+            a128ctr_key,
+            recipients=[r],
+        )
+
+        # The recipient side:
+        recipient = COSE.new()
+        assert b"Hello world!" == recipient.decode(encoded, keys=[a128kw_key])
+
+    def test_recipients_a192ctr_aeskw(self, rsk1, rsk2):
+        a192kw_key = COSEKey.from_symmetric_key(alg="A192KW")
+        a192ctr_key = COSEKey.from_symmetric_key(alg="A192CTR", kid="A192CTR key")
+
+        # The sender side:
+        r = Recipient.new(unprotected={"alg": "A192KW"}, sender_key=a192kw_key)
+        sender = COSE.new(alg_auto_inclusion=True)
+        encoded = sender.encode_and_encrypt(
+            b"Hello world!",
+            a192ctr_key,
+            recipients=[r],
+        )
+
+        # The recipient side:
+        recipient = COSE.new()
+        assert b"Hello world!" == recipient.decode(encoded, keys=[a192kw_key])
+
+    def test_recipients_a256_aeskw(self, rsk1, rsk2):
+        a256kw_key = COSEKey.from_symmetric_key(alg="A256KW")
+        a256ctr_key = COSEKey.from_symmetric_key(alg="A256CTR", kid="A256CTR key")
+
+        # The sender side:
+        r = Recipient.new(unprotected={"alg": "A256KW"}, sender_key=a256kw_key)
+        sender = COSE.new(alg_auto_inclusion=True)
+        encoded = sender.encode_and_encrypt(
+            b"Hello world!",
+            a256ctr_key,
+            recipients=[r],
+        )
+
+        # The recipient side:
+        recipient = COSE.new()
+        assert b"Hello world!" == recipient.decode(encoded, keys=[a256kw_key])
+
+    def test_recipients_a128cbc_aeskw(self, rsk1, rsk2):
+        a128kw_key = COSEKey.from_symmetric_key(alg="A128KW")
+        a128cbc_key = COSEKey.from_symmetric_key(alg="A128CBC", kid="A128CBC key")
+
+        # The sender side:
+        r = Recipient.new(unprotected={"alg": "A128KW"}, sender_key=a128kw_key)
+        sender = COSE.new(alg_auto_inclusion=True)
+        encoded = sender.encode_and_encrypt(
+            b"Hello world!",
+            a128cbc_key,
+            recipients=[r],
+        )
+
+        # The recipient side:
+        recipient = COSE.new()
+        assert b"Hello world!" == recipient.decode(encoded, keys=[a128kw_key])
+
+    def test_recipients_a192cbc_aeskw(self, rsk1, rsk2):
+        a192kw_key = COSEKey.from_symmetric_key(alg="A192KW")
+        a192cbc_key = COSEKey.from_symmetric_key(alg="A192CBC", kid="A192CBC key")
+
+        # The sender side:
+        r = Recipient.new(unprotected={"alg": "A192KW"}, sender_key=a192kw_key)
+        sender = COSE.new(alg_auto_inclusion=True)
+        encoded = sender.encode_and_encrypt(
+            b"Hello world!",
+            a192cbc_key,
+            recipients=[r],
+        )
+
+        # The recipient side:
+        recipient = COSE.new()
+        assert b"Hello world!" == recipient.decode(encoded, keys=[a192kw_key])
+
+    def test_recipients_a256cbc_aeskw(self, rsk1, rsk2):
+        a256kw_key = COSEKey.from_symmetric_key(alg="A256KW")
+        a256cbc_key = COSEKey.from_symmetric_key(alg="A256CBC", kid="A256CBC key")
+
+        # The sender side:
+        r = Recipient.new(unprotected={"alg": "A256KW"}, sender_key=a256kw_key)
+        sender = COSE.new(alg_auto_inclusion=True)
+        encoded = sender.encode_and_encrypt(
+            b"Hello world!",
+            a256cbc_key,
+            recipients=[r],
+        )
+
+        # The recipient side:
+        recipient = COSE.new()
+        assert b"Hello world!" == recipient.decode(encoded, keys=[a256kw_key])
+
     def test_recipients_a128ctr_hpke(self, rsk1, rsk2):
         enc_key = COSEKey.from_symmetric_key(alg="A128CTR")
         rpk = COSEKey.from_jwk(