160 lines
5.6 KiB
C++
160 lines
5.6 KiB
C++
/*
|
|
* Copyright (C) 2016, 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 <array>
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#include <linux/if_ether.h>
|
|
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
#include <wifi_system_test/mock_interface_tool.h>
|
|
|
|
#include "wificond/logging_utils.h"
|
|
#include "wificond/tests/mock_ap_interface_event_callback.h"
|
|
#include "wificond/tests/mock_netlink_manager.h"
|
|
#include "wificond/tests/mock_netlink_utils.h"
|
|
|
|
#include "wificond/ap_interface_impl.h"
|
|
|
|
using android::wifi_system::MockInterfaceTool;
|
|
using android::net::wifi::nl80211::NativeWifiClient;
|
|
using std::array;
|
|
using std::placeholders::_1;
|
|
using std::placeholders::_2;
|
|
using std::unique_ptr;
|
|
using std::vector;
|
|
using testing::NiceMock;
|
|
using testing::Invoke;
|
|
using testing::Return;
|
|
using testing::Sequence;
|
|
using testing::StrEq;
|
|
using testing::_;
|
|
|
|
namespace android {
|
|
namespace wificond {
|
|
namespace {
|
|
|
|
const char kTestInterfaceName[] = "testwifi0";
|
|
const uint32_t kTestInterfaceIndex = 42;
|
|
const array<uint8_t, ETH_ALEN> kFakeMacAddress01 = {0x45, 0x54, 0xad, 0x67, 0x98, 0xf6};
|
|
const array<uint8_t, ETH_ALEN> kFakeMacAddress02 = {0x45, 0x54, 0xad, 0x67, 0x98, 0xf7};
|
|
|
|
void CaptureStationEventHandler(
|
|
OnStationEventHandler* out_handler,
|
|
uint32_t interface_index,
|
|
OnStationEventHandler handler) {
|
|
*out_handler = handler;
|
|
}
|
|
|
|
void CaptureChannelSwitchEventHandler(
|
|
OnChannelSwitchEventHandler* out_handler,
|
|
uint32_t interface_index,
|
|
OnChannelSwitchEventHandler handler) {
|
|
*out_handler = handler;
|
|
}
|
|
|
|
class ApInterfaceImplTest : public ::testing::Test {
|
|
protected:
|
|
unique_ptr<NiceMock<MockInterfaceTool>> if_tool_{
|
|
new NiceMock<MockInterfaceTool>};
|
|
unique_ptr<NiceMock<MockNetlinkManager>> netlink_manager_{
|
|
new NiceMock<MockNetlinkManager>()};
|
|
unique_ptr<NiceMock<MockNetlinkUtils>> netlink_utils_{
|
|
new NiceMock<MockNetlinkUtils>(netlink_manager_.get())};
|
|
|
|
unique_ptr<ApInterfaceImpl> ap_interface_;
|
|
|
|
void SetUp() override {
|
|
ap_interface_.reset(new ApInterfaceImpl(
|
|
kTestInterfaceName,
|
|
kTestInterfaceIndex,
|
|
netlink_utils_.get(),
|
|
if_tool_.get()));
|
|
}
|
|
}; // class ApInterfaceImplTest
|
|
|
|
} // namespace
|
|
|
|
TEST_F(ApInterfaceImplTest, CallbackIsCalledOnConnectedClientsChanged) {
|
|
OnStationEventHandler handler;
|
|
EXPECT_CALL(*netlink_utils_, SubscribeStationEvent(kTestInterfaceIndex, _))
|
|
.WillOnce(Invoke(bind(CaptureStationEventHandler, &handler, _1, _2)));
|
|
ap_interface_.reset(new ApInterfaceImpl(
|
|
kTestInterfaceName, kTestInterfaceIndex, netlink_utils_.get(),
|
|
if_tool_.get()));
|
|
|
|
auto binder = ap_interface_->GetBinder();
|
|
sp<MockApInterfaceEventCallback> callback(new MockApInterfaceEventCallback());
|
|
bool out_success = false;
|
|
EXPECT_TRUE(binder->registerCallback(callback, &out_success).isOk());
|
|
EXPECT_TRUE(out_success);
|
|
|
|
array<uint8_t, ETH_ALEN> fake_mac_address_01 = kFakeMacAddress01;
|
|
array<uint8_t, ETH_ALEN> fake_mac_address_02 = kFakeMacAddress02;
|
|
EXPECT_CALL(*callback, onConnectedClientsChanged(_,_)).Times(3);
|
|
handler(NEW_STATION, fake_mac_address_01);
|
|
handler(NEW_STATION, fake_mac_address_02);
|
|
handler(DEL_STATION, fake_mac_address_01);
|
|
}
|
|
|
|
TEST_F(ApInterfaceImplTest, CallbackIsCalledOnConnectedClientsChangedOnlyOnDiff) {
|
|
OnStationEventHandler handler;
|
|
EXPECT_CALL(*netlink_utils_, SubscribeStationEvent(kTestInterfaceIndex, _))
|
|
.WillOnce(Invoke(bind(CaptureStationEventHandler, &handler, _1, _2)));
|
|
ap_interface_.reset(new ApInterfaceImpl(
|
|
kTestInterfaceName, kTestInterfaceIndex, netlink_utils_.get(),
|
|
if_tool_.get()));
|
|
|
|
auto binder = ap_interface_->GetBinder();
|
|
sp<MockApInterfaceEventCallback> callback(new MockApInterfaceEventCallback());
|
|
bool out_success = false;
|
|
EXPECT_TRUE(binder->registerCallback(callback, &out_success).isOk());
|
|
EXPECT_TRUE(out_success);
|
|
|
|
array<uint8_t, ETH_ALEN> fake_mac_address_01 = kFakeMacAddress01;
|
|
EXPECT_CALL(*callback, onConnectedClientsChanged(_,_)).Times(4);
|
|
handler(NEW_STATION, fake_mac_address_01);
|
|
handler(NEW_STATION, fake_mac_address_01);
|
|
handler(DEL_STATION, fake_mac_address_01);
|
|
handler(DEL_STATION, fake_mac_address_01);
|
|
}
|
|
|
|
TEST_F(ApInterfaceImplTest, CallbackIsCalledOnSoftApChannelSwitched) {
|
|
OnChannelSwitchEventHandler handler;
|
|
EXPECT_CALL(*netlink_utils_, SubscribeChannelSwitchEvent(kTestInterfaceIndex, _))
|
|
.WillOnce(Invoke(bind(CaptureChannelSwitchEventHandler, &handler, _1, _2)));
|
|
ap_interface_.reset(new ApInterfaceImpl(
|
|
kTestInterfaceName, kTestInterfaceIndex, netlink_utils_.get(),
|
|
if_tool_.get()));
|
|
|
|
auto binder = ap_interface_->GetBinder();
|
|
sp<MockApInterfaceEventCallback> callback(new MockApInterfaceEventCallback());
|
|
bool out_success = false;
|
|
EXPECT_TRUE(binder->registerCallback(callback, &out_success).isOk());
|
|
EXPECT_TRUE(out_success);
|
|
|
|
const uint32_t kTestChannelFrequency = 2437;
|
|
const ChannelBandwidth kTestChannelBandwidth = ChannelBandwidth::BW_20;
|
|
EXPECT_CALL(*callback, onSoftApChannelSwitched(kTestChannelFrequency,
|
|
kTestChannelBandwidth));
|
|
handler(kTestChannelFrequency, kTestChannelBandwidth);
|
|
}
|
|
|
|
} // namespace wificond
|
|
} // namespace android
|