22
33#include < atomic>
44
5+ using namespace std ::chrono_literals;
6+
57namespace Task {
68
79TEST_F (SchedulerTest, scheduleOne) {
@@ -58,14 +60,14 @@ TEST_F(SchedulerTest, scheduleDependentInParallel) {
5860 this ->addTasks ({
5961 { [&p2, &f1] {
6062 p2.set_value ();
61- EXPECT_EQ (std::future_status::ready, f1.wait_for (std::chrono::milliseconds ( Async::kTestTimeout ) ));
63+ EXPECT_EQ (std::future_status::ready, f1.wait_for (Async::kTestTimeout ));
6264 },
63- std::chrono::microseconds ( 0 ) },
65+ 0us },
6466 { [&p1, &f2] {
6567 p1.set_value ();
66- EXPECT_EQ (std::future_status::ready, f2.wait_for (std::chrono::milliseconds ( Async::kTestTimeout ) ));
68+ EXPECT_EQ (std::future_status::ready, f2.wait_for (Async::kTestTimeout ));
6769 },
68- std::chrono::microseconds ( 0 ) },
70+ 0us },
6971 });
7072 this ->runTasks ();
7173}
@@ -82,12 +84,12 @@ TEST_F(SchedulerTest, scheduleDependentSequentially) {
8284 p2.set_value ();
8385 EXPECT_EQ (std::future_status::timeout, f1.wait_for (std::chrono::milliseconds (1 )));
8486 },
85- std::chrono::microseconds ( 0 ) },
87+ 0us },
8688 { [&p1, &f2] {
8789 p1.set_value ();
88- EXPECT_EQ (std::future_status::ready, f2.wait_for (std::chrono::milliseconds ( Async::kTestTimeout ) ));
90+ EXPECT_EQ (std::future_status::ready, f2.wait_for (Async::kTestTimeout ));
8991 },
90- std::chrono::microseconds ( 0 ) },
92+ 0us },
9193 });
9294 this ->runTasks ();
9395}
@@ -99,12 +101,11 @@ TEST_F(SchedulerTest, scheduleNested) {
99101 std::promise<void > promise;
100102 auto future = promise.get_future ();
101103
102- _scheduler->scheduleIn (
103- " 0" , std::chrono::microseconds (0 ), [promise = std::move (promise)]() mutable { promise.set_value (); });
104+ _scheduler->scheduleIn (" 0" , 0us, [promise = std::move (promise)]() mutable { promise.set_value (); });
104105
105- EXPECT_EQ (std::future_status::ready, future.wait_for (std::chrono::milliseconds ( Async::kTestTimeout ) ));
106+ EXPECT_EQ (std::future_status::ready, future.wait_for (Async::kTestTimeout ));
106107 },
107- std::chrono::microseconds ( 0 ) },
108+ 0us },
108109 });
109110
110111 this ->runTasks ();
@@ -124,8 +125,8 @@ TEST_F(SchedulerTest, schedulePeriodic) {
124125 }
125126 });
126127
127- EXPECT_EQ (std::future_status::ready, f.wait_for (std::chrono::milliseconds ( Async::kTestTimeout ) ));
128- EXPECT_EQ (std::future_status::ready, scheduler->stop ().wait_for (std::chrono::milliseconds ( Async::kTestTimeout ) ));
128+ EXPECT_EQ (std::future_status::ready, f.wait_for (Async::kTestTimeout ));
129+ EXPECT_EQ (std::future_status::ready, scheduler->stop ().wait_for (Async::kTestTimeout ));
129130}
130131
131132TEST_F (SchedulerTest, multiplePeriodicTasksOneWorker) {
@@ -154,22 +155,22 @@ TEST_F(SchedulerTest, multiplePeriodicTasksOneWorker) {
154155 });
155156 // Unblock the scheduler.
156157 this ->runTasks ();
157- EXPECT_EQ (std::future_status::ready, f.wait_for (std::chrono::milliseconds ( Async::kTestTimeout ) ));
158+ EXPECT_EQ (std::future_status::ready, f.wait_for (Async::kTestTimeout ));
158159
159160 EXPECT_EQ (expected, final );
160- EXPECT_EQ (std::future_status::ready, scheduler->stop ().wait_for (std::chrono::milliseconds ( Async::kTestTimeout ) ));
161+ EXPECT_EQ (std::future_status::ready, scheduler->stop ().wait_for (Async::kTestTimeout ));
161162}
162163
163164TEST_F (SchedulerTest, erase) {
164165 size_t n = 0 ;
165166
166167 this ->setup (1 , 1 );
167168 this ->addTasks ({
168- { [&n] { ++n; }, std::chrono::microseconds ( 0 ) },
169- { [&n] { ++n; }, std::chrono::microseconds ( 1 ) },
170- { [&n] { ++n; }, std::chrono::microseconds ( 2 ) },
171- { [&n] { ++n; }, std::chrono::microseconds ( 3 ) },
172- { [&n] { ++n; }, std::chrono::microseconds ( 4 ) },
169+ { [&n] { ++n; }, 0us },
170+ { [&n] { ++n; }, 1us },
171+ { [&n] { ++n; }, 2us },
172+ { [&n] { ++n; }, 3us },
173+ { [&n] { ++n; }, 4us },
173174 });
174175 _scheduler->remove (" 1" );
175176 _scheduler->remove (" 4" );
@@ -182,7 +183,7 @@ TEST_F(SchedulerTest, erase) {
182183TEST_F (SchedulerTest, checkTaskIsScheduled) {
183184 this ->setup (1 , 1 );
184185 this ->addTasks ({
185- { [] {}, std::chrono::microseconds ( 0 ) },
186+ { [] {}, 0us },
186187 });
187188 EXPECT_TRUE (_scheduler->isScheduled (" 0" ));
188189 this ->runTasks ();
@@ -193,21 +194,21 @@ TEST_F(SchedulerTest, stopAndDiscard) {
193194
194195 this ->setup (1 , 1 );
195196 this ->addTasks ({
196- { [&n] { n = 1 ; }, std::chrono::minutes ( 1 ) },
197+ { [&n] { n = 1 ; }, 1min },
197198 });
198199 auto done = _scheduler->stop (true );
199200
200201 this ->runTasks ();
201- EXPECT_EQ (std::future_status::ready, done.wait_for (std::chrono::milliseconds ( Async::kTestTimeout ) ));
202+ EXPECT_EQ (std::future_status::ready, done.wait_for (Async::kTestTimeout ));
202203 EXPECT_EQ (0 , n);
203204}
204205
205206TEST_F (SchedulerTest, scheduleOnStopped) {
206207 auto threadpool = std::make_shared<Detail::Threadpool>(2 );
207208 auto scheduler = std::make_shared<Scheduler>(threadpool, 1 );
208209
209- EXPECT_EQ (std::future_status::ready, scheduler->stop ().wait_for (std::chrono::milliseconds ( Async::kTestTimeout ) ));
210- auto future = scheduler->scheduleIn (" 0" , std::chrono::milliseconds ( 0 ) , [] {});
210+ EXPECT_EQ (std::future_status::ready, scheduler->stop ().wait_for (Async::kTestTimeout ));
211+ auto future = scheduler->scheduleIn (" 0" , 0us , [] {});
211212
212213 EXPECT_FALSE (scheduler->isScheduled (" 0" ));
213214 EXPECT_TRUE (future.valid ());
0 commit comments