aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/sliding_window/sliding_window_ut.cpp
blob: 1e7343a8d3d4a4c641bfa633d115c9ef50877faa (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
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
127
128
129
130
131
132
#include "sliding_window.h"

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

using namespace NSlidingWindow;

Y_UNIT_TEST_SUITE(TSlidingWindowTest) {
    Y_UNIT_TEST(TestSlidingWindowMax) {
        TSlidingWindow<TMaxOperation<unsigned>> w(TDuration::Minutes(5), 5);
        TInstant start = TInstant::MicroSeconds(TDuration::Hours(1).MicroSeconds());
        TInstant now = start;
        w.Update(5, start);                        // ~ ~ ~ ~ 5
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 5); //         ^
        now += TDuration::Minutes(1) + TDuration::Seconds(1);
        w.Update(5, now);                          // 5 ~ ~ ~ 5
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 5); // ^
        now += TDuration::Minutes(1);
        w.Update(3, now);                          // 5 3 ~ ~ 5
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 5); //   ^
        now += TDuration::Minutes(3);
        w.Update(2, now);                          // 5 3 ~ ~ 2
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 5); //         ^
        now += TDuration::Minutes(1);
        w.Update(2, now);                          // 2 3 ~ ~ 2
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 3); // ^
        now += TDuration::Minutes(1);
        w.Update(2, now);                          // 2 2 ~ ~ 2
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 2); //   ^
        now += TDuration::Minutes(5);
        w.Update(1, now);                          // ~ 1 ~ ~ ~
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 1); //   ^

        // update current bucket
        w.Update(2, now);                          // ~ 2 ~ ~ ~
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 2); //   ^

        w.Update(1, now + TDuration::Seconds(30)); // ~ 2 ~ ~ ~
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 2); //   ^

        // test idle
        now += TDuration::Minutes(1);
        w.Update(now);                             // ~ 2 ~ ~ ~
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 2); //     ^

        now += TDuration::Minutes(5); // ~ ~ ~ ~ ~
        UNIT_ASSERT_VALUES_EQUAL(w.Update(now), 0);
    }

    Y_UNIT_TEST(TestSlidingWindowMin) {
        TSlidingWindow<TMinOperation<unsigned>> w(TDuration::Minutes(5), 5);
        TInstant start = TInstant::MicroSeconds(TDuration::Hours(1).MicroSeconds());
        TInstant now = start;
        w.Update(5, start);                        // ~ ~ ~ ~ 5
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 5); //         ^
        now += TDuration::Minutes(1) + TDuration::Seconds(1);
        w.Update(5, now);                          // 5 ~ ~ ~ 5
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 5); // ^
        now += TDuration::Minutes(1);
        w.Update(7, now);                          // 5 7 ~ ~ 5
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 5); //   ^
        now += TDuration::Minutes(3);
        w.Update(8, now);                          // 5 7 ~ ~ 8
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 5); //         ^
        now += TDuration::Minutes(1);
        w.Update(8, now);                          // 8 7 ~ ~ 8
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 7); // ^
        now += TDuration::Minutes(1);
        w.Update(8, now);                          // 8 8 ~ ~ 8
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 8); //   ^
        now += TDuration::Minutes(5);
        w.Update(6, now);                          // ~ 6 ~ ~ ~
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 6); //   ^

        // update current bucket
        w.Update(5, now);                          // ~ 5 ~ ~ ~
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 5); //   ^

        w.Update(6, now + TDuration::Seconds(30)); // ~ 5 ~ ~ ~
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 5); //   ^

        // test idle
        now += TDuration::Minutes(1);
        w.Update(now);                             // ~ 5 ~ ~ ~
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 5); //     ^

        now += TDuration::Minutes(5); // ~ ~ ~ ~ ~
        UNIT_ASSERT_VALUES_EQUAL(w.Update(now), std::numeric_limits<unsigned>::max());
    }

    Y_UNIT_TEST(TestSlidingWindowSum) {
        TSlidingWindow<TSumOperation<unsigned>> w(TDuration::Minutes(5), 5);
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 0); // current sum

        TInstant start = TInstant::MicroSeconds(TDuration::Hours(1).MicroSeconds());
        TInstant now = start;
        w.Update(5, start);                        // 0 0 0 0 5
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 5); //         ^
        now += TDuration::Minutes(1) + TDuration::Seconds(1);
        w.Update(5, now);                           // 5 0 0 0 5
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 10); // ^
        now += TDuration::Minutes(1);
        w.Update(3, now);                           // 5 3 0 0 5
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 13); //   ^
        now += TDuration::Minutes(3);
        w.Update(2, now);                           // 5 3 0 0 2
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 10); //         ^
        now += TDuration::Minutes(1);
        w.Update(2, now);                          // 2 3 0 0 2
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 7); // ^
        now += TDuration::Minutes(1);
        w.Update(2, now);                          // 2 2 0 0 2
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 6); //   ^
        now += TDuration::Minutes(5);
        w.Update(1, now);                          // 0 1 0 0 0
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 1); //   ^

        // update current bucket
        w.Update(2, now);                          // 0 3 0 0 0
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 3); //   ^

        w.Update(1, now + TDuration::Seconds(30)); // 0 4 0 0 0
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 4); //   ^

        // test idle
        now += TDuration::Minutes(1);
        w.Update(now);                             // 0 4 0 0 0
        UNIT_ASSERT_VALUES_EQUAL(w.GetValue(), 4); //     ^

        now += TDuration::Minutes(5); // 0 0 0 0 0
        UNIT_ASSERT_VALUES_EQUAL(w.Update(now), 0);
    }
}