aboutsummaryrefslogtreecommitdiffstats
path: root/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/container/metrics/MetricsTest.java
blob: 8e23c7e54b6d649129f1f05da65fcfc5dc82de2b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.node.admin.container.metrics;

import org.junit.jupiter.api.Test;

import java.util.Map;
import java.util.stream.Collectors;

import static com.yahoo.vespa.hosted.node.admin.container.metrics.Metrics.APPLICATION_HOST;
import static com.yahoo.vespa.hosted.node.admin.container.metrics.Metrics.DimensionType.DEFAULT;
import static org.junit.jupiter.api.Assertions.assertEquals;

/**
 * @author freva
 */
public class MetricsTest {
    private static final Dimensions hostDimension = new Dimensions.Builder().add("host", "abc.yahoo.com").build();
    private final Metrics metrics = new Metrics();

    @Test
    void testDefaultValue() {
        metrics.declareCounter("some.name", hostDimension);

        assertEquals(getMetricsForDimension(hostDimension).get("some.name"), 0L);
    }

    @Test
    void testSimpleIncrementMetric() {
        Counter counter = metrics.declareCounter("a_counter.value", hostDimension);

        counter.add(5);
        counter.add(8);

        Map<String, Number> latestMetrics = getMetricsForDimension(hostDimension);
        assertEquals(1, latestMetrics.size(), "Expected only 1 metric value to be set");
        assertEquals(latestMetrics.get("a_counter.value"), 13L); // 5 + 8
    }

    @Test
    void testSimpleGauge() {
        Gauge gauge = metrics.declareGauge("test.gauge", hostDimension);

        gauge.sample(42);
        gauge.sample(-342.23);

        Map<String, Number> latestMetrics = getMetricsForDimension(hostDimension);
        assertEquals(1, latestMetrics.size(), "Expected only 1 metric value to be set");
        assertEquals(latestMetrics.get("test.gauge"), -342.23);
    }

    @Test
    void testRedeclaringSameGauge() {
        Gauge gauge = metrics.declareGauge("test.gauge", hostDimension);
        gauge.sample(42);

        // Same as hostDimension, but new instance.
        Dimensions newDimension = new Dimensions.Builder().add("host", "abc.yahoo.com").build();
        Gauge newGauge = metrics.declareGauge("test.gauge", newDimension);
        newGauge.sample(56);

        assertEquals(getMetricsForDimension(hostDimension).get("test.gauge"), 56.);
    }

    @Test
    void testSameMetricNameButDifferentDimensions() {
        Gauge gauge = metrics.declareGauge("test.gauge", hostDimension);
        gauge.sample(42);

        // Not the same as hostDimension.
        Dimensions newDimension = new Dimensions.Builder().add("host", "abcd.yahoo.com").build();
        Gauge newGauge = metrics.declareGauge("test.gauge", newDimension);
        newGauge.sample(56);

        assertEquals(getMetricsForDimension(hostDimension).get("test.gauge"), 42.);
        assertEquals(getMetricsForDimension(newDimension).get("test.gauge"), 56.);
    }

    @Test
    void testDeletingMetric() {
        metrics.declareGauge("test.gauge", hostDimension);

        Dimensions differentDimension = new Dimensions.Builder().add("host", "abcd.yahoo.com").build();
        metrics.declareGauge("test.gauge", differentDimension);

        assertEquals(2, metrics.getMetricsByType(DEFAULT).size());
        metrics.deleteMetricByDimension(APPLICATION_HOST, differentDimension, DEFAULT);
        assertEquals(1, metrics.getMetricsByType(DEFAULT).size());
        assertEquals(getMetricsForDimension(hostDimension).size(), 1);
        assertEquals(getMetricsForDimension(differentDimension).size(), 0);
    }

    private Map<String, Number> getMetricsForDimension(Dimensions dimensions) {
        return metrics.getOrCreateApplicationMetrics(APPLICATION_HOST, DEFAULT)
                .getOrDefault(dimensions, Map.of())
                .entrySet()
                .stream()
                .collect(Collectors.toMap(Map.Entry::getKey, entry -> entry.getValue().getValue()));
    }
}