blob: 7f2201ca46ac19f762538000d4aafe7f7311e470 [file] [log] [blame]
// 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 "chrome/browser/ui/webui/managed_ui_handler.h"
#include "base/test/scoped_feature_list.h"
#include "base/token.h"
#include "base/values.h"
#include "chrome/browser/policy/profile_policy_connector.h"
#include "chrome/browser/policy/profile_policy_connector_factory.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/common/chrome_features.h"
#include "chrome/test/base/testing_profile.h"
#include "components/policy/core/common/mock_configuration_policy_provider.h"
#include "components/policy/core/common/policy_service_impl.h"
#include "content/public/browser/web_contents.h"
#include "content/public/browser/web_ui_data_source.h"
#include "content/public/test/test_browser_thread_bundle.h"
#include "content/public/test/test_web_ui.h"
#include "content/public/test/test_web_ui_data_source.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
#if !defined(OS_CHROMEOS)
#include "components/policy/core/browser/browser_policy_connector_base.h"
#endif // !defined(OS_CHROMEOS)
class TestManagedUIHandler : public ManagedUIHandler {
public:
using ManagedUIHandler::InitializeInternal;
};
class ManagedUIHandlerTest : public testing::Test {
public:
ManagedUIHandlerTest()
: source_(content::TestWebUIDataSource::Create(
base::Token::CreateRandom().ToString())) {
features_.InitAndEnableFeature(features::kShowManagedUi);
// Create a TestingProfile that uses our MockConfigurationPolicyProvider.
policy_provider()->Init();
policy::PolicyServiceImpl::Providers providers = {policy_provider()};
TestingProfile::Builder builder;
builder.SetPolicyService(
std::make_unique<policy::PolicyServiceImpl>(std::move(providers)));
profile_ = builder.Build();
// We use a random source_name here as calling Add() can replace existing
// sources with the same name (which might destroy the memory addressed by
// |source_->GetWebUIDataSource()|.
content::WebUIDataSource::Add(profile(), source_->GetWebUIDataSource());
}
void TearDown() override { policy_provider()->Shutdown(); }
TestingProfile* profile() { return profile_.get(); }
policy::MockConfigurationPolicyProvider* policy_provider() {
return &policy_provider_;
}
policy::ProfilePolicyConnector* profile_policy_connector() {
return policy::ProfilePolicyConnectorFactory::GetForBrowserContext(
profile());
}
void InitializeHandler() {
TestManagedUIHandler::InitializeInternal(
&web_ui_, source_->GetWebUIDataSource(), profile());
web_ui_.HandleReceivedMessage("observeManagedUI", /*args=*/nullptr);
}
bool IsSourceManaged() {
const auto* local_strings = source_->GetLocalizedStrings();
const auto* managed =
local_strings->FindKeyOfType("isManaged", base::Value::Type::BOOLEAN);
if (managed == nullptr) {
ADD_FAILURE();
return false;
}
return managed->GetBool();
}
private:
content::TestBrowserThreadBundle bundle_;
base::test::ScopedFeatureList features_;
testing::NiceMock<policy::MockConfigurationPolicyProvider> policy_provider_;
std::unique_ptr<TestingProfile> profile_;
content::TestWebUI web_ui_;
std::unique_ptr<content::TestWebUIDataSource> source_;
};
TEST_F(ManagedUIHandlerTest, ManagedUIDisabledByDefault) {
InitializeHandler();
EXPECT_FALSE(IsSourceManaged());
}
TEST_F(ManagedUIHandlerTest, ManagedUIEnabledWhenManaged) {
profile_policy_connector()->OverrideIsManagedForTesting(true);
InitializeHandler();
EXPECT_TRUE(IsSourceManaged());
}
TEST_F(ManagedUIHandlerTest, ManagedUIBecomesEnabledByProfile) {
InitializeHandler();
EXPECT_FALSE(IsSourceManaged());
// Make ProfilePolicyConnector::IsManaged() return true.
profile_policy_connector()->OverrideIsManagedForTesting(true);
policy::PolicyMap non_empty_map;
non_empty_map.Set("FakePolicyName", policy::POLICY_LEVEL_MANDATORY,
policy::POLICY_SCOPE_USER, policy::POLICY_SOURCE_CLOUD,
std::make_unique<base::Value>("fake"), nullptr);
policy_provider()->UpdateChromePolicy(non_empty_map);
// Source should auto-update.
EXPECT_TRUE(IsSourceManaged());
}
#if defined(OS_CHROMEOS)
TEST_F(ManagedUIHandlerTest, ManagedUIDisabledForChildAccount) {
profile_policy_connector()->OverrideIsManagedForTesting(true);
profile()->SetSupervisedUserId("supervised");
InitializeHandler();
EXPECT_FALSE(IsSourceManaged());
}
#endif