Prevent networkingPrivate.forgetNetwork from removing shared configs

extensions/browser/api/networking_private/networking_private_chromeos_unittest.cc

 // Copyright 2017 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 "base/macros.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/stringprintf.h"
 #include "chromeos/dbus/dbus_thread_manager.h"
@@ skipping 447 matching lines @@
                                                         "Priority": 1,
                                                         "Type": "WiFi",
                                                         "WiFi": {
                                                           "HexSSID": "%s",
                                                           "Security": "WPA-PSK"
                                                         }
                                                       }])",
                                                    network_hex_ssid.c_str())));
 }
 
+TEST_F(NetworkingPrivateApiTest, ForgetSharedNetwork) {
+  EXPECT_EQ(networking_private::kErrorAccessToSharedConfig,
+            RunFunctionAndReturnError(
+                new NetworkingPrivateForgetNetworkFunction(),
+                base::StringPrintf(R"(["%s"])", kSharedWifiGuid)));
+
+  std::string profile_path;
+  EXPECT_TRUE(GetServiceProfile(kSharedWifiServicePath, &profile_path));
+  EXPECT_EQ(chromeos::ShillProfileClient::GetSharedProfilePath(), profile_path);
+}
+
+TEST_F(NetworkingPrivateApiTest, ForgetPrivateNetwork) {
+  RunFunction(new NetworkingPrivateForgetNetworkFunction(),
+              base::StringPrintf(R"(["%s"])", kPrivateWifiGuid));
+
+  std::string profile_path;
+  EXPECT_FALSE(GetServiceProfile(kPrivateWifiServicePath, &profile_path));
+}
+
+TEST_F(NetworkingPrivateApiTest, ForgetPrivateNetworkWebUI) {
+  scoped_refptr<NetworkingPrivateForgetNetworkFunction> forget_network =
+      new NetworkingPrivateForgetNetworkFunction();
+  forget_network->set_source_context_type(Feature::WEBUI_CONTEXT);
+
+  RunFunction(forget_network.get(),
+              base::StringPrintf(R"(["%s"])", kPrivateWifiGuid));
+
+  std::string profile_path;
+  EXPECT_FALSE(GetServiceProfile(kPrivateWifiServicePath, &profile_path));
+}
+
+TEST_F(NetworkingPrivateApiTest, ForgetUserPolicyNetwork) {
+  EXPECT_EQ(networking_private::kErrorPolicyControlled,
+            RunFunctionAndReturnError(
+                new NetworkingPrivateForgetNetworkFunction(),
+                base::StringPrintf(R"(["%s"])", kManagedUserWifiGuid)));
+
+  const chromeos::NetworkState* network =
+      chromeos::NetworkHandler::Get()
+          ->network_state_handler()
+          ->GetNetworkStateFromGuid(kManagedUserWifiGuid);
+  std::string profile_path;
+  EXPECT_TRUE(GetServiceProfile(network->path(), &profile_path));
+  EXPECT_EQ(kUserProfilePath, profile_path);
+}
+
+TEST_F(NetworkingPrivateApiTest, ForgetUserPolicyNetworkWebUI) {
+  scoped_refptr<NetworkingPrivateForgetNetworkFunction> forget_network =
+      new NetworkingPrivateForgetNetworkFunction();
+  forget_network->set_source_context_type(Feature::WEBUI_CONTEXT);
+
+  EXPECT_EQ(networking_private::kErrorPolicyControlled,
+            RunFunctionAndReturnError(
+                forget_network.get(),
+                base::StringPrintf(R"(["%s"])", kManagedUserWifiGuid)));
+
+  const chromeos::NetworkState* network =
+      chromeos::NetworkHandler::Get()
+          ->network_state_handler()
+          ->GetNetworkStateFromGuid(kManagedUserWifiGuid);
+  std::string profile_path;
+  EXPECT_TRUE(GetServiceProfile(network->path(), &profile_path));
+  EXPECT_EQ(kUserProfilePath, profile_path);
+}
+
+TEST_F(NetworkingPrivateApiTest, ForgetDevicePolicyNetworkWebUI) {
+  scoped_refptr<NetworkingPrivateForgetNetworkFunction> forget_network =
+      new NetworkingPrivateForgetNetworkFunction();
+  forget_network->set_source_context_type(Feature::WEBUI_CONTEXT);
+
+  EXPECT_EQ(networking_private::kErrorPolicyControlled,
+            RunFunctionAndReturnError(
+                forget_network.get(),
+                base::StringPrintf(R"(["%s"])", kManagedDeviceWifiGuid)));
+
+  const chromeos::NetworkState* network =
+      chromeos::NetworkHandler::Get()
+          ->network_state_handler()
+          ->GetNetworkStateFromGuid(kManagedDeviceWifiGuid);
+  std::string profile_path;
+  EXPECT_TRUE(GetServiceProfile(network->path(), &profile_path));
+  EXPECT_EQ(chromeos::ShillProfileClient::GetSharedProfilePath(), profile_path);
+}
+
+// Tests that forgetNetwork in case there is a network config in both user and
+// shared profile - only config from the user profile is expected to be removed.
+TEST_F(NetworkingPrivateApiTest, ForgetNetworkInMultipleProfiles) {
+  AddSharedNetworkToUserProfile();
+
+  std::string profile_path;
+  EXPECT_TRUE(GetServiceProfile(kSharedWifiServicePath, &profile_path));
+  ASSERT_EQ(kUserProfilePath, profile_path);
+
+  RunFunction(new NetworkingPrivateForgetNetworkFunction(),
+              base::StringPrintf(R"(["%s"])", kSharedWifiGuid));
+
+  EXPECT_TRUE(GetServiceProfile(kSharedWifiServicePath, &profile_path));
+  EXPECT_EQ(chromeos::ShillProfileClient::GetSharedProfilePath(), profile_path);
+}
+
+TEST_F(NetworkingPrivateApiTest, ForgetNetworkInMultipleProfilesWebUI) {
+  AddSharedNetworkToUserProfile();
+
+  std::string profile_path;
+  EXPECT_TRUE(GetServiceProfile(kSharedWifiServicePath, &profile_path));
+  ASSERT_EQ(kUserProfilePath, profile_path);
+
+  scoped_refptr<NetworkingPrivateForgetNetworkFunction> forget_network =
+      new NetworkingPrivateForgetNetworkFunction();
+  forget_network->set_source_context_type(Feature::WEBUI_CONTEXT);
+
+  RunFunction(forget_network.get(),
+              base::StringPrintf(R"(["%s"])", kSharedWifiGuid));
+
+  EXPECT_FALSE(GetServiceProfile(kSharedWifiServicePath, &profile_path));
+}
+
 }  // namespace extensions