| Index: ash/system/chromeos/network/network_state_notifier_unittest.cc
|
| diff --git a/ash/system/chromeos/network/network_state_notifier_unittest.cc b/ash/system/chromeos/network/network_state_notifier_unittest.cc
|
| index 2f681bcaf68920a71c9c94cac3a47bd22bb1d935..6e2069deb657d54c2c5b364c9f322fd71cc4ef09 100644
|
| --- a/ash/system/chromeos/network/network_state_notifier_unittest.cc
|
| +++ b/ash/system/chromeos/network/network_state_notifier_unittest.cc
|
| @@ -7,13 +7,14 @@
|
| #include "ash/root_window_controller.h"
|
| #include "ash/shelf/shelf_widget.h"
|
| #include "ash/shell.h"
|
| +#include "ash/system/chromeos/network/network_connect.h"
|
| #include "ash/system/status_area_widget.h"
|
| #include "ash/system/tray/system_tray.h"
|
| #include "ash/test/ash_test_base.h"
|
| #include "chromeos/dbus/dbus_thread_manager.h"
|
| #include "chromeos/dbus/shill_device_client.h"
|
| #include "chromeos/dbus/shill_service_client.h"
|
| -#include "chromeos/network/network_state_handler.h"
|
| +#include "chromeos/network/network_connection_handler.h"
|
| #include "third_party/cros_system_api/dbus/service_constants.h"
|
|
|
| namespace {
|
| @@ -27,7 +28,7 @@ ash::SystemTray* GetSystemTray() {
|
|
|
| using chromeos::DBusThreadManager;
|
| using chromeos::NetworkHandler;
|
| -using chromeos::NetworkStateHandler;
|
| +using chromeos::NetworkConnectionHandler;
|
| using chromeos::ShillDeviceClient;
|
| using chromeos::ShillServiceClient;
|
|
|
| @@ -71,17 +72,17 @@ class NetworkStateNotifierTest : public AshTestBase {
|
| const bool add_to_visible = true;
|
| // Create wifi and cellular networks and set to online.
|
| service_test->AddService("wifi1", "wifi1",
|
| - flimflam::kTypeWifi, flimflam::kStateOnline,
|
| + flimflam::kTypeWifi, flimflam::kStateIdle,
|
| add_to_visible, add_to_watchlist);
|
| - RunAllPendingInMessageLoop();
|
| - }
|
| -
|
| - void SetServiceState(const std::string& service_path,
|
| - const std::string& state) {
|
| - ShillServiceClient::TestInterface* service_test =
|
| - DBusThreadManager::Get()->GetShillServiceClient()->GetTestInterface();
|
| - service_test->SetServiceProperty(service_path, flimflam::kStateProperty,
|
| - base::StringValue(state));
|
| + service_test->SetServiceProperty("wifi1",
|
| + flimflam::kSecurityProperty,
|
| + base::StringValue(flimflam::kSecurityWep));
|
| + service_test->SetServiceProperty("wifi1",
|
| + flimflam::kConnectableProperty,
|
| + base::FundamentalValue(true));
|
| + service_test->SetServiceProperty("wifi1",
|
| + flimflam::kPassphraseProperty,
|
| + base::StringValue("failure"));
|
| RunAllPendingInMessageLoop();
|
| }
|
|
|
| @@ -91,20 +92,10 @@ class NetworkStateNotifierTest : public AshTestBase {
|
|
|
| TEST_F(NetworkStateNotifierTest, ConnectionFailure) {
|
| EXPECT_FALSE(GetSystemTray()->HasNotificationBubble());
|
| - // State -> Failure for non connecting network should not spawn a notification
|
| - SetServiceState("wifi1", flimflam::kStateFailure);
|
| - EXPECT_FALSE(GetSystemTray()->CloseNotificationBubbleForTest());
|
| - // State -> Failure for connecting network should spawn a notification
|
| - SetServiceState("wifi1", flimflam::kStateAssociation);
|
| - NetworkHandler::Get()->network_state_handler()->SetConnectingNetwork("wifi1");
|
| - SetServiceState("wifi1", flimflam::kStateFailure);
|
| + ash::network_connect::ConnectToNetwork("wifi1", NULL /* owning_window */);
|
| + RunAllPendingInMessageLoop();
|
| + // Failure should spawn a notification.
|
| EXPECT_TRUE(GetSystemTray()->CloseNotificationBubbleForTest());
|
| - // Failure -> Idle should not spawn a notification
|
| - SetServiceState("wifi1", flimflam::kStateIdle);
|
| - EXPECT_FALSE(GetSystemTray()->HasNotificationBubble());
|
| - // Idle -> Failure should also not spawn a notification
|
| - SetServiceState("wifi1", flimflam::kStateFailure);
|
| - EXPECT_FALSE(GetSystemTray()->HasNotificationBubble());
|
| }
|
|
|
| } // namespace test
|
|
|