chrome/browser/enterprise/reporting/policy_info_unittest.cc - chromium/src - Git at Google

 // Copyright 2019 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 "components/enterprise/browser/reporting/policy_info.h"

 #include "base/files/file_util.h"
 #include "base/memory/raw_ptr.h"
 #include "base/strings/utf_string_conversions.h"
 #include "build/build_config.h"
 #include "build/chromeos_buildflags.h"
 #include "chrome/browser/policy/chrome_browser_policy_connector.h"
 #include "chrome/browser/policy/chrome_policy_conversions_client.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/test/base/testing_browser_process.h"
 #include "chrome/test/base/testing_profile.h"
 #include "chrome/test/base/testing_profile_manager.h"
 #include "components/policy/core/browser/policy_conversions.h"
 #include "components/policy/core/common/mock_policy_service.h"
 #include "components/policy/core/common/policy_map.h"
 #include "components/policy/proto/device_management_backend.pb.h"
 #include "components/sync_preferences/pref_service_syncable.h"
 #include "content/public/test/browser_task_environment.h"
 #include "testing/gtest/include/gtest/gtest.h"

 #if !defined(OS_ANDROID)
 #include "extensions/browser/extension_registry.h"
 #include "extensions/common/extension_builder.h"
 #include "extensions/common/manifest_constants.h"
 #endif  // !defined(OS_ANDROID)

 namespace em = enterprise_management;

 namespace enterprise_reporting {
 namespace {
 constexpr char kPolicyName1[] = "policy_a";
 constexpr char kPolicyName2[] = "policy_b";

 constexpr char kExtensionId1[] = "abcdefghijklmnoabcdefghijklmnoab";
 constexpr char kExtensionId2[] = "abcdefghijklmnoabcdefghijklmnoac";
 }  // namespace

 using ::testing::_;
 using ::testing::Eq;

 // TODO(crbug.com/1096499): Get rid of chrome/browser dependencies and then
 // move this file to components/enterprise/browser.
 class PolicyInfoTest : public ::testing::Test {
  public:
   void SetUp() override {
     profile_manager_ = std::make_unique<TestingProfileManager>(
         TestingBrowserProcess::GetGlobal());
     ASSERT_TRUE(profile_manager_->SetUp());
     std::string test_profile_name = "test_profile";
     profile_ = profile_manager_->CreateTestingProfile(
         test_profile_name,
         std::unique_ptr<sync_preferences::PrefServiceSyncable>(),
         base::UTF8ToUTF16(test_profile_name), 0, std::string(),
         TestingProfile::TestingFactories(), absl::optional<bool>(),
         GetPolicyService());
     profile_manager_->CreateTestingProfile(chrome::kInitialProfile);
   }

   std::unique_ptr<policy::MockPolicyService> GetPolicyService() {
     auto policy_service = std::make_unique<policy::MockPolicyService>();
     ON_CALL(*policy_service.get(),
             GetPolicies(Eq(policy::PolicyNamespace(policy::POLICY_DOMAIN_CHROME,
                                                    std::string()))))
         .WillByDefault(::testing::ReturnRef(policy_map_));
     ON_CALL(*policy_service.get(),
             GetPolicies(Eq(policy::PolicyNamespace(
                 policy::POLICY_DOMAIN_EXTENSIONS, kExtensionId1))))
         .WillByDefault(::testing::ReturnRef(extension_policy_map_));
     ON_CALL(*policy_service.get(),
             GetPolicies(Eq(policy::PolicyNamespace(
                 policy::POLICY_DOMAIN_EXTENSIONS, kExtensionId2))))
         .WillByDefault(::testing::ReturnRef(empty_policy_map_));

     policy_service_ = policy_service.get();
     return policy_service;
   }

   TestingProfile* profile() { return profile_; }
   policy::PolicyMap* policy_map() { return &policy_map_; }
   policy::PolicyMap* extension_policy_map() { return &extension_policy_map_; }
   policy::MockPolicyService* policy_service() { return policy_service_; }

  private:
   content::BrowserTaskEnvironment task_environment_;
   std::unique_ptr<TestingProfileManager> profile_manager_;
   raw_ptr<TestingProfile> profile_ = nullptr;
   policy::PolicyMap policy_map_;
   policy::PolicyMap extension_policy_map_;
   policy::PolicyMap empty_policy_map_;
   raw_ptr<policy::MockPolicyService> policy_service_;
 };

 // Verify two Chrome policies are appended to the Profile report properly.
 TEST_F(PolicyInfoTest, ChromePolicy) {
   policy_map()->Set(kPolicyName1, policy::POLICY_LEVEL_MANDATORY,
                     policy::POLICY_SCOPE_USER, policy::POLICY_SOURCE_CLOUD,
                     base::Value(std::vector<base::Value>()), nullptr);
   policy_map()->Set(kPolicyName2, policy::POLICY_LEVEL_RECOMMENDED,
                     policy::POLICY_SCOPE_MACHINE, policy::POLICY_SOURCE_MERGED,
                     base::Value(true), nullptr);
   em::ChromeUserProfileInfo profile_info;

   EXPECT_CALL(*policy_service(), GetPolicies(_));

   auto client =
       std::make_unique<policy::ChromePolicyConversionsClient>(profile());
   AppendChromePolicyInfoIntoProfileReport(
       policy::DictionaryPolicyConversions(std::move(client))
           .EnableConvertTypes(false)
           .EnablePrettyPrint(false)
           .ToValue(),
       &profile_info);
   EXPECT_EQ(2, profile_info.chrome_policies_size());

   auto policy1 = profile_info.chrome_policies(0);
   EXPECT_EQ(kPolicyName1, policy1.name());
   EXPECT_EQ("[]", policy1.value());
   EXPECT_EQ(em::Policy_PolicyLevel_LEVEL_MANDATORY, policy1.level());
   EXPECT_EQ(em::Policy_PolicyScope_SCOPE_USER, policy1.scope());
   EXPECT_EQ(em::Policy_PolicySource_SOURCE_CLOUD, policy1.source());
   EXPECT_NE("", policy1.error());

   auto policy2 = profile_info.chrome_policies(1);
   EXPECT_EQ(kPolicyName2, policy2.name());
   EXPECT_EQ("true", policy2.value());
   EXPECT_EQ(em::Policy_PolicyLevel_LEVEL_RECOMMENDED, policy2.level());
   EXPECT_EQ(em::Policy_PolicyScope_SCOPE_MACHINE, policy2.scope());
   EXPECT_EQ(em::Policy_PolicySource_SOURCE_MERGED, policy2.source());
   EXPECT_NE("", policy2.error());
 }

 #if !defined(OS_ANDROID)
 TEST_F(PolicyInfoTest, ExtensionPolicy) {
   EXPECT_CALL(*policy_service(), GetPolicies(_)).Times(3);
   extensions::ExtensionRegistry* extension_registry =
       extensions::ExtensionRegistry::Get(profile());

   extension_registry->AddEnabled(
       extensions::ExtensionBuilder("extension_name")
           .SetID(kExtensionId1)
           .SetManifestPath({"storage", "managed_schema"}, "schema.json")
           .Build());
   extension_registry->AddEnabled(
       extensions::ExtensionBuilder("extension_name")
           .SetID(kExtensionId2)
           .SetManifestPath({"storage", "managed_schema"}, "schema.json")
           .Build());

   extension_policy_map()->Set(kPolicyName1, policy::POLICY_LEVEL_MANDATORY,
                               policy::POLICY_SCOPE_MACHINE,
                               policy::POLICY_SOURCE_PLATFORM, base::Value(3),
                               nullptr);
   em::ChromeUserProfileInfo profile_info;
   auto client =
       std::make_unique<policy::ChromePolicyConversionsClient>(profile());
   AppendExtensionPolicyInfoIntoProfileReport(
       policy::DictionaryPolicyConversions(std::move(client))
           .EnableConvertTypes(false)
           .EnablePrettyPrint(false)
           .ToValue(),
       &profile_info);
   // The second extension is not in the report because it has no policy.
   EXPECT_EQ(1, profile_info.extension_policies_size());
   EXPECT_EQ(kExtensionId1, profile_info.extension_policies(0).extension_id());
   EXPECT_EQ(1, profile_info.extension_policies(0).policies_size());

   auto policy1 = profile_info.extension_policies(0).policies(0);
   EXPECT_EQ(kPolicyName1, policy1.name());
   EXPECT_EQ("3", policy1.value());
   EXPECT_EQ(em::Policy_PolicyLevel_LEVEL_MANDATORY, policy1.level());
   EXPECT_EQ(em::Policy_PolicyScope_SCOPE_MACHINE, policy1.scope());
   EXPECT_EQ(em::Policy_PolicySource_SOURCE_PLATFORM, policy1.source());
   EXPECT_NE(std::string(), policy1.error());
 }
 #endif  // !defined(OS_ANDROID)

 TEST_F(PolicyInfoTest, MachineLevelUserCloudPolicyFetchTimestamp) {
   em::ChromeUserProfileInfo profile_info;
 #if !BUILDFLAG(IS_CHROMEOS_ASH)
   AppendMachineLevelUserCloudPolicyFetchTimestamp(
       &profile_info, g_browser_process->browser_policy_connector()
                          ->machine_level_user_cloud_policy_manager());
 #endif  // !BUILDFLAG(IS_CHROMEOS_ASH)
   EXPECT_EQ(0, profile_info.policy_fetched_timestamps_size());
 }

 }  // namespace enterprise_reporting
	// Copyright 2019 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 "components/enterprise/browser/reporting/policy_info.h"

	#include "base/files/file_util.h"
	#include "base/memory/raw_ptr.h"
	#include "base/strings/utf_string_conversions.h"
	#include "build/build_config.h"
	#include "build/chromeos_buildflags.h"
	#include "chrome/browser/policy/chrome_browser_policy_connector.h"
	#include "chrome/browser/policy/chrome_policy_conversions_client.h"
	#include "chrome/common/chrome_constants.h"
	#include "chrome/test/base/testing_browser_process.h"
	#include "chrome/test/base/testing_profile.h"
	#include "chrome/test/base/testing_profile_manager.h"
	#include "components/policy/core/browser/policy_conversions.h"
	#include "components/policy/core/common/mock_policy_service.h"
	#include "components/policy/core/common/policy_map.h"
	#include "components/policy/proto/device_management_backend.pb.h"
	#include "components/sync_preferences/pref_service_syncable.h"
	#include "content/public/test/browser_task_environment.h"
	#include "testing/gtest/include/gtest/gtest.h"

	#if !defined(OS_ANDROID)
	#include "extensions/browser/extension_registry.h"
	#include "extensions/common/extension_builder.h"
	#include "extensions/common/manifest_constants.h"
	#endif // !defined(OS_ANDROID)

	namespace em = enterprise_management;

	namespace enterprise_reporting {
	namespace {
	constexpr char kPolicyName1[] = "policy_a";
	constexpr char kPolicyName2[] = "policy_b";

	constexpr char kExtensionId1[] = "abcdefghijklmnoabcdefghijklmnoab";
	constexpr char kExtensionId2[] = "abcdefghijklmnoabcdefghijklmnoac";
	} // namespace

	using ::testing::_;
	using ::testing::Eq;

	// TODO(crbug.com/1096499): Get rid of chrome/browser dependencies and then
	// move this file to components/enterprise/browser.
	class PolicyInfoTest : public ::testing::Test {
	public:
	void SetUp() override {
	profile_manager_ = std::make_unique<TestingProfileManager>(
	TestingBrowserProcess::GetGlobal());
	ASSERT_TRUE(profile_manager_->SetUp());
	std::string test_profile_name = "test_profile";
	profile_ = profile_manager_->CreateTestingProfile(
	test_profile_name,
	std::unique_ptr<sync_preferences::PrefServiceSyncable>(),
	base::UTF8ToUTF16(test_profile_name), 0, std::string(),
	TestingProfile::TestingFactories(), absl::optional<bool>(),
	GetPolicyService());
	profile_manager_->CreateTestingProfile(chrome::kInitialProfile);
	}

	std::unique_ptr<policy::MockPolicyService> GetPolicyService() {
	auto policy_service = std::make_unique<policy::MockPolicyService>();
	ON_CALL(*policy_service.get(),
	GetPolicies(Eq(policy::PolicyNamespace(policy::POLICY_DOMAIN_CHROME,
	std::string()))))
	.WillByDefault(::testing::ReturnRef(policy_map_));
	ON_CALL(*policy_service.get(),
	GetPolicies(Eq(policy::PolicyNamespace(
	policy::POLICY_DOMAIN_EXTENSIONS, kExtensionId1))))
	.WillByDefault(::testing::ReturnRef(extension_policy_map_));
	ON_CALL(*policy_service.get(),
	GetPolicies(Eq(policy::PolicyNamespace(
	policy::POLICY_DOMAIN_EXTENSIONS, kExtensionId2))))
	.WillByDefault(::testing::ReturnRef(empty_policy_map_));

	policy_service_ = policy_service.get();
	return policy_service;
	}

	TestingProfile* profile() { return profile_; }
	policy::PolicyMap* policy_map() { return &policy_map_; }
	policy::PolicyMap* extension_policy_map() { return &extension_policy_map_; }
	policy::MockPolicyService* policy_service() { return policy_service_; }

	private:
	content::BrowserTaskEnvironment task_environment_;
	std::unique_ptr<TestingProfileManager> profile_manager_;
	raw_ptr<TestingProfile> profile_ = nullptr;
	policy::PolicyMap policy_map_;
	policy::PolicyMap extension_policy_map_;
	policy::PolicyMap empty_policy_map_;
	raw_ptr<policy::MockPolicyService> policy_service_;
	};

	// Verify two Chrome policies are appended to the Profile report properly.
	TEST_F(PolicyInfoTest, ChromePolicy) {
	policy_map()->Set(kPolicyName1, policy::POLICY_LEVEL_MANDATORY,
	policy::POLICY_SCOPE_USER, policy::POLICY_SOURCE_CLOUD,
	base::Value(std::vector<base::Value>()), nullptr);
	policy_map()->Set(kPolicyName2, policy::POLICY_LEVEL_RECOMMENDED,
	policy::POLICY_SCOPE_MACHINE, policy::POLICY_SOURCE_MERGED,
	base::Value(true), nullptr);
	em::ChromeUserProfileInfo profile_info;

	EXPECT_CALL(*policy_service(), GetPolicies(_));

	auto client =
	std::make_unique<policy::ChromePolicyConversionsClient>(profile());
	AppendChromePolicyInfoIntoProfileReport(
	policy::DictionaryPolicyConversions(std::move(client))
	.EnableConvertTypes(false)
	.EnablePrettyPrint(false)
	.ToValue(),
	&profile_info);
	EXPECT_EQ(2, profile_info.chrome_policies_size());

	auto policy1 = profile_info.chrome_policies(0);
	EXPECT_EQ(kPolicyName1, policy1.name());
	EXPECT_EQ("[]", policy1.value());
	EXPECT_EQ(em::Policy_PolicyLevel_LEVEL_MANDATORY, policy1.level());
	EXPECT_EQ(em::Policy_PolicyScope_SCOPE_USER, policy1.scope());
	EXPECT_EQ(em::Policy_PolicySource_SOURCE_CLOUD, policy1.source());
	EXPECT_NE("", policy1.error());

	auto policy2 = profile_info.chrome_policies(1);
	EXPECT_EQ(kPolicyName2, policy2.name());
	EXPECT_EQ("true", policy2.value());
	EXPECT_EQ(em::Policy_PolicyLevel_LEVEL_RECOMMENDED, policy2.level());
	EXPECT_EQ(em::Policy_PolicyScope_SCOPE_MACHINE, policy2.scope());
	EXPECT_EQ(em::Policy_PolicySource_SOURCE_MERGED, policy2.source());
	EXPECT_NE("", policy2.error());
	}

	#if !defined(OS_ANDROID)
	TEST_F(PolicyInfoTest, ExtensionPolicy) {
	EXPECT_CALL(*policy_service(), GetPolicies(_)).Times(3);
	extensions::ExtensionRegistry* extension_registry =
	extensions::ExtensionRegistry::Get(profile());

	extension_registry->AddEnabled(
	extensions::ExtensionBuilder("extension_name")
	.SetID(kExtensionId1)
	.SetManifestPath({"storage", "managed_schema"}, "schema.json")
	.Build());
	extension_registry->AddEnabled(
	extensions::ExtensionBuilder("extension_name")
	.SetID(kExtensionId2)
	.SetManifestPath({"storage", "managed_schema"}, "schema.json")
	.Build());

	extension_policy_map()->Set(kPolicyName1, policy::POLICY_LEVEL_MANDATORY,
	policy::POLICY_SCOPE_MACHINE,
	policy::POLICY_SOURCE_PLATFORM, base::Value(3),
	nullptr);
	em::ChromeUserProfileInfo profile_info;
	auto client =
	std::make_unique<policy::ChromePolicyConversionsClient>(profile());
	AppendExtensionPolicyInfoIntoProfileReport(
	policy::DictionaryPolicyConversions(std::move(client))
	.EnableConvertTypes(false)
	.EnablePrettyPrint(false)
	.ToValue(),
	&profile_info);
	// The second extension is not in the report because it has no policy.
	EXPECT_EQ(1, profile_info.extension_policies_size());
	EXPECT_EQ(kExtensionId1, profile_info.extension_policies(0).extension_id());
	EXPECT_EQ(1, profile_info.extension_policies(0).policies_size());

	auto policy1 = profile_info.extension_policies(0).policies(0);
	EXPECT_EQ(kPolicyName1, policy1.name());
	EXPECT_EQ("3", policy1.value());
	EXPECT_EQ(em::Policy_PolicyLevel_LEVEL_MANDATORY, policy1.level());
	EXPECT_EQ(em::Policy_PolicyScope_SCOPE_MACHINE, policy1.scope());
	EXPECT_EQ(em::Policy_PolicySource_SOURCE_PLATFORM, policy1.source());
	EXPECT_NE(std::string(), policy1.error());
	}
	#endif // !defined(OS_ANDROID)

	TEST_F(PolicyInfoTest, MachineLevelUserCloudPolicyFetchTimestamp) {
	em::ChromeUserProfileInfo profile_info;
	#if !BUILDFLAG(IS_CHROMEOS_ASH)
	AppendMachineLevelUserCloudPolicyFetchTimestamp(
	&profile_info, g_browser_process->browser_policy_connector()
	->machine_level_user_cloud_policy_manager());
	#endif // !BUILDFLAG(IS_CHROMEOS_ASH)
	EXPECT_EQ(0, profile_info.policy_fetched_timestamps_size());
	}

	} // namespace enterprise_reporting