Skip to content

Commit 7fff38b

Browse files
committed
refactoring
1 parent bcba9a9 commit 7fff38b

File tree

2 files changed

+24
-20
lines changed

2 files changed

+24
-20
lines changed

src/test/java/suite/operator/TestData.java

Lines changed: 24 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -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);

src/test/java/suite/operator/V2ApiOperatorTest.java

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@
77
import app.component.Operator;
88
import com.fasterxml.jackson.databind.JsonNode;
99
import com.uid2.client.*;
10-
import org.junit.jupiter.api.Test;
1110
import org.junit.jupiter.params.ParameterizedTest;
1211
import org.junit.jupiter.params.provider.MethodSource;
1312

0 commit comments

Comments
 (0)