Add phone tests

IABTechLab · Jun 3, 2024 · 1b209bc · 1b209bc
1 parent 7989701
commit 1b209bc
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 10 deletions.
diff --git a/src/test/java/suite/optout/OptoutTest.java b/src/test/java/suite/optout/OptoutTest.java
@@ -123,7 +123,8 @@ public void testV2LogoutWithV2TokenGenerateOldParticipant(String label, Operator
 
     @ParameterizedTest(name = "/v2/token/logout with /v2/identity/map - {0} - {2}")
     @MethodSource({
-            "suite.optout.TestData#identityMapEmailArgs"
+            "suite.optout.TestData#identityMapEmailArgs",
+            "suite.optout.TestData#identityMapPhoneArgs"
     })
     @Order(5)
     public void testV2LogoutWithV2IdentityMap(String label, Operator operator, String operatorName, String type, String emailOrPhone, boolean toOptOut) throws Exception {

diff --git a/src/test/java/suite/optout/TestData.java b/src/test/java/suite/optout/TestData.java
@@ -67,15 +67,7 @@ public static Set<Arguments> tokenPhoneArgs() {
 
     public static Set<Arguments> optoutTokenPhoneArgs() {
         Set<Operator> operators = getPublicOperators();
-        Random random = new Random();
-        StringBuilder phone = new StringBuilder(String.valueOf(random.nextLong((10000000000L - 1000000000L) +1) + 1000000000L));
-        while (phone.length() < 10) {
-            phone.insert(0, "0");
-        }
-        phone.insert(0, "+0");
-        Set<List<String>> inputs = Set.of(
-                List.of("good phone", "phone", phone.toString())
-        );
+        Set<List<String>> inputs = generatePhoneSet(1);
 
         Set<Arguments> args = new HashSet<>();
         if (PHONE_SUPPORT) {
@@ -101,6 +93,19 @@ public static Set<Arguments> identityMapEmailArgs() {
         return args;
     }
 
+    public static Set<Arguments> identityMapPhoneArgs() {
+        Set<Operator> operators = getPublicOperators();
+        Set<List<String>> inputs = generatePhoneSet(4);
+
+        Set<Arguments> args = new HashSet<>();
+        for (Operator operator : operators) {
+            for (List<String> input : inputs) {
+                args.add(Arguments.of(input.get(0), operator, operator.getName(), input.get(1), input.get(2), Boolean.parseBoolean(input.get(3))));
+            }
+        }
+        return args;
+    }
+
     private static Set<List<String>> generateEmailSet(int count) {
         Random random = new Random();
         Set<List<String>> inputs = new HashSet<>(count);
@@ -113,6 +118,21 @@ private static Set<List<String>> generateEmailSet(int count) {
         return inputs;
     }
 
+    private static Set<List<String>> generatePhoneSet(int count) {
+        Random random = new Random();
+        Set<List<String>> inputs = new HashSet<>();
+        final long bound = (10000000000L - 1000000000L) + 1;
+        for (int i = 0; i < count; ++i) {
+            StringBuilder phone = new StringBuilder(String.valueOf(random.nextLong(bound) + 1000000000L));
+            while (phone.length() < 10) {
+                phone.insert(0, "0");
+            }
+            phone.insert(0, "+0");
+            inputs.add(List.of("good phone " + (i + 1), "phone", phone.toString(),  i % 2 == 0 ? "true" : "false"));
+        }
+        return inputs;
+    }
+
     private static String getRandomString(int minLength, int maxLength) {
         String alphabet = "abcdefghijklmnopqrstuvwxyz0123456789";
         StringBuilder s = new StringBuilder();