Skip to content

Commit affdfc7

Browse files
committed
fix: tests moving to unprotected header for non AEAD cipher
1 parent f280719 commit affdfc7

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

tests/test_recipient.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -832,13 +832,13 @@ def test_recipients_hpke(self, rsk1, rsk2, enc_alg):
832832
"y": "BGU5soLgsu_y7GN2I3EPUXS9EZ7Sw0qif-V70JtInFI",
833833
}
834834
)
835-
r = Recipient.new(protected={1: 35}, recipient_key=rpk)
835+
r = Recipient.new(unprotected={1: 35}, recipient_key=rpk)
836836
r.encode(enc_key.key)
837837
sender = COSE.new()
838838
encoded = sender.encode_and_encrypt(
839839
b"This is the content.",
840840
enc_key,
841-
protected={"alg": enc_alg},
841+
unprotected={"alg": enc_alg},
842842
recipients=[r],
843843
)
844844
recipient = COSE.new()
@@ -861,7 +861,7 @@ def test_recipients_ecdh_es(self, key_agreement_alg, key_agreement_alg_id, kw_al
861861
"alg": kw_alg,
862862
"supp_pub": {
863863
"key_data_length": len(enc_key.key) * 8,
864-
"protected": {1: key_agreement_alg_id},
864+
"protected": {},
865865
},
866866
}
867867

@@ -886,15 +886,15 @@ def test_recipients_ecdh_es(self, key_agreement_alg, key_agreement_alg_id, kw_al
886886
"y": "BGU5soLgsu_y7GN2I3EPUXS9EZ7Sw0qif-V70JtInFI",
887887
}
888888
)
889-
r = Recipient.new(protected={"alg": key_agreement_alg}, sender_key=rsk1, recipient_key=rpk2, context=context)
889+
r = Recipient.new(unprotected={"alg": key_agreement_alg}, sender_key=rsk1, recipient_key=rpk2, context=context)
890890

891891
nonce = enc_key.generate_nonce()
892892
sender = COSE.new()
893893
encoded = sender.encode(
894894
b"Hello world!",
895895
enc_key,
896-
protected={"alg": enc_alg},
897-
unprotected={"iv": nonce},
896+
protected={},
897+
unprotected={"alg": enc_alg, "iv": nonce},
898898
recipients=[r],
899899
)
900900

0 commit comments

Comments
 (0)