1 // Copyright 2014 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "base/command_line.h"
6 #include "base/run_loop.h"
7 #include "content/public/common/content_switches.h"
8 #include "content/public/test/browser_test_utils.h"
9 #include "content/public/test/content_browser_test.h"
10 #include "content/public/test/content_browser_test_utils.h"
11 #include "content/shell/browser/shell.h"
12 #include "net/base/network_change_notifier.h"
13 #include "net/base/network_change_notifier_factory.h"
14
15 class NetInfoBrowserTest : public content::ContentBrowserTest {
16 protected:
SetUpCommandLine(base::CommandLine * command_line)17 virtual void SetUpCommandLine(base::CommandLine* command_line) OVERRIDE {
18 // TODO(jkarlin): Once NetInfo is enabled on all platforms remove this
19 // switch.
20 command_line->AppendSwitch(switches::kEnableNetworkInformation);
21 }
22
23 #if defined(OS_CHROMEOS)
SetUp()24 virtual void SetUp() OVERRIDE {
25 // ChromeOS's NetworkChangeNotifier isn't known to content and therefore
26 // doesn't get created in content_browsertests. Insert a mock
27 // NetworkChangeNotifier.
28 net::NetworkChangeNotifier::CreateMock();
29 content::ContentBrowserTest::SetUp();
30 }
31 #endif
32
SetUpOnMainThread()33 virtual void SetUpOnMainThread() OVERRIDE {
34 net::NetworkChangeNotifier::SetTestNotificationsOnly(true);
35 base::RunLoop().RunUntilIdle();
36 }
37
SetConnectionType(net::NetworkChangeNotifier::ConnectionType type)38 static void SetConnectionType(
39 net::NetworkChangeNotifier::ConnectionType type) {
40 net::NetworkChangeNotifier::NotifyObserversOfConnectionTypeChangeForTests(
41 type);
42 base::RunLoop().RunUntilIdle();
43 }
44
RunScriptExtractString(const std::string & script)45 std::string RunScriptExtractString(const std::string& script) {
46 std::string data;
47 EXPECT_TRUE(
48 ExecuteScriptAndExtractString(shell()->web_contents(), script, &data));
49 return data;
50 }
51
RunScriptExtractBool(const std::string & script)52 bool RunScriptExtractBool(const std::string& script) {
53 bool data;
54 EXPECT_TRUE(
55 ExecuteScriptAndExtractBool(shell()->web_contents(), script, &data));
56 return data;
57 }
58 };
59
60 // Make sure that type changes in the browser make their way to
61 // navigator.connection.type.
IN_PROC_BROWSER_TEST_F(NetInfoBrowserTest,NetworkChangePlumbsToNavigator)62 IN_PROC_BROWSER_TEST_F(NetInfoBrowserTest, NetworkChangePlumbsToNavigator) {
63 NavigateToURL(shell(), content::GetTestUrl("", "net_info.html"));
64 SetConnectionType(net::NetworkChangeNotifier::CONNECTION_WIFI);
65 EXPECT_EQ("wifi", RunScriptExtractString("getType()"));
66 SetConnectionType(net::NetworkChangeNotifier::CONNECTION_ETHERNET);
67 EXPECT_EQ("ethernet", RunScriptExtractString("getType()"));
68 }
69
70 // Make sure that type changes in the browser make their way to
71 // navigator.isOnline.
IN_PROC_BROWSER_TEST_F(NetInfoBrowserTest,IsOnline)72 IN_PROC_BROWSER_TEST_F(NetInfoBrowserTest, IsOnline) {
73 NavigateToURL(shell(), content::GetTestUrl("", "net_info.html"));
74 SetConnectionType(net::NetworkChangeNotifier::CONNECTION_ETHERNET);
75 EXPECT_TRUE(RunScriptExtractBool("getOnLine()"));
76 SetConnectionType(net::NetworkChangeNotifier::CONNECTION_NONE);
77 EXPECT_FALSE(RunScriptExtractBool("getOnLine()"));
78 SetConnectionType(net::NetworkChangeNotifier::CONNECTION_WIFI);
79 EXPECT_TRUE(RunScriptExtractBool("getOnLine()"));
80 }
81