Add DiscoveryNetworkMonitor implementation

chrome/browser/extensions/api/dial/dial_apitest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/command_line.h"
 #include "build/build_config.h"
 #include "chrome/browser/extensions/api/dial/dial_api.h"
 #include "chrome/browser/extensions/api/dial/dial_api_factory.h"
 #include "chrome/browser/extensions/api/dial/dial_registry.h"
 #include "chrome/browser/extensions/extension_apitest.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "extensions/common/switches.h"
 #include "extensions/test/extension_test_message_listener.h"
 #include "extensions/test/result_catcher.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "url/gurl.h"
 
 using extensions::Extension;
 using extensions::ResultCatcher;
 using extensions::api::dial::DialDeviceData;
 using extensions::api::dial::DialDeviceDescriptionData;
 using extensions::api::dial::DialRegistry;
 
 namespace {
 
+std::vector<NetworkInfo> FakeGetNetworkInfo(
+    const std::vector<NetworkInfo>& network_info_list) {
+  return network_info_list;
+}
+
 class DialAPITest : public ExtensionApiTest {
  public:
   DialAPITest() {}
 
   void SetUpCommandLine(base::CommandLine* command_line) override {
     ExtensionApiTest::SetUpCommandLine(command_line);
     command_line->AppendSwitchASCII(
         extensions::switches::kWhitelistedExtensionID,
         "ddchlicdkolnonkihahngkmmmjnjlkkf");
   }
+
+ protected:
+  void SetUp() override {
+    net::NetworkChangeNotifier::SetTestNotificationsOnly(true);
+
+    // This will create a NetworkChangeNotifier via BrowserTestBase, so make
+    // sure this is after SetTestNotificationsOnly but before we try to create a
+    // DiscoveryNetworkMonitor.
+    ExtensionApiTest::SetUp();
+  }
+
+  void SetUpOnMainThread() override {
+    // This executes after SetUp() so NetworkChangeNotifier should exist here.
+    auto* discovery_network_monitor =
+        DiscoveryNetworkMonitor::GetInstanceForTest(
+            base::Bind(&FakeGetNetworkInfo, base::ConstRef(fake_network_info)));
+    // See SetUp() in discovery_network_monitor_unittest.cc for an explanation
+    // of this logic.
+    if (should_rebind_to_network_change_monitor) {
+      discovery_network_monitor->RebindNetworkChangeObserverForTest();
+    } else {
+      should_rebind_to_network_change_monitor = true;
+    }
+    ExtensionApiTest::SetUpOnMainThread();
+  }
+
+  bool should_rebind_to_network_change_monitor = false;
+  std::vector<NetworkInfo> fake_network_info;
 };
 
 }  // namespace
 
 // http://crbug.com/177163
 #if defined(OS_WIN) && !defined(NDEBUG)
 #define MAYBE_DeviceListEvents DISABLED_DeviceListEvents
 #else
 #define MAYBE_DeviceListEvents DeviceListEvents
 #endif
@@ skipping 82 matching lines @@
                              base::Time::Now());
   test_device.set_label("testDevice");
 
   DialDeviceDescriptionData test_description("<xml>testDescription</xml>",
                                              GURL("http://127.0.0.1/apps"));
   api->SetDeviceForTest(test_device, test_description);
 
   ASSERT_TRUE(RunExtensionSubtest("dial/experimental",
                                   "fetch_device_description.html"));
 }
+
+IN_PROC_BROWSER_TEST_F(DialAPITest, GetNetworkId) {
+  fake_network_info = std::vector<NetworkInfo>{
+      {{0, net::NetworkChangeNotifier::CONNECTION_ETHERNET,
+        std::string("enp0s2"), std::string("ethernet1")}}};
+
+  ASSERT_TRUE(RunExtensionSubtest("dial/experimental", "get_network_id.html"));
+
+  ResultCatcher catcher;
+  EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
+}
+
+IN_PROC_BROWSER_TEST_F(DialAPITest, NetworkIdChanged) {
+  ASSERT_TRUE(
+      RunExtensionSubtest("dial/experimental", "network_id_changed.html"));
+
+  ResultCatcher catcher;
+  ExtensionTestMessageListener listener("continue", false);
+
+  fake_network_info = std::vector<NetworkInfo>{
+      {{0, net::NetworkChangeNotifier::CONNECTION_ETHERNET,
+        std::string("enp0s2"), std::string("ethernet1")}}};
+  net::NetworkChangeNotifier::NotifyObserversOfNetworkChangeForTests(
+      net::NetworkChangeNotifier::CONNECTION_ETHERNET);
+  EXPECT_TRUE(listener.WaitUntilSatisfied());
+
+  fake_network_info =
+      std::vector<NetworkInfo>{{{0, net::NetworkChangeNotifier::CONNECTION_WIFI,
+                                 std::string("wlp0s3"), std::string("wifi1")}}};
+  net::NetworkChangeNotifier::NotifyObserversOfNetworkChangeForTests(
+      net::NetworkChangeNotifier::CONNECTION_WIFI);
+  EXPECT_TRUE(listener.WaitUntilSatisfied());
+
+  EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
+}