1 #pragma once
2 #include <memory>
3
4 struct alarm_t;
5
6 class AlarmMock {
7 public:
8 MOCK_METHOD1(AlarmNew, alarm_t*(const char*));
9 MOCK_METHOD1(AlarmFree, void(alarm_t*));
10 MOCK_METHOD1(AlarmCancel, void(alarm_t*));
11 MOCK_METHOD4(AlarmSet,
12 void(alarm_t* alarm, uint64_t interval_ms, alarm_callback_t cb, void* data));
13 MOCK_METHOD4(AlarmSetOnMloop,
14 void(alarm_t* alarm, uint64_t interval_ms, alarm_callback_t cb, void* data));
15 MOCK_METHOD1(AlarmIsScheduled, bool(const alarm_t*));
16
Get()17 static inline AlarmMock* Get() {
18 if (!localAlarmMock) {
19 localAlarmMock = std::make_unique<AlarmMock>();
20 }
21 return localAlarmMock.get();
22 }
23
Reset()24 static inline void Reset() { localAlarmMock = std::make_unique<AlarmMock>(); }
25
26 private:
27 static std::unique_ptr<AlarmMock> localAlarmMock;
28 };
29
30 std::unique_ptr<AlarmMock> AlarmMock::localAlarmMock;
31
alarm_new(const char * name)32 alarm_t* alarm_new(const char* name) { return AlarmMock::Get()->AlarmNew(name); }
33
alarm_free(alarm_t * alarm)34 void alarm_free(alarm_t* alarm) { AlarmMock::Get()->AlarmFree(alarm); }
35
alarm_set_on_mloop(alarm_t * alarm,uint64_t interval_ms,alarm_callback_t cb,void * data)36 void alarm_set_on_mloop(alarm_t* alarm, uint64_t interval_ms, alarm_callback_t cb, void* data) {
37 AlarmMock::Get()->AlarmSetOnMloop(alarm, interval_ms, cb, data);
38 }
39
alarm_set(alarm_t * alarm,uint64_t interval_ms,alarm_callback_t cb,void * data)40 void alarm_set(alarm_t* alarm, uint64_t interval_ms, alarm_callback_t cb, void* data) {
41 AlarmMock::Get()->AlarmSet(alarm, interval_ms, cb, data);
42 }
43
alarm_is_scheduled(const alarm_t * alarm)44 bool alarm_is_scheduled(const alarm_t* alarm) { return AlarmMock::Get()->AlarmIsScheduled(alarm); }
45
alarm_cancel(alarm_t * alarm)46 void alarm_cancel(alarm_t* alarm) { AlarmMock::Get()->AlarmCancel(alarm); }
47