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
|
#include "mkql_stats_registry.h"
#include <library/cpp/testing/unittest/registar.h>
using namespace NKikimr;
using namespace NMiniKQL;
Y_UNIT_TEST_SUITE(TStatsRegistryTest) {
Y_UNIT_TEST(Stats) {
static TStatKey key("key", false);
auto stats = CreateDefaultStatsRegistry();
stats->SetStat(key, 37);
UNIT_ASSERT_EQUAL(stats->GetStat(key), 37);
stats->AddStat(key, 3);
UNIT_ASSERT_EQUAL(stats->GetStat(key), 40);
stats->IncStat(key);
UNIT_ASSERT_EQUAL(stats->GetStat(key), 41);
stats->DecStat(key);
UNIT_ASSERT_EQUAL(stats->GetStat(key), 40);
stats->SetMaxStat(key, 3);
UNIT_ASSERT_EQUAL(stats->GetStat(key), 40);
stats->SetMaxStat(key, 43);
UNIT_ASSERT_EQUAL(stats->GetStat(key), 43);
stats->SetMinStat(key, 57);
UNIT_ASSERT_EQUAL(stats->GetStat(key), 43);
stats->SetMinStat(key, 34);
UNIT_ASSERT_EQUAL(stats->GetStat(key), 34);
}
Y_UNIT_TEST(ForEach) {
static TStatKey key1("key1", false), key2("key2", true);
UNIT_ASSERT(!key1.IsDeriv());
UNIT_ASSERT(key2.IsDeriv());
auto stats = CreateDefaultStatsRegistry();
stats->SetStat(key1, 42);
stats->SetStat(key2, 37);
stats->ForEachStat([](const TStatKey& key, i64 value) {
if (key.GetName() == "key1") {
UNIT_ASSERT_EQUAL(value, 42);
} else if (key.GetName() == "key2") {
UNIT_ASSERT_EQUAL(value, 37);
} else {
// depending on how to run tests other keys can be linked
// together and appears here
}
});
}
Y_UNIT_TEST(DuplicateKeys) {
TString error;
try {
static TStatKey key("my_key", false), sameKey("my_key", false);
// avoid variables elimitation
Cerr << key.GetId() << ": " << key.GetName() << '\n'
<< sameKey.GetId() << ": " << sameKey.GetName() << Endl;
} catch (const yexception& e) {
error = e.AsStrBuf();
}
UNIT_ASSERT(error.Contains("duplicated stat key: my_key"));
}
}
|
