uthread/test/task_queue.cpp at master · andreimaximov/uthread · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
#include <gtest/gtest.h>

#include <uthread/task.hpp>
#include <uthread/task_loop.hpp>
#include <uthread/task_queue.hpp>

namespace uthread {

TEST(TaskQueueTest, ParkAndUnpark) {
  int x = 0;
  TaskQueue queue;

  auto f = [&x, &queue]() {
    ASSERT_EQ(x, 0);
    x++;
    queue.park();
    ASSERT_EQ(x, 2);
    x++;
  };

  auto g = [&x, &queue]() {
    ASSERT_EQ(x, 1);
    x++;
    ASSERT_TRUE(queue.unpark());
    Task::yield();
    ASSERT_EQ(x, 3);
    x++;
  };

  TaskLoop taskLoop;
  taskLoop.addTask(f);
  taskLoop.addTask(g);
  taskLoop.runLoop();

  ASSERT_EQ(x, 4);
}

TEST(TaskQueueTest, UnparkNoTasks) {
  TaskQueue queue;

  auto f = [&queue]() { ASSERT_FALSE(queue.unpark()); };

  TaskLoop taskLoop;
  taskLoop.addTask(f);
  taskLoop.runLoop();
}

}  // namespace uthread