From f41a5c8985f73e43c04aff65659740e02c90054c Mon Sep 17 00:00:00 2001 From: Konstantinos Spyrou Date: Wed, 16 Nov 2022 16:33:42 +0200 Subject: [PATCH] fixed prometheus test --- src/test/java/metrics/PrometheusTest.java | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/src/test/java/metrics/PrometheusTest.java b/src/test/java/metrics/PrometheusTest.java index 7552e00..8c34c6c 100644 --- a/src/test/java/metrics/PrometheusTest.java +++ b/src/test/java/metrics/PrometheusTest.java @@ -2,6 +2,7 @@ package metrics; import eu.dnetlib.repo.manager.controllers.PrometheusController; import eu.dnetlib.repo.manager.service.PiWikService; +import eu.dnetlib.repo.manager.service.RepositoryService; import org.junit.Before; import org.junit.Test; import org.mockito.InjectMocks; @@ -23,6 +24,10 @@ public class PrometheusTest { @Mock PiWikService piWikService; + @Autowired + @Mock + RepositoryService repositoryService; + @Autowired @InjectMocks private PrometheusController prometheusController; @@ -34,6 +39,7 @@ public class PrometheusTest { when(piWikService.getTotal()).thenReturn(TOTAL); when(piWikService.getValidated(true)).thenReturn(VALIDATED); when(piWikService.getValidated(false)).thenReturn(TOTAL - VALIDATED); + when(repositoryService.getTotalRegisteredRepositories()).thenReturn(TOTAL); } @Test @@ -47,8 +53,9 @@ public class PrometheusTest { public void testPiwikMetrics() { assertTrue(piWikService.getValidated(false).equals(TOTAL - VALIDATED)); String report = prometheusController.getPiwikMetrics(); - assertTrue(report.contains("total_total " + TOTAL)); - assertTrue(report.contains("validated_total " + VALIDATED)); + assertTrue(report.contains("provide_repositories_registered_total " + TOTAL)); + assertTrue(report.contains("provide_usagecounts_repositories_registered_total " + TOTAL)); + assertTrue(report.contains("provide_usagecounts_repositories_validated_total " + VALIDATED)); System.out.println(report); } -} \ No newline at end of file +}