@@ -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