library/cpp/messagebus/scheduler/scheduler_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

#include <library/cpp/testing/unittest/registar.h>
 
#include "scheduler.h" 
 
#include <library/cpp/messagebus/misc/test_sync.h>
 
using namespace NBus; 
using namespace NBus::NPrivate; 
 
Y_UNIT_TEST_SUITE(TSchedulerTests) {
    struct TSimpleScheduleItem: public IScheduleItem { 
        TTestSync* const TestSync; 
 
        TSimpleScheduleItem(TTestSync* testSync) 
            : IScheduleItem((TInstant::Now() + TDuration::MilliSeconds(1))) 
            , TestSync(testSync) 
        {
        }
 
        void Do() override {
            TestSync->WaitForAndIncrement(0); 
        } 
    }; 
 
    Y_UNIT_TEST(Simple) {
        TTestSync testSync; 
 
        TScheduler scheduler; 
 
        scheduler.Schedule(new TSimpleScheduleItem(&testSync)); 
 
        testSync.WaitForAndIncrement(1); 
 
        scheduler.Stop(); 
    } 
}