Handle device ONC AllowOnlyPolicyNetworksToConnect

chromeos/network/network_connection_handler_unittest.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 "chromeos/network/network_connection_handler.h"
 
 #include <map>
 #include <set>
 
 #include "base/bind.h"
@@ skipping 315 matching lines @@
     "{ \"GUID\": \"wifi0\", \"Type\": \"wifi\", \"State\": \"idle\", "
     "  \"Connectable\": true }";
 const char* kConfigConnected =
     "{ \"GUID\": \"wifi1\", \"Type\": \"wifi\", \"State\": \"online\" }";
 const char* kConfigConnecting =
     "{ \"GUID\": \"wifi2\", \"Type\": \"wifi\", \"State\": \"association\" }";
 const char* kConfigRequiresPassphrase =
     "{ \"GUID\": \"wifi3\", \"Type\": \"wifi\", "
     "  \"PassphraseRequired\": true }";
 
+const char* kPolicyWifi0 =
+    "[{ \"GUID\": \"wifi0\",  \"IPAddressConfigType\": \"DHCP\", "
+    "   \"Type\": \"WiFi\", \"Name\": \"My WiFi Network\","
+    "   \"WiFi\": { \"SSID\": \"wifi0\"}}]";
+
 }  // namespace
 
 TEST_F(NetworkConnectionHandlerTest, NetworkConnectionHandlerConnectSuccess) {
   EXPECT_TRUE(Configure(kConfigConnectable));
   Connect(kWifi0);
   EXPECT_EQ(kSuccessResult, GetResultAndReset());
   EXPECT_EQ(shill::kStateOnline,
             GetServiceStringProperty(kWifi0, shill::kStateProperty));
   // Observer expectations
   EXPECT_TRUE(network_connection_observer_->GetRequested(kWifi0));
   EXPECT_EQ(kSuccessResult, network_connection_observer_->GetResult(kWifi0));
 }
 
+TEST_F(NetworkConnectionHandlerTest,
+       NetworkConnectionHandlerConnectProhibited) {
+  EXPECT_TRUE(Configure(kConfigConnectable));
+  base::DictionaryValue global_config;
+  global_config.SetBooleanWithoutPathExpansion(
+      ::onc::global_network_config::kAllowOnlyPolicyNetworksToConnect, true);
+  SetupPolicy("[]", global_config, false /* load as device policy */);
+  LoginToRegularUser();
+  Connect(kWifi0);
+  EXPECT_EQ(NetworkConnectionHandler::kErrorUnmanagedNetwork,
+            GetResultAndReset());
+
+  SetupPolicy(kPolicyWifi0, global_config, false /* load as device policy */);
+  Connect(kWifi0);
+  EXPECT_EQ(kSuccessResult, GetResultAndReset());
+}
+
 // Handles basic failure cases.
 TEST_F(NetworkConnectionHandlerTest, NetworkConnectionHandlerConnectFailure) {
   Connect(kNoNetwork);
   EXPECT_EQ(NetworkConnectionHandler::kErrorConfigureFailed,
             GetResultAndReset());
   EXPECT_TRUE(network_connection_observer_->GetRequested(kNoNetwork));
   EXPECT_EQ(NetworkConnectionHandler::kErrorConfigureFailed,
             network_connection_observer_->GetResult(kNoNetwork));
 
   EXPECT_TRUE(Configure(kConfigConnected));
@@ skipping 107 matching lines @@
   Connect(kNoNetwork);
   EXPECT_EQ(NetworkConnectionHandler::kErrorConfigureFailed,
             GetResultAndReset());
 
   EXPECT_TRUE(Configure(kConfigConnectable));
   Disconnect(kWifi0);
   EXPECT_EQ(NetworkConnectionHandler::kErrorNotConnected, GetResultAndReset());
 }
 
 }  // namespace chromeos