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 MockApi : public ::aidl::android::hardware::vibrator::Vibrator::HwApi { 24 public: 25 MOCK_METHOD0(destructor, void()); 26 MOCK_METHOD1(setF0, bool(std::string value)); 27 MOCK_METHOD1(setF0Offset, bool(uint32_t value)); 28 MOCK_METHOD1(setRedc, bool(std::string value)); 29 MOCK_METHOD1(setQ, bool(std::string value)); 30 MOCK_METHOD1(getEffectCount, bool(uint32_t *value)); 31 MOCK_METHOD2(pollVibeState, bool(uint32_t value, int32_t timeoutMs)); 32 MOCK_METHOD0(hasOwtFreeSpace, bool()); 33 MOCK_METHOD1(getOwtFreeSpace, bool(uint32_t *value)); 34 MOCK_METHOD1(setF0CompEnable, bool(bool value)); 35 MOCK_METHOD1(setRedcCompEnable, bool(bool value)); 36 MOCK_METHOD1(setMinOnOffInterval, bool(uint32_t value)); 37 MOCK_METHOD0(getContextScale, uint32_t()); 38 MOCK_METHOD0(getContextEnable, bool()); 39 MOCK_METHOD0(getContextSettlingTime, uint32_t()); 40 MOCK_METHOD0(getContextCooldownTime, uint32_t()); 41 MOCK_METHOD0(getContextFadeEnable, bool()); 42 MOCK_METHOD2(setFFGain, bool(int fd, uint16_t value)); 43 MOCK_METHOD3(setFFEffect, bool(int fd, struct ff_effect *effect, uint16_t timeoutMs)); 44 MOCK_METHOD3(setFFPlay, bool(int fd, int8_t index, bool value)); 45 MOCK_METHOD2(getHapticAlsaDevice, bool(int *card, int *device)); 46 MOCK_METHOD4(setHapticPcmAmp, bool(struct pcm **haptic_pcm, bool enable, int card, int device)); 47 MOCK_METHOD6(uploadOwtEffect, 48 bool(int fd, uint8_t *owtData, uint32_t numBytes, struct ff_effect *effect, 49 uint32_t *outEffectIndex, int *status)); 50 MOCK_METHOD3(eraseOwtEffect, bool(int fd, int8_t effectIndex, std::vector<ff_effect> *effect)); 51 MOCK_METHOD1(debug, void(int fd)); 52 ~MockApi()53 ~MockApi() override { destructor(); }; 54 }; 55 56 class MockCal : public ::aidl::android::hardware::vibrator::Vibrator::HwCal { 57 public: 58 MOCK_METHOD0(destructor, void()); 59 MOCK_METHOD1(getVersion, bool(uint32_t *value)); 60 MOCK_METHOD1(getF0, bool(std::string &value)); 61 MOCK_METHOD1(getRedc, bool(std::string &value)); 62 MOCK_METHOD1(getQ, bool(std::string &value)); 63 MOCK_METHOD1(getLongFrequencyShift, bool(int32_t *value)); 64 MOCK_METHOD1(getTickVolLevels, bool(std::array<uint32_t, 2> *value)); 65 MOCK_METHOD1(getClickVolLevels, bool(std::array<uint32_t, 2> *value)); 66 MOCK_METHOD1(getLongVolLevels, bool(std::array<uint32_t, 2> *value)); 67 MOCK_METHOD0(isChirpEnabled, bool()); 68 MOCK_METHOD1(getSupportedPrimitives, bool(uint32_t *value)); 69 MOCK_METHOD0(isF0CompEnabled, bool()); 70 MOCK_METHOD0(isRedcCompEnabled, bool()); 71 MOCK_METHOD1(debug, void(int fd)); 72 ~MockCal()73 ~MockCal() override { destructor(); }; 74 // b/132668253: Workaround gMock Compilation Issue getF0(std::string * value)75 bool getF0(std::string *value) { return getF0(*value); } getRedc(std::string * value)76 bool getRedc(std::string *value) { return getRedc(*value); } getQ(std::string * value)77 bool getQ(std::string *value) { return getQ(*value); } 78 }; 79 80 class MockVibratorCallback : public aidl::android::hardware::vibrator::BnVibratorCallback { 81 public: 82 MOCK_METHOD(ndk::ScopedAStatus, onComplete, ()); 83 }; 84 85 #endif // ANDROID_HARDWARE_VIBRATOR_TEST_MOCKS_H 86