More unit tests for ScannerImpl

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

diff --git a/tests/mock_scan_utils.h b/tests/mock_scan_utils.h
index 4f9e461..fbd1acd 100644 (file)
--- a/tests/mock_scan_utils.h
+++ b/tests/mock_scan_utils.h
@@ -29,15 +29,18 @@ class MockScanUtils : public ScanUtils {
   MockScanUtils(NetlinkManager* netlink_manager);
   ~MockScanUtils() override = default;
 
+  MOCK_METHOD1(AbortScan, bool(uint32_t interface_index));
+
   MOCK_METHOD2(GetScanResult, bool(
       uint32_t interface_index,
       std::vector<::com::android::server::wifi::wificond::NativeScanResult>* out_scan_results));
 
-  MOCK_METHOD4(Scan, bool(
+  MOCK_METHOD5(Scan, bool(
       uint32_t interface_index,
       bool random_mac,
       const std::vector<std::vector<uint8_t>>& ssids,
-      const std::vector<uint32_t>& freqs));
+      const std::vector<uint32_t>& freqs,
+      int* error_code));
 
   MOCK_METHOD2(SubscribeScanResultNotification,void(
       uint32_t interface_index,