1 files changed, 137 insertions, 0 deletions
diff --git a/bluetooth/sar/aidl/vts/VtsHalBluetoothSarTargetTest.cpp b/bluetooth/sar/aidl/vts/VtsHalBluetoothSarTargetTest.cpp
new file mode 100644
index 0000000..817c953
--- /dev/null
+++ b/bluetooth/sar/aidl/vts/VtsHalBluetoothSarTargetTest.cpp
@@ -0,0 +1,137 @@
+/*
+ * Copyright (C) 2023 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.
+ */
+
+#include <aidl/Gtest.h>
+#include <aidl/Vintf.h>
+#include <aidl/hardware/google/bluetooth/sar/IBluetoothSar.h>
+#include <android/binder_auto_utils.h>
+#include <android/binder_manager.h>
+#include <android/binder_process.h>
+#include <binder/IServiceManager.h>
+#include <binder/ProcessState.h>
+
+using aidl::hardware::google::bluetooth::sar::IBluetoothSar;
+using ndk::ScopedAStatus;
+using ndk::SpAIBinder;
+
+class BluetoothSarTest : public ::testing::TestWithParam<std::string> {
+public:
+  virtual void SetUp() override;
+  virtual void TearDown() override;
+
+  ndk::ScopedAStatus setAndCheckBluetoothTxPowerCap(int8_t cap);
+  ndk::ScopedAStatus setAndCheckBluetoothTechBasedTxPowerCap(int8_t br_cap, int8_t edr_cap, int8_t ble_cap);
+  ndk::ScopedAStatus setAndCheckBluetoothModeBasedTxPowerCap(const std::array<uint8_t, 3>& chain_0_cap, const std::array<uint8_t, 3>& chain_1_cap, const std::array<uint8_t, 6>& beamforming_cap);
+  ndk::ScopedAStatus setAndCheckBluetoothModeBasedTxPowerCapPlusHR(const std::array<uint8_t, 4>& chain_0_cap, const std::array<uint8_t, 4>& chain_1_cap, const std::array<uint8_t, 8>& beamforming_cap);
+  ndk::ScopedAStatus setAndCheckBluetoothAreaCode(const std::array<uint8_t, 3>& area_code);
+
+private:
+  std::shared_ptr<IBluetoothSar> bluetooth_sar_;
+};
+
+void BluetoothSarTest::SetUp() {
+  ALOGI("SetUp Bluetooth SAR Test");
+  bluetooth_sar_ = IBluetoothSar::fromBinder(
+      ndk::SpAIBinder(AServiceManager_waitForService(GetParam().c_str())));
+  ASSERT_NE(bluetooth_sar_, nullptr);
+}
+
+void BluetoothSarTest::TearDown() {
+  ALOGI("TearDown Bluetooth SAR Test");
+  bluetooth_sar_ = nullptr;
+  ASSERT_EQ(bluetooth_sar_, nullptr);
+}
+
+ndk::ScopedAStatus BluetoothSarTest::setAndCheckBluetoothTxPowerCap(int8_t cap) {
+  return bluetooth_sar_->setBluetoothTxPowerCap(cap);
+}
+
+ndk::ScopedAStatus BluetoothSarTest::setAndCheckBluetoothTechBasedTxPowerCap(
+    int8_t br_cap, int8_t edr_cap, int8_t ble_cap) {
+  return bluetooth_sar_->setBluetoothTechBasedTxPowerCap(br_cap, edr_cap, ble_cap);
+}
+
+ndk::ScopedAStatus BluetoothSarTest::setAndCheckBluetoothModeBasedTxPowerCap(
+    const std::array<uint8_t, 3>& chain_0_cap,
+    const std::array<uint8_t, 3>& chain_1_cap,
+    const std::array<uint8_t, 6>& beamforming_cap) {
+  return bluetooth_sar_->setBluetoothModeBasedTxPowerCap(chain_0_cap, chain_1_cap, beamforming_cap);
+}
+
+
+ndk::ScopedAStatus BluetoothSarTest::setAndCheckBluetoothModeBasedTxPowerCapPlusHR(
+    const std::array<uint8_t, 4>& chain_0_cap,
+    const std::array<uint8_t, 4>& chain_1_cap,
+    const std::array<uint8_t, 8>& beamforming_cap){
+  return bluetooth_sar_->setBluetoothModeBasedTxPowerCapPlusHR(chain_0_cap, chain_1_cap, beamforming_cap);
+}
+
+TEST_P(BluetoothSarTest, setAndCheckBluetoothTxPowerCap) {
+  ndk::ScopedAStatus status = setAndCheckBluetoothTxPowerCap(40);
+  ASSERT_TRUE(status.isOk());
+  // check invalid power cap (greater than 80)
+  status = setAndCheckBluetoothTxPowerCap(100);
+  ASSERT_FALSE(status.isOk());
+}
+
+
+TEST_P(BluetoothSarTest, setAndCheckBluetoothTechBasedTxPowerCap) {
+  ndk::ScopedAStatus status = setAndCheckBluetoothTechBasedTxPowerCap(10, 20, 30);
+  ASSERT_TRUE(status.isOk());
+  // check invalid power cap (greater than 80)
+  status = setAndCheckBluetoothTechBasedTxPowerCap(10, 120, 30);
+  ASSERT_FALSE(status.isOk());
+}
+
+TEST_P(BluetoothSarTest, setAndCheckBluetoothModeBasedTxPowerCap) {
+  std::array<uint8_t, 3> chain_0_cap = {10, 20, 30};
+  std::array<uint8_t, 3> chain_1_cap = {15, 25, 35};
+  std::array<uint8_t, 6> beamforming_cap = {10, 20, 30, 40, 50, 60};
+  ndk::ScopedAStatus status =
+      setAndCheckBluetoothModeBasedTxPowerCap(chain_0_cap, chain_1_cap, beamforming_cap);
+  ASSERT_TRUE(status.isOk());
+  // check invalid power cap (greater than 80)
+  std::array<uint8_t, 3> bad_cap = {15, 125, 35};
+  status = setAndCheckBluetoothModeBasedTxPowerCap(chain_0_cap, bad_cap, beamforming_cap);
+  ASSERT_FALSE(status.isOk());
+}
+
+TEST_P(BluetoothSarTest, setAndCheckBluetoothModeBasedTxPowerCapPlusHR) {
+  std::array<uint8_t, 4> chain_0_cap = {10, 20, 30, 40};
+  std::array<uint8_t, 4> chain_1_cap = {15, 25, 35, 45};
+  std::array<uint8_t, 8> beamforming_cap = {10, 20, 30, 40, 50, 60, 70, 80};
+  ndk::ScopedAStatus status =
+      setAndCheckBluetoothModeBasedTxPowerCapPlusHR(chain_0_cap, chain_1_cap, beamforming_cap);
+  ASSERT_TRUE(status.isOk());
+  // check invalid power cap (greater than 80)
+  std::array<uint8_t, 4> bad_cap = {10, 20, 30, 200};
+  status = setAndCheckBluetoothModeBasedTxPowerCapPlusHR(bad_cap, chain_1_cap, beamforming_cap);
+  ASSERT_FALSE(status.isOk());
+}
+
+GTEST_ALLOW_UNINSTANTIATED_PARAMETERIZED_TEST(BluetoothSarTest);
+INSTANTIATE_TEST_SUITE_P(PerInstance, BluetoothSarTest,
+                         testing::ValuesIn(android::getAidlHalInstanceNames(
+                             IBluetoothSar::descriptor)),
+                         android::PrintInstanceNameToString);
+
+int main(int argc, char **argv) {
+  ::testing::InitGoogleTest(&argc, argv);
+  ABinderProcess_startThreadPool();
+  int status = RUN_ALL_TESTS();
+  ALOGI("Test result = %d", status);
+  return status;
+}