Batch email test for identity map

Author: aulme

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

@@ -3,14 +3,15 @@
 import app.AppsMap;
 import app.component.App;
 import app.component.Operator;
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
 import com.uid2.client.DecryptionStatus;
 import org.junit.jupiter.params.provider.Arguments;
 
-import java.util.HashSet;
-import java.util.List;
-import java.util.Optional;
-import java.util.Set;
+import java.util.*;
 import java.util.stream.Collectors;
+import java.util.stream.IntStream;
 import java.util.stream.Stream;
 
 import static org.junit.jupiter.api.Named.named;
@@ -326,6 +327,27 @@ public static Set<Arguments> identityMapBatchEmailArgs() {
         return args;
     }
 
+    public static Set<Arguments> identityMapBigBatchArgs() throws JsonProcessingException {
+        Set<Operator> operators = AppsMap.getApps(Operator.class);
+
+        List<String> emails = IntStream.range(0, 10_000)
+                .mapToObj(i -> "email_" + i + "_" + UUID.randomUUID() + "@example.com")
+                .collect(Collectors.toList());
+
+        ObjectMapper mapper = new ObjectMapper();
+        ObjectNode root = mapper.createObjectNode();
+        root.putPOJO("email", emails);
+        root.put("policy", 1);
+
+        String emailsPayload = mapper.writeValueAsString(root);
+
+        Set<Arguments> args = new HashSet<>();
+        for (Operator operator : operators) {
+            args.add(Arguments.of("10k emails", operator, operator.getName(), emailsPayload, emails));
+        }
+        return args;
+    }
+
     public static Set<Arguments> identityMapBatchEmailArgsBadPolicy() {
         Set<Operator> operators = getPublicOperators();
         Set<List<String>> inputs = Set.of(

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

@@ -7,9 +7,13 @@
 import app.component.Operator;
 import com.fasterxml.jackson.databind.JsonNode;
 import com.uid2.client.*;
+import org.junit.jupiter.api.Test;
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.MethodSource;
 
+import java.time.Duration;
+import java.util.List;
+
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.junit.jupiter.api.Assertions.*;
 
@@ -105,6 +109,28 @@ public void testV2IdentityMap(String label, Operator operator, String operatorNa
         assertThat(response.at("/status").asText()).isEqualTo("success");
     }
 
+    @ParameterizedTest(name = "/v2/identity/map - {0} - {2}")
+    @MethodSource({
+            "suite.operator.TestData#identityMapBigBatchArgs"
+    })
+    public void testV2IdentityMapLargeBatch(String label, Operator operator, String operatorName, String payload, List<String> diis) {
+        assertTimeoutPreemptively(Duration.ofSeconds(5), () ->  { // Validate we didn't make mapping too slow.
+            JsonNode response = operator.v2IdentityMap(payload);
+
+            assertThat(response.at("/status").asText()).isEqualTo("success");
+
+            var mapped = response.at("/body/mapped");
+            assertThat(mapped.size()).isEqualTo(10_000);
+
+            for (int i = 0; i < 10_000; i++) {
+                assertThat(mapped.get(i).get("identifier").asText()).isEqualTo(diis.get(i));
+                assertThat(mapped.get(i).get("advertising_id").asText()).isNotNull().isNotEmpty();
+                assertThat(mapped.get(i).get("bucket_id").asText()).isNotNull().isNotEmpty();
+            }
+        });
+    }
+
+
     @ParameterizedTest(name = "/v2/identity/map - VALIDATE EMAIL - {0} - {2}")
     @MethodSource({
             "suite.operator.TestData#tokenValidateEmailArgs",