WifiCond: Setup polling on HW binder FD

[android-x86/system-connectivity-wificond.git] / tests / mock_netlink_utils.h

diff --git a/tests/mock_netlink_utils.h b/tests/mock_netlink_utils.h
index a8fd63c..1a0642e 100644 (file)
--- a/tests/mock_netlink_utils.h
+++ b/tests/mock_netlink_utils.h
@@ -32,6 +32,7 @@ class MockNetlinkUtils : public NetlinkUtils {
   MOCK_METHOD1(GetWiphyIndex, bool(uint32_t* out_wiphy_index));
   MOCK_METHOD1(UnsubscribeMlmeEvent, void(uint32_t interface_index));
   MOCK_METHOD1(UnsubscribeRegDomainChange, void(uint32_t wiphy_index));
+  MOCK_METHOD1(UnsubscribeStationEvent, void(uint32_t interface_index));
   MOCK_METHOD2(SetInterfaceMode,
                bool(uint32_t interface_index, InterfaceMode mode));
   MOCK_METHOD2(SubscribeMlmeEvent,
@@ -40,6 +41,9 @@ class MockNetlinkUtils : public NetlinkUtils {
   MOCK_METHOD2(SubscribeRegDomainChange,
                void(uint32_t wiphy_index,
                     OnRegDomainChangedHandler handler));
+  MOCK_METHOD2(SubscribeStationEvent,
+               void(uint32_t interface_index,
+                    OnStationEventHandler handler));
 
   MOCK_METHOD2(GetInterfaces,
                bool(uint32_t wiphy_index,