Merge "Add VTS tests for the untested IWifi methods." into main

    ],

}

cc_test {

    name: "VtsHalWifiTargetTest",

    defaults: ["wifi_vendor_hal_vts_test_defaults"],

    srcs: ["wifi_aidl_test.cpp"],

}

cc_test {

    name: "VtsHalWifiChipTargetTest",

    defaults: ["wifi_vendor_hal_vts_test_defaults"],

/*

 * Copyright (C) 2025 The Android Open Source Project

 *

 * Licensed under the Staache 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 <vector>

#include <VtsCoreUtil.h>

#include <aidl/Gtest.h>

#include <aidl/Vintf.h>

#include <aidl/android/hardware/wifi/BnWifiEventCallback.h>

#include <android/binder_manager.h>

#include <android/binder_status.h>

#include <binder/IServiceManager.h>

#include <binder/ProcessState.h>

#include "wifi_aidl_test_utils.h"

using aidl::android::hardware::wifi::BnWifiEventCallback;

using aidl::android::hardware::wifi::WifiStatusCode;

class WifiAidlTest : public testing::TestWithParam<std::string> {

  public:

    void SetUp() override {

        instance_name_ = GetParam().c_str();

        stopWifiService(instance_name_);

        wifi_ = getWifi(instance_name_);

        ASSERT_NE(wifi_, nullptr);

    }

    void TearDown() override { stopWifiService(instance_name_); }

  protected:

    std::shared_ptr<IWifi> wifi_;

    const char* instance_name_;

};

class WifiEventCallback : public BnWifiEventCallback {

  public:

    WifiEventCallback() = default;

    ::ndk::ScopedAStatus onFailure(WifiStatusCode /* status */) override {

        return ndk::ScopedAStatus::ok();

    }

    ::ndk::ScopedAStatus onStart() override { return ndk::ScopedAStatus::ok(); }

    ::ndk::ScopedAStatus onStop() override { return ndk::ScopedAStatus::ok(); }

    ::ndk::ScopedAStatus onSubsystemRestart(WifiStatusCode /* status */) override {

        return ndk::ScopedAStatus::ok();

    }

};

/*

 * RegisterEventCallback

 */

TEST_P(WifiAidlTest, RegisterEventCallback) {

    const std::shared_ptr<WifiEventCallback> callback =

            ndk::SharedRefBase::make<WifiEventCallback>();

    ASSERT_NE(callback, nullptr);

    EXPECT_TRUE(wifi_->registerEventCallback(callback).isOk());

}

/*

 * IsStarted

 */

TEST_P(WifiAidlTest, IsStarted) {

    // HAL should not be started by default

    bool isStarted;

    EXPECT_TRUE(wifi_->isStarted(&isStarted).isOk());

    EXPECT_FALSE(isStarted);

    // Start wifi by setting up the chip, and verify isStarted

    auto wifiChip = getWifiChip(instance_name_);

    EXPECT_NE(wifiChip, nullptr);

    EXPECT_TRUE(wifi_->isStarted(&isStarted).isOk());

    EXPECT_TRUE(isStarted);

}

GTEST_ALLOW_UNINSTANTIATED_PARAMETERIZED_TEST(WifiAidlTest);

INSTANTIATE_TEST_SUITE_P(WifiTest, WifiAidlTest,

                         testing::ValuesIn(android::getAidlHalInstanceNames(IWifi::descriptor)),

                         android::PrintInstanceNameToString);

int main(int argc, char** argv) {

    ::testing::InitGoogleTest(&argc, argv);

    android::ProcessState::self()->setThreadPoolMaxThreadCount(1);

    android::ProcessState::self()->startThreadPool();

    return RUN_ALL_TESTS();

}