mirror of https://github.com/PentHertz/srsLTE.git
86 lines
2.9 KiB
C++
86 lines
2.9 KiB
C++
|
/*
|
||
|
|
* Copyright 2013-2020 Software Radio Systems Limited
|
||
|
|
*
|
||
|
|
* This file is part of srsLTE.
|
||
|
|
*
|
||
|
|
* srsLTE is free software: you can redistribute it and/or modify
|
||
|
|
* it under the terms of the GNU Affero General Public License as
|
||
|
|
* published by the Free Software Foundation, either version 3 of
|
||
|
|
* the License, or (at your option) any later version.
|
||
|
|
*
|
||
|
|
* srsLTE is distributed in the hope that it will be useful,
|
||
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
* GNU Affero General Public License for more details.
|
||
|
|
*
|
||
|
|
* A copy of the GNU Affero General Public License can be found in
|
||
|
|
* the LICENSE file in the top-level directory of this distribution
|
||
|
|
* and at http://www.gnu.org/licenses/.
|
||
|
|
*
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "srslte/common/task_scheduler.h"
|
||
|
|
#include "srslte/common/test_common.h"
|
||
|
|
|
||
|
|
enum class task_result { null, internal, external, timer };
|
||
|
|
|
||
|
|
int test_task_scheduler_no_pool()
|
||
|
|
{
|
||
|
|
srslte::task_scheduler task_sched{5, 0};
|
||
|
|
task_result state = task_result::null;
|
||
|
|
|
||
|
|
// TEST: deferring task does not run the task until the next tic
|
||
|
|
task_sched.defer_task([&state]() { state = task_result::internal; });
|
||
|
|
TESTASSERT(state == task_result::null);
|
||
|
|
task_sched.tic();
|
||
|
|
TESTASSERT(state == task_result::internal);
|
||
|
|
|
||
|
|
// TEST: check delaying of task
|
||
|
|
state = task_result::null;
|
||
|
|
int dur = 5;
|
||
|
|
task_sched.defer_callback(dur, [&state]() { state = task_result::timer; });
|
||
|
|
for (int i = 0; i < dur; ++i) {
|
||
|
|
TESTASSERT(state == task_result::null);
|
||
|
|
task_sched.tic();
|
||
|
|
}
|
||
|
|
TESTASSERT(state == task_result::timer);
|
||
|
|
|
||
|
|
// TEST: background task is run, despite there are no pool workers
|
||
|
|
state = task_result::null;
|
||
|
|
task_sched.enqueue_background_task([&task_sched, &state](uint32_t worker_id) {
|
||
|
|
task_sched.notify_background_task_result([&state]() { state = task_result::external; });
|
||
|
|
});
|
||
|
|
TESTASSERT(state == task_result::null);
|
||
|
|
task_sched.tic();
|
||
|
|
TESTASSERT(state == task_result::null);
|
||
|
|
task_sched.run_next_external_task(); // runs background task
|
||
|
|
TESTASSERT(state == task_result::null);
|
||
|
|
task_sched.run_next_external_task(); // runs notification
|
||
|
|
TESTASSERT(state == task_result::external);
|
||
|
|
|
||
|
|
return SRSLTE_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
int test_task_scheduler_with_pool()
|
||
|
|
{
|
||
|
|
srslte::task_scheduler task_sched{5, 2};
|
||
|
|
task_result state = task_result::null;
|
||
|
|
|
||
|
|
task_sched.enqueue_background_task([&task_sched, &state](uint32_t worker_id) {
|
||
|
|
task_sched.notify_background_task_result([&state]() { state = task_result::external; });
|
||
|
|
});
|
||
|
|
TESTASSERT(state == task_result::null);
|
||
|
|
task_sched.tic();
|
||
|
|
TESTASSERT(state == task_result::null);
|
||
|
|
task_sched.run_next_external_task(); // waits and runs notification
|
||
|
|
TESTASSERT(state == task_result::external);
|
||
|
|
|
||
|
|
return SRSLTE_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
int main()
|
||
|
|
{
|
||
|
|
TESTASSERT(test_task_scheduler_no_pool() == SRSLTE_SUCCESS);
|
||
|
|
TESTASSERT(test_task_scheduler_with_pool() == SRSLTE_SUCCESS);
|
||
|
|
}
|