From a6f63598adb880a75e1571aac58338c17fa7ad53 Mon Sep 17 00:00:00 2001 From: Amiti Uttarwar Date: Tue, 28 Jan 2020 15:26:32 -0800 Subject: [util] allow scheduler to be mocked Add MockForward method to the scheduler that mimics going into the future by rescheduling all items on the taskQueue to be sooner. --- src/scheduler.h | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'src/scheduler.h') diff --git a/src/scheduler.h b/src/scheduler.h index 7080adf34c..d18be0ea5e 100644 --- a/src/scheduler.h +++ b/src/scheduler.h @@ -55,6 +55,13 @@ public: // need more accurate scheduling, don't use this method. void scheduleEvery(Function f, int64_t deltaMilliSeconds); + /** + * Mock the scheduler to fast forward in time. + * Iterates through items on taskQueue and reschedules them + * to be delta_seconds sooner. + */ + void MockForward(boost::chrono::seconds delta_seconds); + // To keep things as simple as possible, there is no unschedule. // Services the queue 'forever'. Should be run in a thread, -- cgit v1.2.3