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 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122
|
/*
* Copyright (C) 2020 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef VIBRATORSERVICE_UNITTEST_UTIL_H_
#define VIBRATORSERVICE_UNITTEST_UTIL_H_
#include <android/hardware/vibrator/IVibrator.h>
#include <vibratorservice/VibratorHalWrapper.h>
namespace android {
namespace vibrator {
using ::android::hardware::vibrator::ActivePwle;
using ::android::hardware::vibrator::Braking;
using ::android::hardware::vibrator::BrakingPwle;
using ::android::hardware::vibrator::CompositeEffect;
using ::android::hardware::vibrator::CompositePrimitive;
using ::android::hardware::vibrator::PrimitivePwle;
// -------------------------------------------------------------------------------------------------
class MockCallbackScheduler : public vibrator::CallbackScheduler {
public:
MOCK_METHOD(void, schedule, (std::function<void()> callback, std::chrono::milliseconds delay),
(override));
};
ACTION(TriggerSchedulerCallback) {
arg0();
}
// -------------------------------------------------------------------------------------------------
class TestFactory {
public:
static CompositeEffect createCompositeEffect(CompositePrimitive primitive,
std::chrono::milliseconds delay, float scale) {
CompositeEffect effect;
effect.primitive = primitive;
effect.delayMs = delay.count();
effect.scale = scale;
return effect;
}
static PrimitivePwle createActivePwle(float startAmplitude, float startFrequency,
float endAmplitude, float endFrequency,
std::chrono::milliseconds duration) {
ActivePwle pwle;
pwle.startAmplitude = startAmplitude;
pwle.endAmplitude = endAmplitude;
pwle.startFrequency = startFrequency;
pwle.endFrequency = endFrequency;
pwle.duration = duration.count();
return pwle;
}
static PrimitivePwle createBrakingPwle(Braking braking, std::chrono::milliseconds duration) {
BrakingPwle pwle;
pwle.braking = braking;
pwle.duration = duration.count();
return pwle;
}
static std::function<void()> createCountingCallback(int32_t* counter) {
return [counter]() { *counter += 1; };
}
private:
TestFactory() = delete;
~TestFactory() = delete;
};
class TestCounter {
public:
TestCounter(int32_t init = 0) : mMutex(), mCondVar(), mCount(init) {}
int32_t get() {
std::lock_guard<std::mutex> lock(mMutex);
return mCount;
}
void increment() {
{
std::lock_guard<std::mutex> lock(mMutex);
mCount += 1;
}
mCondVar.notify_all();
}
void tryWaitUntilCountIsAtLeast(int32_t count, std::chrono::milliseconds timeout) {
std::unique_lock<std::mutex> lock(mMutex);
mCondVar.wait_for(lock, timeout, [&] { return mCount >= count; });
}
private:
std::mutex mMutex;
std::condition_variable mCondVar;
int32_t mCount GUARDED_BY(mMutex);
};
// -------------------------------------------------------------------------------------------------
} // namespace vibrator
} // namespace android
#endif // VIBRATORSERVICE_UNITTEST_UTIL_H_
|