summaryrefslogtreecommitdiff
path: root/system/osi/test/alarm_mock.h
blob: b5b3b2a9f850de93f570beffda11e1d34e410ede (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
#pragma once
#include <memory>

struct alarm_t;

class AlarmMock {
public:
  MOCK_METHOD1(AlarmNew, alarm_t*(const char*));
  MOCK_METHOD1(AlarmFree, void(alarm_t*));
  MOCK_METHOD1(AlarmCancel, void(alarm_t*));
  MOCK_METHOD4(AlarmSet,
               void(alarm_t* alarm, uint64_t interval_ms, alarm_callback_t cb, void* data));
  MOCK_METHOD4(AlarmSetOnMloop,
               void(alarm_t* alarm, uint64_t interval_ms, alarm_callback_t cb, void* data));
  MOCK_METHOD1(AlarmIsScheduled, bool(const alarm_t*));

  static inline AlarmMock* Get() {
    if (!localAlarmMock) {
      localAlarmMock = std::make_unique<AlarmMock>();
    }
    return localAlarmMock.get();
  }

  static inline void Reset() { localAlarmMock = std::make_unique<AlarmMock>(); }

private:
  static std::unique_ptr<AlarmMock> localAlarmMock;
};

std::unique_ptr<AlarmMock> AlarmMock::localAlarmMock;

alarm_t* alarm_new(const char* name) { return AlarmMock::Get()->AlarmNew(name); }

void alarm_free(alarm_t* alarm) { AlarmMock::Get()->AlarmFree(alarm); }

void alarm_set_on_mloop(alarm_t* alarm, uint64_t interval_ms, alarm_callback_t cb, void* data) {
  AlarmMock::Get()->AlarmSetOnMloop(alarm, interval_ms, cb, data);
}

void alarm_set(alarm_t* alarm, uint64_t interval_ms, alarm_callback_t cb, void* data) {
  AlarmMock::Get()->AlarmSet(alarm, interval_ms, cb, data);
}

bool alarm_is_scheduled(const alarm_t* alarm) { return AlarmMock::Get()->AlarmIsScheduled(alarm); }

void alarm_cancel(alarm_t* alarm) { AlarmMock::Get()->AlarmCancel(alarm); }