add unit tests for new metric labels

Author: swibi-ttd

src/test/java/com/uid2/operator/UIDOperatorVerticleTest.java

@@ -32,6 +32,7 @@
 import com.uid2.shared.store.*;
 import com.uid2.shared.store.reader.RotatingKeysetProvider;
 import com.uid2.shared.store.salt.ISaltProvider;
+import io.micrometer.core.instrument.DistributionSummary;
 import io.micrometer.core.instrument.Metrics;
 import io.micrometer.core.instrument.simple.SimpleMeterRegistry;
 import io.vertx.core.AsyncResult;
@@ -2904,6 +2905,81 @@ void identityMapBatchRequestTooLargeForPhone(Vertx vertx, VertxTestContext testC
         send(vertx, "v2/identity/map", req, 413, json -> testContext.completeNow());
     }
 
+    @Test
+    void identityMapV2InputMetricTaggedWithEmailDiiType(Vertx vertx, VertxTestContext testContext) {
+        fakeAuth(201, Role.MAPPER);
+        setupKeys();
+        setupSalts();
+
+        JsonObject req = new JsonObject()
+                .put("email", new JsonArray().add("test1@uid2.com").add("test2@uid2.com"));
+
+        send(vertx, "v2/identity/map", req, 200, json -> {
+            DistributionSummary summary = Metrics.globalRegistry
+                    .get("uid2_operator_identity_map_inputs")
+                    .tag("path", "/v2/identity/map")
+                    .tag("dii_type", "email")
+                    .summary();
+            assertNotNull(summary);
+            assertEquals(2.0, summary.totalAmount());
+            testContext.completeNow();
+        });
+    }
+
+    @Test
+    void identityMapV2InputMetricTaggedWithEmailHashDiiType(Vertx vertx, VertxTestContext testContext) {
+        fakeAuth(201, Role.MAPPER);
+        setupKeys();
+        setupSalts();
+
+        JsonObject req = new JsonObject()
+                .put("email_hash", new JsonArray()
+                        .add(TokenUtils.getIdentityHashString("test1@uid2.com")));
+
+        send(vertx, "v2/identity/map", req, 200, json -> {
+            DistributionSummary summary = Metrics.globalRegistry
+                    .get("uid2_operator_identity_map_inputs")
+                    .tag("path", "/v2/identity/map")
+                    .tag("dii_type", "email_hash")
+                    .summary();
+            assertNotNull(summary);
+            assertEquals(1.0, summary.totalAmount());
+            testContext.completeNow();
+        });
+    }
+
+    @Test
+    void identityMapV3InputMetricTaggedWithPathAndDiiType(Vertx vertx, VertxTestContext testContext) {
+        fakeAuth(201, Role.MAPPER);
+        SaltEntry salt = setupSalts();
+        when(saltProviderSnapshot.getRotatingSalt(any())).thenReturn(salt);
+
+        JsonObject req = new JsonObject()
+                .put("email", new JsonArray().add("test1@uid2.com").add("test2@uid2.com"))
+                .put("phone_hash", new JsonArray()
+                        .add(TokenUtils.getIdentityHashString("+15555555555")));
+
+        send(vertx, "v3/identity/map", req, 200, json -> {
+            DistributionSummary emailSummary = Metrics.globalRegistry
+                    .get("uid2_operator_identity_map_inputs")
+                    .tag("path", "/v3/identity/map")
+                    .tag("dii_type", "email")
+                    .summary();
+            assertNotNull(emailSummary);
+            assertEquals(2.0, emailSummary.totalAmount());
+
+            DistributionSummary phoneHashSummary = Metrics.globalRegistry
+                    .get("uid2_operator_identity_map_inputs")
+                    .tag("path", "/v3/identity/map")
+                    .tag("dii_type", "phone_hash")
+                    .summary();
+            assertNotNull(phoneHashSummary);
+            assertEquals(1.0, phoneHashSummary.totalAmount());
+
+            testContext.completeNow();
+        });
+    }
+
     @ParameterizedTest
     @ValueSource(booleans = {true, false})
     void identityMapOptoutDefaultOption(boolean useV4Uid, Vertx vertx, VertxTestContext testContext) {