Skip to content

Commit e9b6fc3

Browse files
committed
improved tests to run for all KEM variants
1 parent debaa45 commit e9b6fc3

1 file changed

Lines changed: 4 additions & 7 deletions

File tree

src/test/java/de/hsheilbronn/mi/utils/crypto/hpke/HpkeTest.java

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -73,8 +73,7 @@ Arguments toArguments()
7373
private static Stream<Arguments> forTestEncryptDecryptInputStream() throws KeyNotFoundException
7474
{
7575
List<Mode> modes = List.of(Mode.base(), Mode.psk(PSK_ID));
76-
KemId[] kemIds = EnumSet.of(KemId.RSAKEM_1024_KDF2_SHA256, KemId.RSAKEM_2048_KDF2_SHA256,
77-
KemId.RSAKEM_3072_KDF2_SHA512, KemId.RSAKEM_4096_KDF2_SHA512).toArray(KemId[]::new);
76+
KemId[] kemIds = KemId.values();
7877
KdfId[] kdfIds = KdfId.values();
7978
AeadId[] aeadIds = AeadId.values();
8079
ChunkLength[] chunkLengths = ChunkLength.values();
@@ -163,16 +162,15 @@ else if (i > 2048 + 16 + 16)
163162
fail("Truncated by " + i, e);
164163
};
165164

166-
return Stream.of(AeadId.values()).map(aeadId ->
165+
return Stream.of(AeadId.values()).flatMap(aeadId -> Stream.of(KemId.values()).map(kemId ->
167166
{
168-
KemId kemId = KemId.RSAKEM_1024_KDF2_SHA256;
169167
KeyPair keyPair = kemId.getKeyPairGeneratorFactory().initialize().generateKeyPair();
170168
ProtocolV1 header = new ProtocolV1(handleException(() -> Mode.psk(PSK_ID)), kemId, KdfId.HKDF_SHA256,
171169
aeadId, ChunkLength.KiB_1, RECEIVER_KEY_ID);
172170
ProtocolAndKeyPair hkp = new ProtocolAndKeyPair(header, keyPair);
173171

174172
return Arguments.of(hkp, AeadId.ChaCha20Poly1305.equals(aeadId) ? chaCha20ErrorHandler : aesErrorHandler);
175-
});
173+
}));
176174
}
177175

178176
public interface SupplierWithException<T>
@@ -230,8 +228,7 @@ void decryptionTruncatedStreamTest(ProtocolAndKeyPair headerAndKeyPair,
230228
private static Stream<Arguments> forTestModifiedMessages() throws KeyNotFoundException
231229
{
232230
List<Mode> modes = List.of(Mode.base(), Mode.psk(PSK_ID));
233-
KemId[] kemIds = EnumSet.of(KemId.RSAKEM_1024_KDF2_SHA256, KemId.RSAKEM_2048_KDF2_SHA256,
234-
KemId.RSAKEM_3072_KDF2_SHA512, KemId.RSAKEM_4096_KDF2_SHA512).toArray(KemId[]::new);
231+
KemId[] kemIds = KemId.values();
235232
KdfId[] kdfIds = KdfId.values();
236233
AeadId[] aeadIds = AeadId.values();
237234
ChunkLength[] chunkLengths = ChunkLength.values();

0 commit comments

Comments
 (0)