Update unit tests

IABTechLab · Oct 8, 2024 · 9078e80 · 9078e80
1 parent 3b2f090
commit 9078e80
Show file tree

Hide file tree

Showing 2 changed files with 97 additions and 0 deletions.
diff --git a/src/test/java/com/uid2/client/BidstreamClientTests.java b/src/test/java/com/uid2/client/BidstreamClientTests.java
@@ -206,6 +206,21 @@ private static Stream<Arguments> data_IdentityScopeAndType_TestCases() {
         );
     }
 
+    @ParameterizedTest
+    @ValueSource(strings = {"V2", "V3", "V4"})
+    public void userOptedOutTest(TokenVersionForTesting tokenVersion) throws Exception {
+        refresh(keyBidstreamResponse(IdentityScope.UID2, MASTER_KEY, SITE_KEY));
+        int privacyBits = PrivacyBitsBuilder.Builder().WithOptedOut(true).Build();
+
+        String advertisingToken = AdvertisingTokenBuilder.builder().withVersion(tokenVersion).withPrivacyBits(privacyBits).build();
+
+        validateAdvertisingToken(advertisingToken, IdentityScope.UID2, IdentityType.Email, tokenVersion);
+        DecryptionResponse res = bidstreamClient.decryptTokenIntoRawUid(advertisingToken, null);
+        assertFalse(res.isSuccess());
+        assertEquals(DecryptionStatus.SUCCESS, res.getStatus());
+        assertNull(res.getUid());
+    }
+
     // These are the domain or app names associated with site SITE_ID, as defined by keyBidstreamResponse()
     @ParameterizedTest
     @CsvSource({

diff --git a/src/test/java/com/uid2/client/KeyParserTests.java b/src/test/java/com/uid2/client/KeyParserTests.java
@@ -96,6 +96,88 @@ public void parseErrorKeyList() {
         assertThrows(Exception.class, () -> parse("{\"body\": [{\"id\": 5}]}"));
     }
 
+    @Test
+    public void parseMissingSiteData() {
+        String json = "{ \"body\": {\n" +
+                "  \"keys\": [\n" +
+                "       {\n" +
+                "       \"id\": 3,\n" +
+                "       \"keyset_id\": 99999,\n" +
+                "       \"created\": 1609459200,\n" +
+                "       \"activates\": 1609459210,\n" +
+                "       \"expires\": 1893456000,\n" +
+                "       \"secret\": \"o8HsvkwJ5Ulnrd0uui3GpukpwDapj+JLqb7qfN/GJKo=\"\n" +
+                "          }\n" +
+                "      ]\n" +
+                "  }\n" +
+                "}";
+        KeyContainer keyContainer = parse(json);
+        boolean isAllowed = keyContainer.isDomainOrAppNameAllowedForSite(1, "example.com");
+        assertFalse(isAllowed);
+        assertNotNull(keyContainer.getKey(3));
+    }
+
+    @Test
+    public void parseEmptySiteData() {
+        String json = "{ \"body\": {\n" +
+                 "  \"keys\": [\n" +
+                 "       {\n" +
+                 "       \"id\": 3,\n" +
+                 "       \"keyset_id\": 99999,\n" +
+                 "       \"created\": 1609459200,\n" +
+                 "       \"activates\": 1609459210,\n" +
+                 "       \"expires\": 1893456000,\n" +
+                 "       \"secret\": \"o8HsvkwJ5Ulnrd0uui3GpukpwDapj+JLqb7qfN/GJKo=\"\n" +
+                 "          }\n" +
+                 "      ],\n" +
+                 "      \"site_data\": []\n" +
+                 "     }\n" +
+                 "}";
+        KeyContainer keyContainer = parse(json);
+        boolean isAllowed = keyContainer.isDomainOrAppNameAllowedForSite(1, "example.com");
+        assertFalse(isAllowed);
+        assertFalse(keyContainer.isDomainOrAppNameAllowedForSite(1,  null));
+        assertNotNull(keyContainer.getKey(3));
+    }
+
+    @Test
+    public void parseSiteDataSharingEndpoint() {
+        String json = "{\n" +
+                "        \"body\": {\n" +
+                "            \"keys\": [\n" +
+                "            {\n" +
+                "                \"id\": 3,\n" +
+                "                \"keyset_id\": 99999,\n" +
+                "                \"created\": 1609459200,\n" +
+                "                \"activates\": 1609459210,\n" +
+                "                \"expires\": 1893456000,\n" +
+                "                \"secret\": \"o8HsvkwJ5Ulnrd0uui3GpukpwDapj+JLqb7qfN/GJKo=\"\n" +
+                "            }\n" +
+                "        ],\n" +
+                "            \"site_data\": [\n" +
+                "            {\n" +
+                "                \"id\": 9,\n" +
+                "                \"domain_names\": [\"example.com\"]\n" +
+                "            },\n" +
+                "            {\n" +
+                "                \"id\": 100,\n" +
+                "                \"domain_names\": [\"example.org\", \"example.net\"]\n" +
+                "            }\n" +
+                "        ]\n" +
+                "    }\n" +
+                "}";
+        KeyContainer keyContainer = parse(json);
+        assertTrue(keyContainer.isDomainOrAppNameAllowedForSite(9, "example.com"));
+        assertFalse(keyContainer.isDomainOrAppNameAllowedForSite(9, "example.org"));
+        assertFalse(keyContainer.isDomainOrAppNameAllowedForSite(9, "example.net"));
+
+        assertFalse(keyContainer.isDomainOrAppNameAllowedForSite(100, "example.com"));
+        assertTrue(keyContainer.isDomainOrAppNameAllowedForSite(100, "example.org"));
+        assertTrue(keyContainer.isDomainOrAppNameAllowedForSite(100, "example.net"));
+
+        assertNotNull(keyContainer.getKey(3));
+    }
+
     @Test
     public void parseWithNullTokenExpirySecondField() {
         String s = "{ \"body\": { " +