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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
// Copyright (C) 2015-2020 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
#include <config.h>
#include <stats/context.h>
#include <gtest/gtest.h>
#include <util/chrono_time_utils.h>
#include <string>
using namespace isc::data;
using namespace isc::stats;
using namespace std;
using namespace std::chrono;
// Basic test that checks get, add, del methods
TEST(ContextTest, basic) {
// Let's create a couple observations. Using floating point,
// as they're easiest to initialize.
ObservationPtr a(new Observation("alpha", 1.11));
ObservationPtr b(new Observation("beta", 2.22));
ObservationPtr c(new Observation("gamma", 3.33));
string expected_a = a->getJSON()->str();
string expected_b = b->getJSON()->str();
string expected_c = c->getJSON()->str();
// Context where we will store the observations.
StatContext ctx;
// By default the context does not hold any statistics.
EXPECT_EQ(0, ctx.size());
// It should be possible to add 'a' statistic
EXPECT_NO_THROW(ctx.add(a));
// We can't add a duplicate.
EXPECT_THROW(ctx.add(a), DuplicateStat);
// It should be ok to add other statistics
EXPECT_NO_THROW(ctx.add(b));
EXPECT_NO_THROW(ctx.add(c));
// By now we should have 3 statistics recorded
EXPECT_EQ(3, ctx.size());
// Let's try to retrieve them
ObservationPtr from_ctx;
EXPECT_NO_THROW(from_ctx = ctx.get("alpha"));
ASSERT_TRUE(from_ctx);
EXPECT_EQ(expected_a, from_ctx->getJSON()->str());
EXPECT_NO_THROW(from_ctx = ctx.get("beta"));
ASSERT_TRUE(from_ctx);
EXPECT_EQ(expected_b, from_ctx->getJSON()->str());
EXPECT_NO_THROW(from_ctx = ctx.get("gamma"));
ASSERT_TRUE(from_ctx);
EXPECT_EQ(expected_c, from_ctx->getJSON()->str());
// Let's try to retrieve non-existing stat
EXPECT_NO_THROW(from_ctx = ctx.get("delta"));
EXPECT_FALSE(from_ctx);
// Now delete one of the stats...
EXPECT_TRUE(ctx.del("beta"));
// ... and check that it's really gone.
EXPECT_FALSE(ctx.get("beta"));
// Attempt to delete non-existing stat should fail.
EXPECT_FALSE(ctx.del("beta"));
ConstElementPtr result;
EXPECT_NO_THROW(result = ctx.getAll());
ASSERT_TRUE(result);
ElementPtr expected_result = Element::createMap();
expected_result->set("alpha", a->getJSON());
expected_result->set("gamma", c->getJSON());
EXPECT_EQ(result->str(), expected_result->str());
// Reset all statistics.
EXPECT_NO_THROW(ctx.resetAll());
EXPECT_NO_THROW(from_ctx = ctx.get("alpha"));
ASSERT_TRUE(from_ctx);
EXPECT_NE(expected_a, from_ctx->getJSON()->str());
EXPECT_EQ(0.0, a->getFloat().first);
EXPECT_NO_THROW(from_ctx = ctx.get("gamma"));
ASSERT_TRUE(from_ctx);
EXPECT_NE(expected_c, from_ctx->getJSON()->str());
EXPECT_EQ(0.0, c->getFloat().first);
// Set sample count for all statistics
EXPECT_NO_THROW(ctx.setMaxSampleCountAll(50));
EXPECT_NO_THROW(from_ctx = ctx.get("alpha"));
ASSERT_TRUE(from_ctx);
EXPECT_EQ(from_ctx->getMaxSampleCount().second, 50);
EXPECT_NO_THROW(from_ctx = ctx.get("gamma"));
ASSERT_TRUE(from_ctx);
EXPECT_EQ(from_ctx->getMaxSampleCount().second, 50);
// Set sample age for all statistics
const StatsDuration& dur(minutes(4) + seconds(5) + milliseconds(3));
EXPECT_NO_THROW(ctx.setMaxSampleAgeAll(dur));
EXPECT_NO_THROW(from_ctx = ctx.get("alpha"));
ASSERT_TRUE(from_ctx);
EXPECT_EQ(from_ctx->getMaxSampleAge().second, dur);
EXPECT_NO_THROW(from_ctx = ctx.get("gamma"));
ASSERT_TRUE(from_ctx);
EXPECT_EQ(from_ctx->getMaxSampleAge().second, dur);
// Clear all statistics.
EXPECT_NO_THROW(ctx.clear());
EXPECT_EQ(0, ctx.size());
}
|
