aboutsummaryrefslogtreecommitdiff
path: root/tests/worker_test.cpp
blob: d1eb2b3d4b5c3c724f631b0f8eb491796b7bc8f0 (plain)
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#include "utils/Worker.h"

#include <gtest/gtest.h>
#include <hardware/hardware.h>

#include <chrono>

using android::Worker;

struct TestWorker : public Worker {
  TestWorker() : Worker("test-worker", HAL_PRIORITY_URGENT_DISPLAY){};

  int Init() {
    return InitWorker();
  }

  void Routine() override {
    Lock();
    if (!enabled_) {
      int ret = WaitForSignalOrExitLocked();
      if (ret == -EINTR) {
        Unlock();
        return;
      }
      // should only reached here if it was enabled
      if (!enabled_)
        printf("Shouldn't reach here while disabled %d %d\n", value, ret);
    }
    value++;
    Unlock();
  }

  void Control(bool enable) {
    bool changed = false;
    Lock();
    if (enabled_ != enable) {
      enabled_ = enable;
      changed = true;
    }
    Unlock();

    if (enable && changed)
      Signal();
  }

  // NOLINTNEXTLINE: should not be public
  int value{};

 private:
  bool enabled_{};
};

struct WorkerTest : public testing::Test {
  TestWorker worker;

  void SetUp() override {
    worker.Init();
  }

  void small_delay() {
    std::this_thread::sleep_for(std::chrono::milliseconds(20));
  }
};

// NOLINTNEXTLINE: required by gtest macros
TEST_F(WorkerTest, TestWorker) {
  // already isInitialized so should succeed
  ASSERT_TRUE(worker.initialized());

  int val = worker.value;
  small_delay();

  // value shouldn't change when isInitialized
  ASSERT_EQ(val, worker.value);

  worker.Control(true);
  small_delay();

  // while locked, value shouldn't be changing
  worker.Lock();
  val = worker.value;
  small_delay();
  ASSERT_EQ(val, worker.value);
  worker.Unlock();

  small_delay();
  // value should be different now
  ASSERT_NE(val, worker.value);

  worker.Control(false);
  worker.Lock();
  val = worker.value;
  worker.Unlock();
  small_delay();

  // value should be same
  ASSERT_EQ(val, worker.value);

  worker.Exit();
  ASSERT_FALSE(worker.initialized());
}

// NOLINTNEXTLINE: required by gtest macros
TEST_F(WorkerTest, ExitWhileRunning) {
  worker.Control(true);

  std::this_thread::sleep_for(std::chrono::milliseconds(50));
  worker.Exit();
}