@@ -335,26 +335,16 @@ public static Set<Arguments> identityMapBigBatchArgs() throws JsonProcessingExce
335335 List <String > emailHashes = new ArrayList <>();
336336 List <String > phoneHashes = new ArrayList <>();
337337 for (int i = 0 ; i < 10_000 ; i ++) {
338- emails .add ("email_" + Math .abs (RANDOM .nextLong ()) + "@example.com" );
339-
340- // Phone numbers with 15 digits are technically valid but are not used in any country
341- phones .add ("+" + String .format ("%015d" , Math .abs (RANDOM .nextLong () % 1_000_000_000_000_000L )));
342-
343- emailHashes .add (fakeRandomHash ());
344- phoneHashes .add (fakeRandomHash ());
338+ emails .add (random_email ());
339+ phones .add (randomPhoneNumber ());
340+ emailHashes .add (randomHash ());
341+ phoneHashes .add (randomHash ());
345342 }
346343
347- var emailsJson = mapper .createObjectNode ().putPOJO ("email" , emails ).put ("policy" , 1 );
348- String emailsPayload = mapper .writeValueAsString (emailsJson );
349-
350- var phonesJson = mapper .createObjectNode ().putPOJO ("phone" , phones ).put ("policy" , 1 );
351- String phonesPayload = mapper .writeValueAsString (phonesJson );
352-
353- var emailHashesJson = mapper .createObjectNode ().putPOJO ("email_hash" , emailHashes ).put ("policy" , 1 );
354- String emailHashesPayload = mapper .writeValueAsString (emailHashesJson );
355-
356- var phoneHashesJson = mapper .createObjectNode ().putPOJO ("phone_hash" , phoneHashes ).put ("policy" , 1 );
357- String phoneHashesPayload = mapper .writeValueAsString (phoneHashesJson );
344+ var emailsPayload = identityMapPayload (mapper , "email" , emails );
345+ var phonesPayload = identityMapPayload (mapper , "phone" , phones );
346+ var emailHashesPayload = identityMapPayload (mapper , "email_hash" , emailHashes );
347+ var phoneHashesPayload = identityMapPayload (mapper , "phone_hash" , phoneHashes );
358348
359349 Set <Arguments > args = new HashSet <>();
360350 for (Operator operator : operators ) {
@@ -366,7 +356,22 @@ public static Set<Arguments> identityMapBigBatchArgs() throws JsonProcessingExce
366356 return args ;
367357 }
368358
369- private static String fakeRandomHash () {
359+ private static String identityMapPayload (ObjectMapper mapper , String field , List <String > diis ) throws JsonProcessingException {
360+ var json = mapper .createObjectNode ().putPOJO (field , diis ).put ("policy" , 1 );
361+ return mapper .writeValueAsString (json );
362+ }
363+
364+ private static String random_email () {
365+ return "email_" + Math .abs (RANDOM .nextLong ()) + "@example.com" ;
366+ }
367+
368+ private static String randomPhoneNumber () {
369+ // Phone numbers with 15 digits are technically valid but are not used in any country
370+ return "+" + String .format ("%015d" , Math .abs (RANDOM .nextLong () % 1_000_000_000_000_000L ));
371+ }
372+
373+ private static String randomHash () {
374+ // This isn't really a hashed DII but looks like one ot UID2
370375 byte [] randomBytes = new byte [32 ];
371376 RANDOM .nextBytes (randomBytes );
372377 return Base64 .getEncoder ().encodeToString (randomBytes );
0 commit comments