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 806da7f..1a0642e 100644 (file)
--- a/tests/mock_netlink_utils.h
+++ b/tests/mock_netlink_utils.h
@@ -30,11 +30,29 @@ class MockNetlinkUtils : public NetlinkUtils {
   ~MockNetlinkUtils() override = default;
 
   MOCK_METHOD1(GetWiphyIndex, bool(uint32_t* out_wiphy_index));
-  MOCK_METHOD4(GetInterfaceInfo,
+  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,
+               void(uint32_t interface_index,
+                    MlmeEventHandler* handler));
+  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,
-                    std::string* name,
-                    uint32_t* index,
-                    std::vector<uint8_t>* mac_address));
+                    std::vector<InterfaceInfo>* interfaces));
+  MOCK_METHOD4(GetWiphyInfo,
+               bool(uint32_t wiphy_index,
+                    BandInfo* band_info,
+                    ScanCapabilities* scan_capabilities,
+                    WiphyFeatures* wiphy_features));
 
 };  // class MockNetlinkUtils