library/cpp/tvmauth/client/ut/exponential_backoff_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

#include <library/cpp/tvmauth/client/misc/exponential_backoff.h>

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

#include <thread>

using namespace NTvmAuth;

Y_UNIT_TEST_SUITE(PasspUtilsExpBackoff) {
    Y_UNIT_TEST(common) {
        TExponentialBackoff b({TDuration::Seconds(1), TDuration::Seconds(60), 2, 0.01});

        UNIT_ASSERT_VALUES_EQUAL(TDuration::Seconds(1), b.GetCurrentValue());

        TDuration dur = b.GetCurrentValue();
        for (size_t idx = 0; idx < 6; ++idx) {
            TDuration newValue = b.Increase();
            UNIT_ASSERT_LT(dur, newValue);
            dur = newValue;
        }

        UNIT_ASSERT_LT(TDuration::Seconds(60) - TDuration::Seconds(3), dur);
        UNIT_ASSERT_LT(dur, TDuration::Seconds(60) + TDuration::Seconds(3));
    }

    Y_UNIT_TEST(sleep) {
        TExponentialBackoff b({TDuration::Seconds(60), TDuration::Seconds(600), 2, 0.01});

        const TInstant start = TInstant::Now();

        TAutoEvent started;
        std::thread t([&b, &started]() {
            started.Signal();
            b.Sleep();
        });

        started.WaitT(TDuration::Seconds(30));
        b.Interrupt();
        t.join();
        TDuration dur = TInstant::Now() - start;

        UNIT_ASSERT_LT(dur, TDuration::Seconds(60));
    }
}