diff --git a/biometrics/fingerprint/2.1/vts/functional/VtsHalBiometricsFingerprintV2_1TargetTest.cpp b/biometrics/fingerprint/2.1/vts/functional/VtsHalBiometricsFingerprintV2_1TargetTest.cpp index 638af9f518..a7f40319af 100644 --- a/biometrics/fingerprint/2.1/vts/functional/VtsHalBiometricsFingerprintV2_1TargetTest.cpp +++ b/biometrics/fingerprint/2.1/vts/functional/VtsHalBiometricsFingerprintV2_1TargetTest.cpp @@ -16,12 +16,13 @@ #define LOG_TAG "fingerprint_hidl_hal_test" +#include +#include #include #include #include #include #include -#include #include #include @@ -179,10 +180,23 @@ class RemoveCallback : public FingerprintCallbackBase { std::promise promise; }; +// Test environment for Fingerprint HIDL HAL. +class FingerprintHidlEnvironment : public ::testing::VtsHalHidlTargetTestEnvBase { + public: + // get the test environment singleton + static FingerprintHidlEnvironment* Instance() { + static FingerprintHidlEnvironment* instance = new FingerprintHidlEnvironment; + return instance; + } + + virtual void registerTestServices() override { registerTestService(); } +}; + class FingerprintHidlTest : public ::testing::VtsHalHidlTargetTestBase { public: virtual void SetUp() override { - mService = ::testing::VtsHalHidlTargetTestBase::getService(); + mService = ::testing::VtsHalHidlTargetTestBase::getService( + FingerprintHidlEnvironment::Instance()->getServiceName()); ASSERT_FALSE(mService == nullptr); // Create an active group @@ -454,7 +468,9 @@ TEST_F(FingerprintHidlTest, CancelRemoveAllTest) { } // anonymous namespace int main(int argc, char **argv) { + ::testing::AddGlobalTestEnvironment(FingerprintHidlEnvironment::Instance()); ::testing::InitGoogleTest(&argc, argv); + FingerprintHidlEnvironment::Instance()->init(&argc, argv); int status = RUN_ALL_TESTS(); LOG(INFO) << "Test result = " << status; return status;