1 // Copyright 2013 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/bind.h" 6 #include "base/bind_helpers.h" 7 #include "base/logging.h" 8 #include "base/message_loop/message_loop.h" 9 #include "base/run_loop.h" 10 #include "components/onc/onc_constants.h" 11 #include "components/wifi/wifi_service.h" 12 #include "testing/gtest/include/gtest/gtest.h" 13 14 namespace wifi { 15 16 class WiFiServiceTest : public testing::Test { 17 public: 18 void LogError(const std::string& error_name, 19 scoped_ptr<base::DictionaryValue> error_data) { 20 LOG(ERROR) << "WiFi Error: " << error_name; 21 QuitRunLoop(); 22 } 23 24 void TestError(const std::string& error_name, 25 scoped_ptr<base::DictionaryValue> error_data) { 26 LOG(ERROR) << "WiFi Error: " << error_name; 27 FAIL(); 28 } 29 30 void OnDictionaryResult(const std::string& network_guid, 31 const base::DictionaryValue& dictionary) {} 32 33 void OnNetworkProperties(const std::string& network_guid, 34 const WiFiService::NetworkProperties& properties) { 35 LOG(INFO) << "OnNetworkProperties" << *properties.ToValue(false).release(); 36 } 37 38 void OnNetworksChangedEventWaitingForConnect( 39 const WiFiService::NetworkGuidList& network_guid_list) { 40 LOG(INFO) << "Networks Changed: " << network_guid_list.front(); 41 // Check that network is now connected. 42 wifi_service_->GetProperties( 43 connected_network_guid_, 44 base::Bind(&WiFiServiceTest::WaitForConnect, base::Unretained(this)), 45 base::Bind(&WiFiServiceTest::TestError, base::Unretained(this))); 46 } 47 48 void OnNetworkConnectStarted(const std::string& network_guid) { 49 LOG(INFO) << "Started Network Connect:" << network_guid; 50 wifi_service_->SetNetworksChangedObserver( 51 base::Bind(&WiFiServiceTest::OnNetworksChangedEventWaitingForConnect, 52 base::Unretained(this))); 53 base::MessageLoop::current()->PostDelayedTask( 54 FROM_HERE, 55 base::MessageLoop::QuitClosure(), 56 base::TimeDelta::FromSeconds(10)); 57 } 58 59 void WaitForConnect(const std::string& network_guid, 60 const WiFiService::NetworkProperties& properties) { 61 LOG(INFO) << "WaitForConnect" << *properties.ToValue(false).release(); 62 if (properties.connection_state == onc::connection_state::kConnected) 63 QuitRunLoop(); 64 } 65 66 void WaitForDisconnect(const std::string& network_guid, 67 const WiFiService::NetworkProperties& properties) { 68 LOG(INFO) << "WaitForDisconnect" << *properties.ToValue(false).release(); 69 EXPECT_TRUE(properties.connection_state == 70 onc::connection_state::kNotConnected); 71 } 72 73 void OnNetworkDisconnectStarted(const std::string& network_guid) { 74 LOG(INFO) << "Started Network Disconnect:" << network_guid; 75 // Check that Network state has changed to 'Not Connected'. 76 wifi_service_->GetProperties( 77 connected_network_guid_, 78 base::Bind(&WiFiServiceTest::WaitForDisconnect, base::Unretained(this)), 79 base::Bind(&WiFiServiceTest::TestError, base::Unretained(this))); 80 // Start connect back to the same network. 81 wifi_service_->StartConnect( 82 connected_network_guid_, 83 base::Bind(&WiFiServiceTest::OnNetworkConnectStarted, 84 base::Unretained(this)), 85 base::Bind(&WiFiServiceTest::TestError, base::Unretained(this))); 86 } 87 88 void OnVisibleNetworks(const WiFiService::NetworkList& network_list) { 89 LOG(INFO) << "Visible WiFi Networks: " << network_list.size(); 90 } 91 92 void FindConnectedNetwork(const WiFiService::NetworkList& network_list) { 93 for (WiFiService::NetworkList::const_iterator net = network_list.begin(); 94 net != network_list.end(); 95 ++net) { 96 if (net->connection_state == onc::connection_state::kConnected) { 97 connected_network_guid_ = net->guid; 98 LOG(INFO) << "Connected Network:\n" << *(net->ToValue(false).release()); 99 } 100 } 101 102 if (!connected_network_guid_.empty()) { 103 wifi_service_->StartDisconnect( 104 connected_network_guid_, 105 base::Bind(&WiFiServiceTest::OnNetworkDisconnectStarted, 106 base::Unretained(this)), 107 base::Bind(&WiFiServiceTest::TestError, base::Unretained(this))); 108 } else { 109 LOG(INFO) << "No Connected Networks, skipping disconnect."; 110 QuitRunLoop(); 111 } 112 } 113 114 void QuitRunLoop() { 115 base::MessageLoop::current()->PostTask(FROM_HERE, 116 base::MessageLoop::QuitClosure()); 117 } 118 119 protected: 120 virtual void SetUp() { wifi_service_.reset(WiFiService::CreateService()); } 121 virtual void TearDown() {} 122 123 std::string connected_network_guid_; 124 scoped_ptr<WiFiService> wifi_service_; 125 base::MessageLoop loop_; 126 }; 127 128 // Test getting list of visible networks. 129 TEST_F(WiFiServiceTest, GetVisibleNetworks) { 130 wifi_service_->GetVisibleNetworks( 131 base::Bind(&WiFiServiceTest::OnVisibleNetworks, base::Unretained(this)), 132 base::Bind(&WiFiServiceTest::LogError, base::Unretained(this))); 133 } 134 135 // Test that connected WiFi network can be disconnected and reconnected. 136 // Disabled to avoid network connection interruption unless enabled explicitly. 137 TEST_F(WiFiServiceTest, DISABLED_Reconnect) { 138 base::RunLoop run_loop; 139 wifi_service_->GetVisibleNetworks( 140 base::Bind(&WiFiServiceTest::FindConnectedNetwork, 141 base::Unretained(this)), 142 base::Bind(&WiFiServiceTest::LogError, base::Unretained(this))); 143 run_loop.Run(); 144 } 145 146 } // namespace wifi 147
