blob: d5afe7e4b1153ea74d390fb309248f77b7540c35 (
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
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#define BOOST_TEST_DYN_LINK
#define BOOST_TEST_MAIN
#include <boost/test/unit_test.hpp>
#include <vespa/filedistribution/distributor/scheduler.h>
#include <iostream>
#include <boost/thread/barrier.hpp>
using filedistribution::Scheduler;
using namespace std::literals;
namespace asio = boost::asio;
class TestException {};
struct CallRun {
volatile bool _caughtException;
CallRun()
:_caughtException(false)
{}
void operator()(asio::io_service& ioService) {
try {
//No reset needed after handling exceptions.
ioService.run();
} catch(const TestException& e ) {
_caughtException = true;
}
}
};
struct Fixture {
CallRun callRun;
Scheduler scheduler;
Fixture()
: scheduler(std::ref(callRun))
{}
};
BOOST_FIXTURE_TEST_SUITE(SchedulerTest, Fixture)
struct RepeatedTask : Scheduler::Task {
void doHandle() override {
std::cout <<"RepeatedTask::doHandle " <<std::endl;
schedule(boost::posix_time::seconds(1));
}
RepeatedTask(Scheduler& scheduler) : Task(scheduler) {}
};
BOOST_AUTO_TEST_CASE(require_tasks_does_not_keep_scheduler_alive) {
RepeatedTask::SP task(new RepeatedTask(scheduler));
task->schedule(boost::posix_time::hours(10));
}
struct EnsureInvokedTask : Scheduler::Task {
boost::barrier& _barrier;
void doHandle() override {
_barrier.wait();
}
EnsureInvokedTask(Scheduler& scheduler, boost::barrier& barrier) :
Task(scheduler),
_barrier(barrier)
{}
};
BOOST_AUTO_TEST_CASE(require_task_invoked) {
boost::barrier barrier(2);
EnsureInvokedTask::SP task(new EnsureInvokedTask(scheduler, barrier));
task->schedule(boost::posix_time::milliseconds(50));
barrier.wait();
}
struct ThrowExceptionTask : Scheduler::Task {
void doHandle() override {
throw TestException();
}
ThrowExceptionTask(Scheduler& scheduler) :
Task(scheduler)
{}
};
BOOST_AUTO_TEST_CASE(require_exception_from_tasks_can_be_caught) {
ThrowExceptionTask::SP task(new ThrowExceptionTask(scheduler));
task->scheduleNow();
for (int i=0; i<200 && !callRun._caughtException; ++i) {
std::this_thread::sleep_for(100ms);
}
BOOST_CHECK(callRun._caughtException);
}
BOOST_AUTO_TEST_SUITE_END()
|