add PR IQSS#11358

Author: Benedikt Kruse

src/main/java/edu/harvard/iq/dataverse/api/Metrics.java

@@ -661,6 +661,13 @@ public Response getMakeDataCountMetricTimeSeries(@Context Request req, @Context
         } catch (IllegalArgumentException ex) {
             return error(Response.Status.BAD_REQUEST, ex.getMessage());
         }
+        if (country != null) {
+            country = country.toLowerCase();
+
+            if (!MakeDataCountUtil.isValidCountryCode(country)) {
+                return error(Response.Status.BAD_REQUEST, "Country must be one of the ISO 1366 Country Codes");
+            }
+        }
         String metricName = "MDC-" + metricType.toString() + ((country == null) ? "" : "-" + country);
 
         JsonArray jsonArray = MetricsUtil.stringToJsonArray(metricsSvc.returnUnexpiredCacheAllTime(metricName, null, d));

src/test/java/edu/harvard/iq/dataverse/api/MetricsIT.java

@@ -4,13 +4,15 @@
 import static jakarta.ws.rs.core.Response.Status.OK;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 
+import org.hamcrest.Matchers;
 import org.junit.jupiter.api.AfterAll;
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
 
 import edu.harvard.iq.dataverse.metrics.MetricsUtil;
 import edu.harvard.iq.dataverse.util.FileUtil;
 import io.restassured.RestAssured;
+import io.restassured.http.ContentType;
 import io.restassured.response.Response;
 import jakarta.ws.rs.core.MediaType;
 
@@ -381,4 +383,41 @@ public void testGetDatasetsBySubjectToMonth() {
         response.then().assertThat()
                 .statusCode(OK.getStatusCode());
     }
+
+    @Test
+    public void testUnsupportedQueryParam() {
+        Response response = UtilIT.makeDataCountMetricTimeSeries("viewCount", "unsupportedParam=value");
+
+        response.then().assertThat()
+                .statusCode(BAD_REQUEST.getStatusCode())
+                .body("message", Matchers.containsString("queryParameter unsupportedParam not supported for this endpoint"));
+    }
+
+    @Test
+    public void testInvalidMetric() {
+        Response response = UtilIT.makeDataCountMetricTimeSeries("invalidMetric", null);
+
+        response.then().assertThat()
+                .statusCode(BAD_REQUEST.getStatusCode())
+                .body("message", Matchers.containsString("MetricType must be one of these values"));
+    }
+
+    @Test
+    public void testInvalidCountryCode() {
+        Response response = UtilIT.makeDataCountMetricTimeSeries("viewCount", "country=INVALID");
+
+        response.then().assertThat()
+                .statusCode(BAD_REQUEST.getStatusCode())
+                .body("message", Matchers.containsString("Country must be one of the ISO 1366 Country Codes"));
+    }
+
+    @Test
+    public void testValidRequest() {
+        Response response = UtilIT.makeDataCountMetricTimeSeries("viewCount", "country=us");
+
+        response.then().assertThat()
+                .statusCode(OK.getStatusCode())
+                .contentType(ContentType.JSON);
+    }
+
 }

src/test/java/edu/harvard/iq/dataverse/api/UtilIT.java

@@ -2702,6 +2702,15 @@ static Response metricsDatasetsBySubjectToMonth(String month, String queryParams
         RequestSpecification requestSpecification = given();
         return requestSpecification.get("/api/info/metrics/datasets/bySubject/toMonth/" + month + optionalQueryParams);
     }
+
+    public static Response makeDataCountMetricTimeSeries(String metricType, String queryParams) {
+        String apiPath = "/api/v1/metrics/makeDataCount/" + metricType + "/monthly";
+
+        Response response = given()
+                .get(apiPath + (queryParams != null && !queryParams.isEmpty() ? "?" + queryParams : ""));
+
+        return response;
+    }
 
     static Response clearMetricCache() {
         RequestSpecification requestSpecification = given();
@@ -3986,4 +3995,5 @@ public static Response getOpenAPI(String accept, String format) {
                 .get("/openapi");
         return response;
     }
+    
 }