update test case

Author: clarkxuyang

pom.xml

@@ -5,7 +5,7 @@
 
     <groupId>com.uid2</groupId>
     <artifactId>uid2-shared</artifactId>
-    <version>8.0.6</version>
+    <version>8.1.0</version>
     <name>${project.groupId}:${project.artifactId}</name>
     <description>Library for all the shared uid2 operations</description>
     <url>https://github.com/IABTechLab/uid2docs</url>

src/test/java/com/uid2/shared/vertx/RequestCapturingHandlerTest.java

@@ -4,6 +4,8 @@
 import com.uid2.shared.auth.ClientKey;
 import com.uid2.shared.auth.OperatorKey;
 import com.uid2.shared.middleware.AuthMiddleware;
+import com.uid2.shared.model.Site;
+import com.uid2.shared.store.ISiteStore;
 import io.micrometer.core.instrument.Metrics;
 import io.micrometer.core.instrument.simple.SimpleMeterRegistry;
 import io.vertx.core.Handler;
@@ -14,35 +16,50 @@
 import io.vertx.ext.web.client.WebClient;
 import io.vertx.junit5.VertxExtension;
 import io.vertx.junit5.VertxTestContext;
+import org.assertj.core.condition.AnyOf;
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.ExtendWith;
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.Arguments;
 import org.junit.jupiter.params.provider.MethodSource;
+import org.mockito.Mock;
+import org.mockito.Mockito;
 
 import java.time.Instant;
+import java.util.HashSet;
 import java.util.Set;
 import java.util.stream.Stream;
 
+import static org.mockito.ArgumentMatchers.anyInt;
+import static org.mockito.Mockito.when;
+
 @ExtendWith(VertxExtension.class)
 public class RequestCapturingHandlerTest {
     private static final int Port = 8080;
     private static final Handler<RoutingContext> dummyResponseHandler = routingContext -> {
         routingContext.response().setStatusCode(200).end();
     };
 
+    private ISiteStore siteStore;
+
     @BeforeEach
     public void before() {
         Metrics.globalRegistry.forEachMeter(Metrics.globalRegistry::remove);
         Metrics.globalRegistry.add(new SimpleMeterRegistry());
+
+        Site site100 = new Site(100, "test-100", true);
+        Site site200 = new Site(200, "test-200", true);
+        siteStore = Mockito.mock(ISiteStore.class);
+        when(siteStore.getSite(100)).thenReturn(site100);
+        when(siteStore.getSite(200)).thenReturn(site200);
     }
 
     @Test
     public void captureSimplePath(Vertx vertx, VertxTestContext testContext) {
         Router router = Router.router(vertx);
-        router.route().handler(new RequestCapturingHandler());
+        router.route().handler(new RequestCapturingHandler(siteStore));
         router.get("/v1/token/generate").handler(dummyResponseHandler);
 
         vertx.createHttpServer().requestHandler(router).listen(Port, testContext.succeeding(id -> {
@@ -64,7 +81,7 @@ public void captureSimplePath(Vertx vertx, VertxTestContext testContext) {
     @Test
     public void captureSubRouterPath(Vertx vertx, VertxTestContext testContext) {
         Router router = Router.router(vertx);
-        router.route().handler(new RequestCapturingHandler());
+        router.route().handler(new RequestCapturingHandler(siteStore));
 
         Router v2Router = Router.router(vertx);
         v2Router.post("/token/generate").handler(dummyResponseHandler);
@@ -90,7 +107,7 @@ public void captureSubRouterPath(Vertx vertx, VertxTestContext testContext) {
     @Test
     public void captureStaticPath(Vertx vertx, VertxTestContext testContext) {
         Router router = Router.router(vertx);
-        router.route().handler(new RequestCapturingHandler());
+        router.route().handler(new RequestCapturingHandler(siteStore));
         router.get("/static/*").handler(dummyResponseHandler);
 
         vertx.createHttpServer().requestHandler(router).listen(Port, testContext.succeeding(id -> {
@@ -112,7 +129,7 @@ public void captureStaticPath(Vertx vertx, VertxTestContext testContext) {
     @Test
     public void captureUnknownPath(Vertx vertx, VertxTestContext testContext) {
         Router router = Router.router(vertx);
-        router.route().handler(new RequestCapturingHandler());
+        router.route().handler(new RequestCapturingHandler(siteStore));
 
         vertx.createHttpServer().requestHandler(router).listen(Port, testContext.succeeding(id -> {
             WebClient client = WebClient.create(vertx);
@@ -132,9 +149,10 @@ public void captureUnknownPath(Vertx vertx, VertxTestContext testContext) {
 
     @ParameterizedTest
     @MethodSource("siteIdRoutingContextData")
-    public void getSiteIdFromRoutingContextData(String key, Object value, String siteId, Vertx vertx, VertxTestContext testContext) {
+    public void getSiteIdFromRoutingContextData(String key, Object value, String siteId, String siteName,
+                                                Vertx vertx, VertxTestContext testContext) {
         Router router = Router.router(vertx);
-        router.route().handler(new RequestCapturingHandler());
+        router.route().handler(new RequestCapturingHandler(siteStore));
         router.get("/test").handler(ctx -> {
             if (key != null) {
                 ctx.put(key, value);
@@ -149,6 +167,7 @@ public void getSiteIdFromRoutingContextData(String key, Object value, String sit
                         double actual = Metrics.globalRegistry
                                 .get("uid2.http_requests")
                                 .tag("site_id", siteId)
+                                .tag("site_name", siteName)
                                 .counter()
                                 .count();
                         Assertions.assertEquals(1, actual);
@@ -160,11 +179,14 @@ public void getSiteIdFromRoutingContextData(String key, Object value, String sit
     private static Stream<Arguments> siteIdRoutingContextData() {
         // Arguments are: routing context data key, routing context data value, site ID tag.
         return Stream.of(
-                Arguments.of(Const.RoutingContextData.SiteId, 100, "100"),
-                Arguments.of(AuthMiddleware.API_CLIENT_PROP, new ClientKey("keyHash", "keySalt", "secret", "", Instant.MIN, Set.of(), 200, "test-key-id-1"), "200"),
-                Arguments.of(AuthMiddleware.API_CLIENT_PROP, new OperatorKey("test-keyHash", "test-keySalt", "name", "contact", "protocol", 0, false, "test-key-id-2"), "null"),
-                Arguments.of(AuthMiddleware.API_CLIENT_PROP, new OperatorKey("test-keyHash", "test-keySalt", "name", "contact", "protocol", 0, false, 300, "test-key-id-3"), "300"),
-                Arguments.of(null, null, "null")
+                Arguments.of(Const.RoutingContextData.SiteId, 100, "100", "test-100"),
+                Arguments.of(AuthMiddleware.API_CLIENT_PROP, new ClientKey("keyHash", "keySalt",
+                        "secret", "", Instant.MIN, Set.of(), 200, "test-key-id-1"), "200", "test-200"),
+                Arguments.of(AuthMiddleware.API_CLIENT_PROP, new OperatorKey("test-keyHash", "test-keySalt",
+                        "name", "contact", "protocol", 0, false, "test-key-id-2"), "null", "unknown"),
+                Arguments.of(AuthMiddleware.API_CLIENT_PROP, new OperatorKey("test-keyHash", "test-keySalt",
+                        "name", "contact", "protocol", 0, false, 300, "test-key-id-3"), "300", "unknown"),
+                Arguments.of(null, null, "null", "unknown")
         );
     }
 }