chromium/chrome/browser/enterprise/data_protection/data_protection_page_user_data_unittest.cc

// Copyright 2024 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/enterprise/data_protection/data_protection_page_user_data.h"

#include "chrome/test/base/testing_profile.h"
#include "content/public/browser/web_contents.h"
#include "content/public/test/test_renderer_host.h"

namespace enterprise_data_protection {

namespace {

// RenderViewHostTestHarness might not be necessary, but the Page constructor is
// private, so I looked at other tests and found that this is how it's created.
class DataProtectionPageUserDataTest
    : public content::RenderViewHostTestHarness { … };

std::unique_ptr<safe_browsing::RTLookupResponse> BuildDummyResponse(
    const char* watermark_text,
    bool allow_screenshot) { … }

}  // namespace

TEST_F(DataProtectionPageUserDataTest, TestCreateForPage) { … }

TEST_F(DataProtectionPageUserDataTest, NoRTURLLookupResponse) { … }

TEST_F(DataProtectionPageUserDataTest, NoThreatInfo) { … }

TEST_F(DataProtectionPageUserDataTest, UpdateRTLookupResponse) { … }

TEST_F(DataProtectionPageUserDataTest, UpdateScreenshotState) { … }

TEST_F(DataProtectionPageUserDataTest, MergedScreenshotState) { … }

}  // namespace enterprise_data_protection