Commit 63e41b4e by Sachin Basil John

more formatting

parent 0382e892
...@@ -37,96 +37,97 @@ ...@@ -37,96 +37,97 @@
#define TASK_TEST_ID 23 #define TASK_TEST_ID 23
static void testTaskAction( static void testTaskAction(
char const * msg, char const * msg,
std::string * output, std::string * output,
embb::tasks::TaskContext & /*context*/) { embb::tasks::TaskContext & /*context*/) {
*output = msg; *output = msg;
} }
static void testRecursiveTaskAction( static void testRecursiveTaskAction(
int * value, int * value,
embb::tasks::TaskContext & /*context*/) { embb::tasks::TaskContext & /*context*/) {
embb::tasks::Node & node = embb::tasks::Node::GetInstance(); embb::tasks::Node & node = embb::tasks::Node::GetInstance();
*value = *value + 1; *value = *value + 1;
if (*value < 1000) { if (*value < 1000) {
embb::tasks::Task task = node.Spawn( embb::tasks::Task task = node.Spawn(
embb::base::Bind( embb::base::Bind(
testRecursiveTaskAction, value, embb::base::Placeholder::_1)); testRecursiveTaskAction, value, embb::base::Placeholder::_1));
task.Wait(MTAPI_INFINITE); task.Wait(MTAPI_INFINITE);
} }
PT_EXPECT(*value == 1000); PT_EXPECT(*value == 1000);
} }
static void testErrorTaskAction(embb::tasks::TaskContext & context) { static void testErrorTaskAction(embb::tasks::TaskContext & context) {
context.SetStatus(MTAPI_ERR_ACTION_FAILED); context.SetStatus(MTAPI_ERR_ACTION_FAILED);
} }
static void testDoSomethingElse() { } static void testDoSomethingElse() {
}
TaskTest::TaskTest() { TaskTest::TaskTest() {
CreateUnit("tasks_cpp task test").Add(&TaskTest::TestBasic, this); CreateUnit("tasks_cpp task test").Add(&TaskTest::TestBasic, this);
} }
void TaskTest::TestBasic() { void TaskTest::TestBasic() {
embb::tasks::Node::Initialize(THIS_DOMAIN_ID, THIS_NODE_ID); embb::tasks::Node::Initialize(THIS_DOMAIN_ID, THIS_NODE_ID);
embb::tasks::Node & node = embb::tasks::Node::GetInstance(); embb::tasks::Node & node = embb::tasks::Node::GetInstance();
embb::tasks::ExecutionPolicy policy(false); embb::tasks::ExecutionPolicy policy(false);
PT_EXPECT_EQ(policy.GetAffinity(), 0u); PT_EXPECT_EQ(policy.GetAffinity(), 0u);
PT_EXPECT_EQ(policy.GetPriority(), 0u); PT_EXPECT_EQ(policy.GetPriority(), 0u);
policy.AddWorker(0u); policy.AddWorker(0u);
PT_EXPECT_EQ(policy.GetAffinity(), 1u); PT_EXPECT_EQ(policy.GetAffinity(), 1u);
if (policy.GetCoreCount() > 1) { if (policy.GetCoreCount() > 1) {
policy.AddWorker(1u); policy.AddWorker(1u);
PT_EXPECT_EQ(policy.GetAffinity(), 3u); PT_EXPECT_EQ(policy.GetAffinity(), 3u);
} }
policy.RemoveWorker(0u); policy.RemoveWorker(0u);
PT_EXPECT_EQ(policy.IsSetWorker(0), false); PT_EXPECT_EQ(policy.IsSetWorker(0), false);
if (policy.GetCoreCount() > 1) { if (policy.GetCoreCount() > 1) {
PT_EXPECT_EQ(policy.GetAffinity(), 2u); PT_EXPECT_EQ(policy.GetAffinity(), 2u);
PT_EXPECT_EQ(policy.IsSetWorker(1), true); PT_EXPECT_EQ(policy.IsSetWorker(1), true);
} }
std::string test; std::string test;
embb::tasks::Task task = node.Spawn( embb::tasks::Task task = node.Spawn(
embb::base::Bind( embb::base::Bind(
testTaskAction, "simple", &test, embb::base::Placeholder::_1)); testTaskAction, "simple", &test, embb::base::Placeholder::_1));
testDoSomethingElse(); testDoSomethingElse();
task.Wait(MTAPI_INFINITE); task.Wait(MTAPI_INFINITE);
PT_EXPECT(test == "simple"); PT_EXPECT(test == "simple");
std::string test1, test2, test3; std::string test1, test2, test3;
task = node.First( task = node.First(
embb::base::Bind( embb::base::Bind(
testTaskAction, "first", &test1, embb::base::Placeholder::_1)). testTaskAction, "first", &test1, embb::base::Placeholder::_1)).
Then(embb::base::Bind( Then(embb::base::Bind(
testTaskAction, "second", &test2, embb::base::Placeholder::_1)). testTaskAction, "second", &test2, embb::base::Placeholder::_1)).
Then(embb::base::Bind( Then(embb::base::Bind(
testTaskAction, "third", &test3, embb::base::Placeholder::_1)). testTaskAction, "third", &test3, embb::base::Placeholder::_1)).
Spawn(); Spawn();
testDoSomethingElse(); testDoSomethingElse();
task.Wait(MTAPI_INFINITE); task.Wait(MTAPI_INFINITE);
PT_EXPECT(test1 == "first"); PT_EXPECT(test1 == "first");
PT_EXPECT(test2 == "second"); PT_EXPECT(test2 == "second");
PT_EXPECT(test3 == "third"); PT_EXPECT(test3 == "third");
int value = 0; int value = 0;
task = node.Spawn( task = node.Spawn(
embb::base::Bind( embb::base::Bind(
testRecursiveTaskAction, &value, embb::base::Placeholder::_1)); testRecursiveTaskAction, &value, embb::base::Placeholder::_1));
task.Wait(MTAPI_INFINITE); task.Wait(MTAPI_INFINITE);
PT_EXPECT(value == 1000); PT_EXPECT(value == 1000);
mtapi_status_t status; mtapi_status_t status;
task = node.Spawn(testErrorTaskAction); task = node.Spawn(testErrorTaskAction);
testDoSomethingElse(); testDoSomethingElse();
status = task.Wait(MTAPI_INFINITE); status = task.Wait(MTAPI_INFINITE);
PT_EXPECT(MTAPI_ERR_ACTION_FAILED == status); PT_EXPECT(MTAPI_ERR_ACTION_FAILED == status);
embb::tasks::Node::Finalize(); embb::tasks::Node::Finalize();
PT_EXPECT(embb_get_bytes_allocated() == 0); PT_EXPECT(embb_get_bytes_allocated() == 0);
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment