aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/threading/future/legacy_future_ut.cpp
blob: b0c9dc21aa1bd567706e86f993bb086ac61ce10d (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
#include "legacy_future.h"

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

namespace NThreading {
    Y_UNIT_TEST_SUITE(TLegacyFutureTest) { 
        int intf() {
            return 17;
        }

        Y_UNIT_TEST(TestIntFunction) { 
            TLegacyFuture<int> f((&intf));
            UNIT_ASSERT_VALUES_EQUAL(17, f.Get());
        }

        static int r;

        void voidf() {
            r = 18;
        }

        Y_UNIT_TEST(TestVoidFunction) { 
            r = 0;
            TLegacyFuture<> f((&voidf));
            f.Get();
            UNIT_ASSERT_VALUES_EQUAL(18, r);
        }

        struct TSampleClass {
            int mValue;

            TSampleClass(int value)
                : mValue(value)
            {
            }

            int Calc() {
                return mValue + 1;
            }
        };

        Y_UNIT_TEST(TestMethod) { 
            TLegacyFuture<int> f11(std::bind(&TSampleClass::Calc, TSampleClass(3)));
            UNIT_ASSERT_VALUES_EQUAL(4, f11.Get());

            TLegacyFuture<int> f12(std::bind(&TSampleClass::Calc, TSampleClass(3)), SystemThreadFactory());
            UNIT_ASSERT_VALUES_EQUAL(4, f12.Get());

            TSampleClass c(5);

            TLegacyFuture<int> f21(std::bind(&TSampleClass::Calc, std::ref(c)));
            UNIT_ASSERT_VALUES_EQUAL(6, f21.Get());

            TLegacyFuture<int> f22(std::bind(&TSampleClass::Calc, std::ref(c)), SystemThreadFactory());
            UNIT_ASSERT_VALUES_EQUAL(6, f22.Get());
        }

        struct TSomeThreadPool: public IThreadFactory {};

        Y_UNIT_TEST(TestFunction) { 
            std::function<int()> f((&intf));

            UNIT_ASSERT_VALUES_EQUAL(17, TLegacyFuture<int>(f).Get());
            UNIT_ASSERT_VALUES_EQUAL(17, TLegacyFuture<int>(f, SystemThreadFactory()).Get());

            if (false) {
                TSomeThreadPool* q = nullptr;
                TLegacyFuture<int>(f, q); // just check compiles, do not start
            }
        }
    }

}