library/cpp/monlib/metrics/metric_sub_registry_ut.cpp


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

#include "metric_sub_registry.h"

#include <library/cpp/testing/unittest/registar.h>

using namespace NMonitoring;

Y_UNIT_TEST_SUITE(TMetricSubRegistryTest) {
    Y_UNIT_TEST(WrapRegistry) {
        TMetricRegistry registry;

        {
            TMetricSubRegistry subRegistry{{{"common", "label"}}, &registry};
            IIntGauge* g = subRegistry.IntGauge(MakeLabels({{"my", "gauge"}}));
            UNIT_ASSERT(g);
            g->Set(42);
        }

        TIntGauge* g = registry.IntGauge({{"my", "gauge"}, {"common", "label"}});
        UNIT_ASSERT(g);
        UNIT_ASSERT_VALUES_EQUAL(g->Get(), 42);
    }

    Y_UNIT_TEST(CommonLabelsDoNotOverrideGeneralLabel) {
        TMetricRegistry registry;

        {
            TMetricSubRegistry subRegistry{{{"common", "label"}, {"my", "notOverride"}}, &registry};
            IIntGauge* g = subRegistry.IntGauge(MakeLabels({{"my", "gauge"}}));
            UNIT_ASSERT(g);
            g->Set(1234);
        }

        TIntGauge* knownGauge = registry.IntGauge({{"my", "gauge"}, {"common", "label"}});
        UNIT_ASSERT(knownGauge);
        UNIT_ASSERT_VALUES_EQUAL(knownGauge->Get(), 1234);

        TIntGauge* newGauge = registry.IntGauge({{"common", "label"}, {"my", "notOverride"}});
        UNIT_ASSERT(newGauge);
        UNIT_ASSERT_VALUES_EQUAL(newGauge->Get(), 0);
    }

    Y_UNIT_TEST(RemoveMetric) {
        TMetricRegistry registry;

        {
            TMetricSubRegistry subRegistry{{{"common", "label"}}, &registry};
            IIntGauge* g = subRegistry.IntGauge(MakeLabels({{"my", "gauge"}}));
            UNIT_ASSERT(g);
            g->Set(1234);
        }

        IIntGauge* g1 = registry.IntGauge({{"my", "gauge"}, {"common", "label"}});
        UNIT_ASSERT(g1);
        UNIT_ASSERT_VALUES_EQUAL(g1->Get(), 1234);

        {
            TMetricSubRegistry subRegistry{{{"common", "label"}}, &registry};
            subRegistry.RemoveMetric(TLabels{{"my", "gauge"}});
        }

        IIntGauge* g2 = registry.IntGauge({{"my", "gauge"}, {"common", "label"}});
        UNIT_ASSERT(g2);
        UNIT_ASSERT_VALUES_EQUAL(g2->Get(), 0);
    }
}