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(AlarmSetOnMloop, void(alarm_t* alarm, uint64_t interval_ms,
12 alarm_callback_t cb, void* data));
13
AlarmNewImpl(const char * name)14 alarm_t* AlarmNewImpl(const char* name) {
15 AlarmNew(name);
16 // We must return something from alarm_new in tests, if we just return null,
17 // unique_ptr will misbehave. Just reserve few bits they will be freed in
18 // AlarmFreeImpl
19 return (alarm_t*)new uint8_t[30];
20 }
21
AlarmFreeImpl(alarm_t * alarm)22 void AlarmFreeImpl(alarm_t* alarm) {
23 uint8_t* ptr = (uint8_t*)alarm;
24 delete[] ptr;
25 return AlarmFree(alarm);
26 }
27
Get()28 static inline AlarmMock* Get() {
29 if (!localAlarmMock) {
30 localAlarmMock = std::make_unique<AlarmMock>();
31 }
32 return localAlarmMock.get();
33 }
34
Reset()35 static inline void Reset() { localAlarmMock = std::make_unique<AlarmMock>(); }
36
37 private:
38 static std::unique_ptr<AlarmMock> localAlarmMock;
39 };
40
41 std::unique_ptr<AlarmMock> AlarmMock::localAlarmMock;
42
alarm_new(const char * name)43 alarm_t* alarm_new(const char* name) {
44 return AlarmMock::Get()->AlarmNewImpl(name);
45 }
46
alarm_free(alarm_t * alarm)47 void alarm_free(alarm_t* alarm) { AlarmMock::Get()->AlarmFreeImpl(alarm); }
48
alarm_set_on_mloop(alarm_t * alarm,uint64_t interval_ms,alarm_callback_t cb,void * data)49 void alarm_set_on_mloop(alarm_t* alarm, uint64_t interval_ms,
50 alarm_callback_t cb, void* data) {
51 AlarmMock::Get()->AlarmSetOnMloop(alarm, interval_ms, cb, data);
52 }
53