1 /* 2 * Copyright (C) 2022 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 #ifndef ANDROID_HARDWARE_VIBRATOR_TEST_MOCKS_H 17 #define ANDROID_HARDWARE_VIBRATOR_TEST_MOCKS_H 18 19 #include <aidl/android/hardware/vibrator/BnVibratorCallback.h> 20 21 #include "Vibrator.h" 22 23 class MockGPIO : public ::aidl::android::hardware::vibrator::Vibrator::HwGPIO { 24 public: 25 MOCK_METHOD0(destructor, void()); 26 MOCK_METHOD0(getGPIO, bool()); 27 MOCK_METHOD0(initGPIO, bool()); 28 MOCK_METHOD1(setGPIOOutput, bool(bool value)); 29 MOCK_METHOD1(debug, void(int fd)); 30 ~MockGPIO()31 ~MockGPIO() override { destructor(); }; 32 }; 33 34 class MockApi : public ::aidl::android::hardware::vibrator::Vibrator::HwApi { 35 public: 36 MOCK_METHOD0(destructor, void()); 37 MOCK_METHOD1(setF0, bool(std::string value)); 38 MOCK_METHOD1(setF0Offset, bool(uint32_t value)); 39 MOCK_METHOD1(setRedc, bool(std::string value)); 40 MOCK_METHOD1(setQ, bool(std::string value)); 41 MOCK_METHOD1(getEffectCount, bool(uint32_t *value)); 42 MOCK_METHOD0(hasEffectBrakingTimeBank, bool()); 43 MOCK_METHOD1(setEffectBrakingTimeBank, bool(uint32_t value)); 44 MOCK_METHOD1(setEffectBrakingTimeIndex, bool(uint32_t value)); 45 MOCK_METHOD1(getEffectBrakingTimeMs, bool(uint32_t *value)); 46 MOCK_METHOD2(pollVibeState, bool(uint32_t value, int32_t timeoutMs)); 47 MOCK_METHOD0(hasOwtFreeSpace, bool()); 48 MOCK_METHOD1(getOwtFreeSpace, bool(uint32_t *value)); 49 MOCK_METHOD1(setF0CompEnable, bool(bool value)); 50 MOCK_METHOD1(setRedcCompEnable, bool(bool value)); 51 MOCK_METHOD1(setMinOnOffInterval, bool(uint32_t value)); 52 MOCK_METHOD2(setFFGain, bool(int fd, uint16_t value)); 53 MOCK_METHOD3(setFFEffect, bool(int fd, struct ff_effect *effect, uint16_t timeoutMs)); 54 MOCK_METHOD3(setFFPlay, bool(int fd, int8_t index, bool value)); 55 MOCK_METHOD2(getHapticAlsaDevice, bool(int *card, int *device)); 56 MOCK_METHOD4(setHapticPcmAmp, bool(struct pcm **haptic_pcm, bool enable, int card, int device)); 57 MOCK_METHOD6(uploadOwtEffect, 58 bool(int fd, const uint8_t *owtData, const uint32_t numBytes, struct ff_effect *effect, 59 uint32_t *outEffectIndex, int *status)); 60 MOCK_METHOD3(eraseOwtEffect, bool(int fd, int8_t effectIndex, std::vector<ff_effect> *effect)); 61 MOCK_METHOD1(debug, void(int fd)); 62 ~MockApi()63 ~MockApi() override { destructor(); }; 64 }; 65 66 class MockCal : public ::aidl::android::hardware::vibrator::Vibrator::HwCal { 67 public: 68 MOCK_METHOD0(destructor, void()); 69 MOCK_METHOD1(getVersion, bool(uint32_t *value)); 70 MOCK_METHOD1(getF0, bool(std::string &value)); 71 MOCK_METHOD1(getF0SyncOffset, bool(uint32_t *value)); 72 MOCK_METHOD1(getRedc, bool(std::string &value)); 73 MOCK_METHOD1(getQ, bool(std::string &value)); 74 MOCK_METHOD1(getLongFrequencyShift, bool(int32_t *value)); 75 MOCK_METHOD1(getTickVolLevels, bool(std::array<uint32_t, 2> *value)); 76 MOCK_METHOD1(getClickVolLevels, bool(std::array<uint32_t, 2> *value)); 77 MOCK_METHOD1(getLongVolLevels, bool(std::array<uint32_t, 2> *value)); 78 MOCK_METHOD0(isChirpEnabled, bool()); 79 MOCK_METHOD1(getSupportedPrimitives, bool(uint32_t *value)); 80 MOCK_METHOD0(isF0CompEnabled, bool()); 81 MOCK_METHOD0(isRedcCompEnabled, bool()); 82 MOCK_METHOD1(debug, void(int fd)); 83 ~MockCal()84 ~MockCal() override { destructor(); }; 85 // b/132668253: Workaround gMock Compilation Issue getF0(std::string * value)86 bool getF0(std::string *value) { return getF0(*value); } getRedc(std::string * value)87 bool getRedc(std::string *value) { return getRedc(*value); } getQ(std::string * value)88 bool getQ(std::string *value) { return getQ(*value); } 89 }; 90 91 class MockVibratorCallback : public aidl::android::hardware::vibrator::BnVibratorCallback { 92 public: 93 MOCK_METHOD(ndk::ScopedAStatus, onComplete, ()); 94 }; 95 96 #endif // ANDROID_HARDWARE_VIBRATOR_TEST_MOCKS_H 97