// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROME_BROWSER_POLICY_SAFE_SEARCH_POLICY_TEST_H_
#define CHROME_BROWSER_POLICY_SAFE_SEARCH_POLICY_TEST_H_

#include <optional>
#include <string>

#include "base/test/scoped_feature_list.h"
#include "base/values.h"
#include "chrome/browser/policy/policy_test_utils.h"
#include "chrome/test/base/chrome_test_utils.h"
#include "url/gurl.h"

class BrowserWindowInterface;

namespace policy {

class SafeSearchPolicyTest : public PolicyTest {
 protected:
  SafeSearchPolicyTest();
  ~SafeSearchPolicyTest() override;

  void ApplySafeSearchPolicy(std::optional<base::Value> legacy_safe_search,
                             std::optional<base::Value> google_safe_search,
                             std::optional<base::Value> legacy_youtube,
                             std::optional<base::Value> youtube_restrict);

  static GURL GetExpectedSearchURL(bool expect_safe_search);

  static void CheckSafeSearch(BrowserWindowInterface* browser,
                              bool expect_safe_search,
                              const std::string& url = "http://google.com/");

 private:
  base::test::ScopedFeatureList feature_list_;
};

}  // namespace policy

#endif  // CHROME_BROWSER_POLICY_SAFE_SEARCH_POLICY_TEST_H_