diff --git a/tetheroffload/config/1.0/vts/functional/VtsHalTetheroffloadConfigV1_0TargetTest.cpp b/tetheroffload/config/1.0/vts/functional/VtsHalTetheroffloadConfigV1_0TargetTest.cpp index 2cbe479dc9..0ee9eb4eea 100644 --- a/tetheroffload/config/1.0/vts/functional/VtsHalTetheroffloadConfigV1_0TargetTest.cpp +++ b/tetheroffload/config/1.0/vts/functional/VtsHalTetheroffloadConfigV1_0TargetTest.cpp @@ -17,6 +17,7 @@ #define LOG_TAG "VtsOffloadConfigV1_0TargetTest" #include +#include #include #include #include @@ -77,10 +78,25 @@ int netlinkSocket(unsigned groups) { return netlinkSocket(NETLINK_NETFILTER, groups); } +// Test environment for OffloadConfig HIDL HAL. +class OffloadConfigHidlEnvironment : public ::testing::VtsHalHidlTargetTestEnvBase { + public: + // get the test environment singleton + static OffloadConfigHidlEnvironment* Instance() { + static OffloadConfigHidlEnvironment* instance = new OffloadConfigHidlEnvironment; + return instance; + } + + virtual void registerTestServices() override { registerTestService(); } + private: + OffloadConfigHidlEnvironment() {} +}; + class OffloadConfigHidlTest : public testing::VtsHalHidlTargetTestBase { public: virtual void SetUp() override { - config = testing::VtsHalHidlTargetTestBase::getService(); + config = testing::VtsHalHidlTargetTestBase::getService( + OffloadConfigHidlEnvironment::Instance()->getServiceName()); ASSERT_NE(nullptr, config.get()) << "Could not get HIDL instance"; } @@ -175,7 +191,9 @@ TEST_F(OffloadConfigHidlTest, TestSetHandle2OnlyNotOk) { } int main(int argc, char** argv) { - testing::InitGoogleTest(&argc, argv); + ::testing::AddGlobalTestEnvironment(OffloadConfigHidlEnvironment::Instance()); + ::testing::InitGoogleTest(&argc, argv); + OffloadConfigHidlEnvironment::Instance()->init(&argc, argv); int status = RUN_ALL_TESTS(); ALOGE("Test result with status=%d", status); return status; diff --git a/tetheroffload/control/1.0/vts/functional/VtsHalTetheroffloadControlV1_0TargetTest.cpp b/tetheroffload/control/1.0/vts/functional/VtsHalTetheroffloadControlV1_0TargetTest.cpp index c009af5d91..c5e8021801 100644 --- a/tetheroffload/control/1.0/vts/functional/VtsHalTetheroffloadControlV1_0TargetTest.cpp +++ b/tetheroffload/control/1.0/vts/functional/VtsHalTetheroffloadControlV1_0TargetTest.cpp @@ -18,6 +18,7 @@ #include #include +#include #include #include #include @@ -111,6 +112,23 @@ class TetheringOffloadCallbackArgs { NatTimeoutUpdate last_params; }; +// Test environment for OffloadControl HIDL HAL. +class OffloadControlHidlEnvironment : public ::testing::VtsHalHidlTargetTestEnvBase { + public: + // get the test environment singleton + static OffloadControlHidlEnvironment* Instance() { + static OffloadControlHidlEnvironment* instance = new OffloadControlHidlEnvironment; + return instance; + } + + virtual void registerTestServices() override { + registerTestService(); + registerTestService(); + } + private: + OffloadControlHidlEnvironment() {} +}; + class OffloadControlHidlTestBase : public testing::VtsHalHidlTargetTestBase { public: virtual void SetUp() override { @@ -128,7 +146,8 @@ class OffloadControlHidlTestBase : public testing::VtsHalHidlTargetTestBase { // The IOffloadConfig HAL is tested more thoroughly elsewhere. He we just // setup everything correctly and verify basic readiness. void setupConfigHal() { - config = testing::VtsHalHidlTargetTestBase::getService(); + config = testing::VtsHalHidlTargetTestBase::getService( + OffloadControlHidlEnvironment::Instance()->getServiceName()); ASSERT_NE(nullptr, config.get()) << "Could not get HIDL instance"; unique_fd fd1(conntrackSocket(NFNLGRP_CONNTRACK_NEW | NFNLGRP_CONNTRACK_DESTROY)); @@ -156,7 +175,8 @@ class OffloadControlHidlTestBase : public testing::VtsHalHidlTargetTestBase { } void prepareControlHal() { - control = testing::VtsHalHidlTargetTestBase::getService(); + control = testing::VtsHalHidlTargetTestBase::getService( + OffloadControlHidlEnvironment::Instance()->getServiceName()); ASSERT_NE(nullptr, control.get()) << "Could not get HIDL instance"; control_cb = new TetheringOffloadCallback(); @@ -660,7 +680,9 @@ TEST_F(OffloadControlHidlTest, RemoveDownstreamBogusPrefixFails) { } int main(int argc, char** argv) { - testing::InitGoogleTest(&argc, argv); + ::testing::AddGlobalTestEnvironment(OffloadControlHidlEnvironment::Instance()); + ::testing::InitGoogleTest(&argc, argv); + OffloadControlHidlEnvironment::Instance()->init(&argc, argv); int status = RUN_ALL_TESTS(); ALOGE("Test result with status=%d", status); return status;