From c578a913568a5611f396e97b4ee5851930559b97 Mon Sep 17 00:00:00 2001
From: Jimmy Chen <jimmycmchen@google.com>
Date: Thu, 16 Apr 2020 11:48:14 +0800
Subject: [PATCH] wifi: skip tests if RTT is not supported

Bug: 153418156
Test: atest VtsHalWifiApV1_4TargetTest

Change-Id: I2aa58c501f44a9b32c1a7ebf48aa8fcbfb08b046
---
 wifi/1.4/vts/functional/wifi_rtt_controller_hidl_test.cpp | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/wifi/1.4/vts/functional/wifi_rtt_controller_hidl_test.cpp b/wifi/1.4/vts/functional/wifi_rtt_controller_hidl_test.cpp
index 295c86e1a7..4035fb8f71 100644
--- a/wifi/1.4/vts/functional/wifi_rtt_controller_hidl_test.cpp
+++ b/wifi/1.4/vts/functional/wifi_rtt_controller_hidl_test.cpp
@@ -64,6 +64,14 @@ class WifiRttControllerHidlTest : public ::testing::TestWithParam<std::string> {
 
         wifi_rtt_controller_ = getWifiRttController();
         ASSERT_NE(nullptr, wifi_rtt_controller_.get());
+
+        // Check RTT support before we run the test.
+        std::pair<WifiStatus, RttCapabilities> status_and_caps;
+        status_and_caps =
+            HIDL_INVOKE(wifi_rtt_controller_, getCapabilities_1_4);
+        if (status_and_caps.first.code == WifiStatusCode::ERROR_NOT_SUPPORTED) {
+            GTEST_SKIP() << "Skipping this test since RTT is not supported.";
+        }
     }
 
     virtual void TearDown() override { stopWifi(GetInstanceName()); }