Files
openthread/tests/unit/test_tasklet.cpp
T
Abtin Keshavarzian fc7249861c [tasklet] add Unpost() method (#11101)
This commit adds `Tasklet::Unpost()` to remove a tasklet from the
tasklet scheduler run queue if previously posted.

It also adds a detailed unit test, `test_tasklet`, covering various
uses of `Tasklet`, such as posting single or multiple tasks.,
re-posting a task from its own handler, Un-posting tasks in different
orders.
2025-01-08 10:43:11 -08:00

572 lines
16 KiB
C++

/*
* Copyright (c) 2024, The OpenThread Authors.
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. Neither the name of the copyright holder nor the
* names of its contributors may be used to endorse or promote products
* derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/
#include "test_platform.h"
#include "test_util.h"
#include <openthread/config.h>
#include "instance/instance.hpp"
namespace ot {
#define Log(aMessage) fprintf(stderr, aMessage "\n\r")
static Instance *sInstance = nullptr;
static bool sTask1Handled = false;
static bool sTask2Handled = false;
static bool sTask3Handled = false;
static bool sSignalPendingCalled = false;
static bool sShouldTask3RepostItself = false;
extern "C" void otTaskletsSignalPending(otInstance *aInstance)
{
Log(" otTaskletsSignalPending()");
if (sInstance != nullptr)
{
VerifyOrQuit(aInstance == sInstance);
}
sSignalPendingCalled = true;
}
void ResetTestFlags(void)
{
sTask1Handled = false;
sTask2Handled = false;
sTask3Handled = false;
sSignalPendingCalled = false;
}
void CheckTaskeltFromHandler(Tasklet &aTasklet)
{
VerifyOrQuit(&aTasklet.GetInstance() == sInstance);
VerifyOrQuit(!aTasklet.IsPosted());
}
void HandleTask1(Tasklet &aTasklet)
{
Log(" HandleTask1()");
CheckTaskeltFromHandler(aTasklet);
VerifyOrQuit(!sTask1Handled);
sTask1Handled = true;
}
void HandleTask2(Tasklet &aTasklet)
{
Log(" HandleTask2()");
CheckTaskeltFromHandler(aTasklet);
VerifyOrQuit(!sTask2Handled);
sTask2Handled = true;
}
void HandleTask3(Tasklet &aTasklet)
{
Log(" HandleTask3()");
CheckTaskeltFromHandler(aTasklet);
VerifyOrQuit(!sTask3Handled);
sTask3Handled = true;
if (sShouldTask3RepostItself)
{
aTasklet.Post();
}
}
void TestTasklet(void)
{
Log("TestTasklet");
sInstance = static_cast<Instance *>(testInitInstance());
VerifyOrQuit(sInstance != nullptr);
sInstance->Get<Tasklet::Scheduler>().ProcessQueuedTasklets();
{
Tasklet::Scheduler &scheduler = sInstance->Get<Tasklet::Scheduler>();
Tasklet task1(*sInstance, HandleTask1);
Tasklet task2(*sInstance, HandleTask2);
Tasklet task3(*sInstance, HandleTask3);
VerifyOrQuit(!task1.IsPosted());
VerifyOrQuit(!task2.IsPosted());
VerifyOrQuit(!task3.IsPosted());
VerifyOrQuit(!scheduler.AreTaskletsPending());
//- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Log("Posting a single task");
ResetTestFlags();
task1.Post();
VerifyOrQuit(task1.IsPosted());
VerifyOrQuit(!task2.IsPosted());
VerifyOrQuit(!task3.IsPosted());
VerifyOrQuit(sSignalPendingCalled);
VerifyOrQuit(scheduler.AreTaskletsPending());
scheduler.ProcessQueuedTasklets();
VerifyOrQuit(sTask1Handled);
VerifyOrQuit(!sTask2Handled);
VerifyOrQuit(!sTask3Handled);
VerifyOrQuit(!scheduler.AreTaskletsPending());
//- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Log("Posting multiple tasks");
ResetTestFlags();
task3.Post();
VerifyOrQuit(sSignalPendingCalled);
sSignalPendingCalled = false;
VerifyOrQuit(scheduler.AreTaskletsPending());
task2.Post();
task1.Post();
VerifyOrQuit(!sSignalPendingCalled);
VerifyOrQuit(task1.IsPosted());
VerifyOrQuit(task2.IsPosted());
VerifyOrQuit(task3.IsPosted());
scheduler.ProcessQueuedTasklets();
VerifyOrQuit(sTask1Handled);
VerifyOrQuit(sTask2Handled);
VerifyOrQuit(sTask3Handled);
VerifyOrQuit(!scheduler.AreTaskletsPending());
VerifyOrQuit(!sSignalPendingCalled);
//- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Log("Posting the same task multiple times");
ResetTestFlags();
task2.Post();
VerifyOrQuit(sSignalPendingCalled);
sSignalPendingCalled = false;
VerifyOrQuit(scheduler.AreTaskletsPending());
task2.Post();
VerifyOrQuit(!task1.IsPosted());
VerifyOrQuit(task2.IsPosted());
VerifyOrQuit(!task3.IsPosted());
task1.Post();
task2.Post();
task1.Post();
VerifyOrQuit(!sSignalPendingCalled);
VerifyOrQuit(task1.IsPosted());
VerifyOrQuit(task2.IsPosted());
VerifyOrQuit(!task3.IsPosted());
scheduler.ProcessQueuedTasklets();
VerifyOrQuit(sTask1Handled);
VerifyOrQuit(sTask2Handled);
VerifyOrQuit(!sTask3Handled);
VerifyOrQuit(!scheduler.AreTaskletsPending());
VerifyOrQuit(!sSignalPendingCalled);
//- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Log("Task posting itself from its handler");
ResetTestFlags();
sShouldTask3RepostItself = true;
task3.Post();
VerifyOrQuit(sSignalPendingCalled);
sSignalPendingCalled = false;
VerifyOrQuit(scheduler.AreTaskletsPending());
task2.Post();
VerifyOrQuit(!sSignalPendingCalled);
VerifyOrQuit(scheduler.AreTaskletsPending());
VerifyOrQuit(!task1.IsPosted());
VerifyOrQuit(task2.IsPosted());
VerifyOrQuit(task3.IsPosted());
scheduler.ProcessQueuedTasklets();
VerifyOrQuit(!sTask1Handled);
VerifyOrQuit(sTask2Handled);
VerifyOrQuit(sTask3Handled);
VerifyOrQuit(scheduler.AreTaskletsPending());
VerifyOrQuit(sSignalPendingCalled);
ResetTestFlags();
sShouldTask3RepostItself = false;
scheduler.ProcessQueuedTasklets();
VerifyOrQuit(!sTask1Handled);
VerifyOrQuit(!sTask2Handled);
VerifyOrQuit(sTask3Handled);
VerifyOrQuit(!sSignalPendingCalled);
VerifyOrQuit(!scheduler.AreTaskletsPending());
//- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Log("Un-posting a single posted task");
ResetTestFlags();
task1.Post();
VerifyOrQuit(sSignalPendingCalled);
sSignalPendingCalled = false;
VerifyOrQuit(scheduler.AreTaskletsPending());
VerifyOrQuit(task1.IsPosted());
task1.Unpost();
VerifyOrQuit(!sSignalPendingCalled);
VerifyOrQuit(!scheduler.AreTaskletsPending());
VerifyOrQuit(!task1.IsPosted());
scheduler.ProcessQueuedTasklets();
VerifyOrQuit(!sTask1Handled);
VerifyOrQuit(!sTask2Handled);
VerifyOrQuit(!sTask3Handled);
VerifyOrQuit(!sSignalPendingCalled);
VerifyOrQuit(!scheduler.AreTaskletsPending());
//- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Log("Post multiple tasks and then un-post the first one");
ResetTestFlags();
task1.Post();
task2.Post();
task3.Post();
VerifyOrQuit(sSignalPendingCalled);
sSignalPendingCalled = false;
VerifyOrQuit(scheduler.AreTaskletsPending());
VerifyOrQuit(task1.IsPosted());
VerifyOrQuit(task2.IsPosted());
VerifyOrQuit(task3.IsPosted());
task1.Unpost();
VerifyOrQuit(!task1.IsPosted());
VerifyOrQuit(task2.IsPosted());
VerifyOrQuit(task3.IsPosted());
scheduler.ProcessQueuedTasklets();
VerifyOrQuit(!sTask1Handled);
VerifyOrQuit(sTask2Handled);
VerifyOrQuit(sTask3Handled);
VerifyOrQuit(!sSignalPendingCalled);
VerifyOrQuit(!scheduler.AreTaskletsPending());
//- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Log("Post multiple tasks and then un-post the middle one");
ResetTestFlags();
task1.Post();
task2.Post();
task3.Post();
VerifyOrQuit(sSignalPendingCalled);
sSignalPendingCalled = false;
VerifyOrQuit(scheduler.AreTaskletsPending());
VerifyOrQuit(task1.IsPosted());
VerifyOrQuit(task2.IsPosted());
VerifyOrQuit(task3.IsPosted());
task2.Unpost();
VerifyOrQuit(task1.IsPosted());
VerifyOrQuit(!task2.IsPosted());
VerifyOrQuit(task3.IsPosted());
scheduler.ProcessQueuedTasklets();
VerifyOrQuit(sTask1Handled);
VerifyOrQuit(!sTask2Handled);
VerifyOrQuit(sTask3Handled);
VerifyOrQuit(!sSignalPendingCalled);
VerifyOrQuit(!scheduler.AreTaskletsPending());
//- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Log("Post multiple tasks and then un-post the last one");
ResetTestFlags();
task1.Post();
task2.Post();
task3.Post();
VerifyOrQuit(sSignalPendingCalled);
sSignalPendingCalled = false;
VerifyOrQuit(scheduler.AreTaskletsPending());
VerifyOrQuit(task1.IsPosted());
VerifyOrQuit(task2.IsPosted());
VerifyOrQuit(task3.IsPosted());
task3.Unpost();
VerifyOrQuit(task1.IsPosted());
VerifyOrQuit(task2.IsPosted());
VerifyOrQuit(!task3.IsPosted());
scheduler.ProcessQueuedTasklets();
VerifyOrQuit(sTask1Handled);
VerifyOrQuit(sTask2Handled);
VerifyOrQuit(!sTask3Handled);
VerifyOrQuit(!sSignalPendingCalled);
VerifyOrQuit(!scheduler.AreTaskletsPending());
//- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Log("Post multiple tasks and then un-post the first and last ones");
ResetTestFlags();
task1.Post();
task2.Post();
task3.Post();
VerifyOrQuit(sSignalPendingCalled);
sSignalPendingCalled = false;
VerifyOrQuit(scheduler.AreTaskletsPending());
VerifyOrQuit(task1.IsPosted());
VerifyOrQuit(task2.IsPosted());
VerifyOrQuit(task3.IsPosted());
task1.Unpost();
task3.Unpost();
VerifyOrQuit(!task1.IsPosted());
VerifyOrQuit(task2.IsPosted());
VerifyOrQuit(!task3.IsPosted());
scheduler.ProcessQueuedTasklets();
VerifyOrQuit(!sTask1Handled);
VerifyOrQuit(sTask2Handled);
VerifyOrQuit(!sTask3Handled);
VerifyOrQuit(!sSignalPendingCalled);
VerifyOrQuit(!scheduler.AreTaskletsPending());
//- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Log("Post multiple tasks and then un-post all in the same order added");
ResetTestFlags();
task1.Post();
task2.Post();
task3.Post();
VerifyOrQuit(sSignalPendingCalled);
sSignalPendingCalled = false;
VerifyOrQuit(scheduler.AreTaskletsPending());
VerifyOrQuit(task1.IsPosted());
VerifyOrQuit(task2.IsPosted());
VerifyOrQuit(task3.IsPosted());
task1.Unpost();
task2.Unpost();
task3.Unpost();
VerifyOrQuit(!task1.IsPosted());
VerifyOrQuit(!task2.IsPosted());
VerifyOrQuit(!task3.IsPosted());
VerifyOrQuit(!scheduler.AreTaskletsPending());
scheduler.ProcessQueuedTasklets();
VerifyOrQuit(!sTask1Handled);
VerifyOrQuit(!sTask2Handled);
VerifyOrQuit(!sTask3Handled);
VerifyOrQuit(!sSignalPendingCalled);
VerifyOrQuit(!scheduler.AreTaskletsPending());
//- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Log("Post multiple tasks and then un-post all in the reverse order added");
ResetTestFlags();
task1.Post();
task2.Post();
task3.Post();
VerifyOrQuit(sSignalPendingCalled);
sSignalPendingCalled = false;
VerifyOrQuit(scheduler.AreTaskletsPending());
VerifyOrQuit(task1.IsPosted());
VerifyOrQuit(task2.IsPosted());
VerifyOrQuit(task3.IsPosted());
task3.Unpost();
task2.Unpost();
task1.Unpost();
VerifyOrQuit(!task1.IsPosted());
VerifyOrQuit(!task2.IsPosted());
VerifyOrQuit(!task3.IsPosted());
VerifyOrQuit(!scheduler.AreTaskletsPending());
scheduler.ProcessQueuedTasklets();
VerifyOrQuit(!sTask1Handled);
VerifyOrQuit(!sTask2Handled);
VerifyOrQuit(!sTask3Handled);
VerifyOrQuit(!sSignalPendingCalled);
VerifyOrQuit(!scheduler.AreTaskletsPending());
//- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Log("Post multiple tasks and then un-post all in different order (middle first)");
ResetTestFlags();
task1.Post();
task2.Post();
task3.Post();
VerifyOrQuit(sSignalPendingCalled);
sSignalPendingCalled = false;
VerifyOrQuit(scheduler.AreTaskletsPending());
VerifyOrQuit(task1.IsPosted());
VerifyOrQuit(task2.IsPosted());
VerifyOrQuit(task3.IsPosted());
task2.Unpost();
task3.Unpost();
task1.Unpost();
VerifyOrQuit(!task1.IsPosted());
VerifyOrQuit(!task2.IsPosted());
VerifyOrQuit(!task3.IsPosted());
VerifyOrQuit(!scheduler.AreTaskletsPending());
scheduler.ProcessQueuedTasklets();
VerifyOrQuit(!sTask1Handled);
VerifyOrQuit(!sTask2Handled);
VerifyOrQuit(!sTask3Handled);
VerifyOrQuit(!sSignalPendingCalled);
VerifyOrQuit(!scheduler.AreTaskletsPending());
//- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Log("Un-posting tasks not yet posted");
ResetTestFlags();
task1.Unpost();
VerifyOrQuit(!task1.IsPosted());
VerifyOrQuit(!sSignalPendingCalled);
VerifyOrQuit(!scheduler.AreTaskletsPending());
task2.Post();
VerifyOrQuit(sSignalPendingCalled);
sSignalPendingCalled = false;
VerifyOrQuit(scheduler.AreTaskletsPending());
task1.Unpost();
task3.Unpost();
VerifyOrQuit(!task1.IsPosted());
VerifyOrQuit(task2.IsPosted());
VerifyOrQuit(!task3.IsPosted());
VerifyOrQuit(scheduler.AreTaskletsPending());
task3.Post();
VerifyOrQuit(!task1.IsPosted());
VerifyOrQuit(task2.IsPosted());
VerifyOrQuit(task3.IsPosted());
VerifyOrQuit(!sSignalPendingCalled);
VerifyOrQuit(scheduler.AreTaskletsPending());
scheduler.ProcessQueuedTasklets();
VerifyOrQuit(!sTask1Handled);
VerifyOrQuit(sTask2Handled);
VerifyOrQuit(sTask3Handled);
VerifyOrQuit(!sSignalPendingCalled);
VerifyOrQuit(!scheduler.AreTaskletsPending());
}
}
} // namespace ot
int main(void)
{
ot::TestTasklet();
printf("All tests passed\n");
return 0;
}