From 002ba6cf2e144e0d86e867011418ba536b55b573 Mon Sep 17 00:00:00 2001 From: mabofu Date: Thu, 27 Mar 2025 11:06:57 +0800 Subject: [PATCH] Fixed code coverage: fixed RangeQueryBuilder code coverage --- .../builder/RangeQueryBuilderTest.java | 33 +++++++++++++++++++ 1 file changed, 33 insertions(+) create mode 100644 dsms-engine-application/src/test/java/com/dsms/common/prometheus/builder/RangeQueryBuilderTest.java diff --git a/dsms-engine-application/src/test/java/com/dsms/common/prometheus/builder/RangeQueryBuilderTest.java b/dsms-engine-application/src/test/java/com/dsms/common/prometheus/builder/RangeQueryBuilderTest.java new file mode 100644 index 00000000..6b470919 --- /dev/null +++ b/dsms-engine-application/src/test/java/com/dsms/common/prometheus/builder/RangeQueryBuilderTest.java @@ -0,0 +1,33 @@ +package com.dsms.common.prometheus.builder; + +import static com.dsms.common.constant.PromQL.NODE_MEMORY_TOTAL; + +import java.net.URI; +import org.assertj.core.api.Assertions; +import org.junit.jupiter.api.Test; + +class RangeQueryBuilderTest { + + @Test + void build() { + String prometheusUrl = "http://prometheus:9095"; + RangeQueryBuilder rangeQueryBuilder = new RangeQueryBuilder(prometheusUrl); + URI build1 = rangeQueryBuilder.withQuery(NODE_MEMORY_TOTAL).build(); + Assertions.assertThat(build1) + .hasHost("prometheus") + .hasPort(9095) + .hasParameter("query", NODE_MEMORY_TOTAL); + org.junit.jupiter.api.Assertions.assertTrue(rangeQueryBuilder.getTargetUriPattern().contains(prometheusUrl)); + long currentTimeMillis = System.currentTimeMillis(); + URI noStep = rangeQueryBuilder.build(prometheusUrl, currentTimeMillis, currentTimeMillis + 1000); + Assertions.assertThat(noStep) + .hasParameter("start", String.valueOf(currentTimeMillis)) + .hasParameter("end", String.valueOf(currentTimeMillis + 1000)) + .hasParameter("step", String.valueOf(Math.max(1000 / 250, 1))); + URI epochTime = rangeQueryBuilder.build(prometheusUrl, currentTimeMillis, currentTimeMillis + 1000, String.valueOf(7)); + Assertions.assertThat(epochTime) + .hasParameter("start", String.valueOf(currentTimeMillis)) + .hasParameter("end", String.valueOf(currentTimeMillis + 1000)) + .hasParameter("step", "7"); + } +} \ No newline at end of file -- Gitee