Index: chromeos/components/tether/wifi_hotspot_connector_unittest.cc |
diff --git a/chromeos/components/tether/wifi_hotspot_connector_unittest.cc b/chromeos/components/tether/wifi_hotspot_connector_unittest.cc |
index 628d1329b8ddc7bc494524303171b4d210844e77..029ca82b261a676eab3127be59111fbe6d0225f5 100644 |
--- a/chromeos/components/tether/wifi_hotspot_connector_unittest.cc |
+++ b/chromeos/components/tether/wifi_hotspot_connector_unittest.cc |
@@ -12,6 +12,7 @@ |
#include "base/test/scoped_task_environment.h" |
#include "base/timer/mock_timer.h" |
#include "base/values.h" |
+#include "chromeos/components/tether/mock_host_connection_metrics_logger.h" |
#include "chromeos/dbus/dbus_thread_manager.h" |
#include "chromeos/dbus/shill_device_client.h" |
#include "chromeos/dbus/shill_service_client.h" |
@@ -139,8 +140,12 @@ class WifiHotspotConnectorTest : public NetworkStateTest { |
"tetherNetworkCarrier2" /* carrier */, 100 /* full battery */, |
100 /* full signal strength */, false /* has_connected_to_host */); |
+ mock_host_connection_metrics_logger_ = |
+ base::MakeUnique<MockHostConnectionMetricsLogger>(); |
+ |
wifi_hotspot_connector_ = base::WrapUnique(new WifiHotspotConnector( |
- network_state_handler(), test_network_connect_.get())); |
+ network_state_handler(), test_network_connect_.get(), |
+ mock_host_connection_metrics_logger_.get())); |
mock_timer_ = new base::MockTimer(true /* retain_user_task */, |
false /* is_repeating */); |
@@ -256,6 +261,8 @@ class WifiHotspotConnectorTest : public NetworkStateTest { |
base::MockTimer* mock_timer_; |
std::unique_ptr<TestNetworkConnect> test_network_connect_; |
+ std::unique_ptr<MockHostConnectionMetricsLogger> |
+ mock_host_connection_metrics_logger_; |
std::unique_ptr<WifiHotspotConnector> wifi_hotspot_connector_; |
@@ -264,6 +271,11 @@ class WifiHotspotConnectorTest : public NetworkStateTest { |
}; |
TEST_F(WifiHotspotConnectorTest, TestConnect_NetworkDoesNotBecomeConnectable) { |
+ EXPECT_CALL(*mock_host_connection_metrics_logger_, |
+ RecordConnectionToHostResult( |
+ HostConnectionMetricsLogger::ConnectionToHostResult:: |
+ CONNECTION_RESULT_FAILURE_CLIENT_CONNECTION_TIMEOUT)); |
+ |
wifi_hotspot_connector_->ConnectToWifiHotspot( |
std::string(kSsid), std::string(kPassword), kTetherNetworkGuid, |
base::Bind(&WifiHotspotConnectorTest::WifiConnectionCallback, |
@@ -284,6 +296,11 @@ TEST_F(WifiHotspotConnectorTest, TestConnect_NetworkDoesNotBecomeConnectable) { |
} |
TEST_F(WifiHotspotConnectorTest, TestConnect_AnotherNetworkBecomesConnectable) { |
+ EXPECT_CALL(*mock_host_connection_metrics_logger_, |
+ RecordConnectionToHostResult( |
+ HostConnectionMetricsLogger::ConnectionToHostResult:: |
+ CONNECTION_RESULT_FAILURE_CLIENT_CONNECTION_TIMEOUT)); |
+ |
wifi_hotspot_connector_->ConnectToWifiHotspot( |
std::string(kSsid), std::string(kPassword), kTetherNetworkGuid, |
base::Bind(&WifiHotspotConnectorTest::WifiConnectionCallback, |
@@ -311,6 +328,11 @@ TEST_F(WifiHotspotConnectorTest, TestConnect_AnotherNetworkBecomesConnectable) { |
} |
TEST_F(WifiHotspotConnectorTest, TestConnect_CannotConnectToNetwork) { |
+ EXPECT_CALL(*mock_host_connection_metrics_logger_, |
+ RecordConnectionToHostResult( |
+ HostConnectionMetricsLogger::ConnectionToHostResult:: |
+ CONNECTION_RESULT_FAILURE_CLIENT_CONNECTION_TIMEOUT)); |
+ |
wifi_hotspot_connector_->ConnectToWifiHotspot( |
std::string(kSsid), std::string(kPassword), kTetherNetworkGuid, |
base::Bind(&WifiHotspotConnectorTest::WifiConnectionCallback, |
@@ -384,6 +406,12 @@ TEST_F(WifiHotspotConnectorTest, TestConnect_Success_EmptyPassword) { |
TEST_F(WifiHotspotConnectorTest, |
TestConnect_SecondConnectionWhileWaitingForFirstToBecomeConnectable) { |
+ EXPECT_CALL( |
+ *mock_host_connection_metrics_logger_, |
+ RecordConnectionToHostResult( |
+ HostConnectionMetricsLogger::ConnectionToHostResult:: |
+ CONNECTION_RESULT_FAILURE_CLIENT_CONNECTION_CANCELED_BY_NEW_ATTEMPT)); |
+ |
wifi_hotspot_connector_->ConnectToWifiHotspot( |
"ssid1", "password1", "tetherNetworkGuid1", |
base::Bind(&WifiHotspotConnectorTest::WifiConnectionCallback, |
@@ -439,6 +467,12 @@ TEST_F(WifiHotspotConnectorTest, |
TEST_F(WifiHotspotConnectorTest, |
TestConnect_SecondConnectionWhileWaitingForFirstToConnect) { |
+ EXPECT_CALL( |
+ *mock_host_connection_metrics_logger_, |
+ RecordConnectionToHostResult( |
+ HostConnectionMetricsLogger::ConnectionToHostResult:: |
+ CONNECTION_RESULT_FAILURE_CLIENT_CONNECTION_CANCELED_BY_NEW_ATTEMPT)); |
+ |
wifi_hotspot_connector_->ConnectToWifiHotspot( |
"ssid1", "password1", kTetherNetworkGuid, |
base::Bind(&WifiHotspotConnectorTest::WifiConnectionCallback, |