File: prevent_close_test_base.cc

package info (click to toggle)
chromium 120.0.6099.224-1~deb11u1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 6,112,112 kB
  • sloc: cpp: 32,907,025; ansic: 8,148,123; javascript: 3,679,536; python: 2,031,248; asm: 959,718; java: 804,675; xml: 617,256; sh: 111,417; objc: 100,835; perl: 88,443; cs: 53,032; makefile: 29,579; fortran: 24,137; php: 21,162; tcl: 21,147; sql: 20,809; ruby: 17,735; pascal: 12,864; yacc: 8,045; lisp: 3,388; lex: 1,323; ada: 727; awk: 329; jsp: 267; csh: 117; exp: 43; sed: 37
file content (90 lines) | stat: -rw-r--r-- 3,636 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ui/prevent_close_test_base.h"

#include "base/json/json_reader.h"
#include "base/test/scoped_feature_list.h"
#include "base/values.h"
#include "chrome/browser/policy/policy_test_utils.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/web_applications/test/web_app_browsertest_util.h"
#include "chrome/browser/web_applications/test/web_app_install_test_utils.h"
#include "chrome/browser/web_applications/web_app_id_constants.h"
#include "chrome/browser/web_applications/web_app_install_info.h"
#include "chrome/common/chrome_features.h"
#include "components/policy/core/browser/browser_policy_connector.h"
#include "components/policy/core/browser/browser_policy_connector_base.h"
#include "components/policy/core/common/mock_configuration_policy_provider.h"
#include "components/policy/core/common/policy_map.h"
#include "components/policy/policy_constants.h"
#include "components/webapps/common/web_app_id.h"
#include "content/public/test/browser_test.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "ui/base/window_open_disposition.h"
#include "url/gurl.h"

PreventCloseTestBase::PreventCloseTestBase() {
  scoped_feature_list_.InitWithFeatures(
      /*enabled_features=*/{features::kDesktopPWAsEnforceWebAppSettingsPolicy,
                            features::kDesktopPWAsPreventClose},
      /*disabled_features=*/{});
}

PreventCloseTestBase::~PreventCloseTestBase() = default;

void PreventCloseTestBase::SetUpInProcessBrowserTestFixture() {
  provider_.SetDefaultReturns(
      /*is_initialization_complete_return=*/true,
      /*is_first_policy_load_complete_return=*/true);
  policy::BrowserPolicyConnector::SetPolicyProviderForTesting(&provider_);
}

void PreventCloseTestBase::TearDownInProcessBrowserTestFixture() {
  ClearWebAppSettings();
  policy::PolicyTest::TearDownInProcessBrowserTestFixture();
}

void PreventCloseTestBase::SetWebAppSettings(base::StringPiece config) {
  policy::PolicyMap policies;
  SetPolicy(&policies, policy::key::kWebAppSettings,
            ReturnPolicyValueFromJson(config));
  provider_.UpdateChromePolicy(policies);
}

void PreventCloseTestBase::ClearWebAppSettings() {
  SetWebAppSettings(/*config=*/"[]");
}

void PreventCloseTestBase::InstallPWA(const GURL& app_url,
                                      const webapps::AppId& app_id) {
  auto web_app_info = std::make_unique<web_app::WebAppInstallInfo>();
  web_app_info->start_url = app_url;
  web_app_info->scope = app_url.GetWithoutFilename();
  webapps::AppId installed_app_id = web_app::test::InstallWebApp(
      browser()->profile(), std::move(web_app_info));
  EXPECT_EQ(app_id, installed_app_id);
}

Browser* PreventCloseTestBase::LaunchPWA(const webapps::AppId& app_id,
                                         bool launch_in_window) {
  return launch_in_window
             ? web_app::LaunchWebAppBrowserAndWait(
                   profile(), app_id, WindowOpenDisposition::NEW_WINDOW)
             : web_app::LaunchBrowserForWebAppInTab(profile(), app_id);
}

base::Value PreventCloseTestBase::ReturnPolicyValueFromJson(
    base::StringPiece policy) {
  auto result = base::JSONReader::ReadAndReturnValueWithError(
      policy, base::JSONParserOptions::JSON_ALLOW_TRAILING_COMMAS);
  DCHECK(result.has_value()) << result.error().message;
  DCHECK(result->is_list());
  return std::move(*result);
}

Profile* PreventCloseTestBase::profile() {
  return browser()->profile();
}