chrome/browser/chromeos/policy/dlp/mock_dlp_content_manager.h

// Copyright 2020 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.

#ifndef CHROME_BROWSER_CHROMEOS_POLICY_DLP_MOCK_DLP_CONTENT_MANAGER_H_
#define CHROME_BROWSER_CHROMEOS_POLICY_DLP_MOCK_DLP_CONTENT_MANAGER_H_

#include "chrome/browser/chromeos/policy/dlp/dlp_content_manager.h"
#include "chrome/browser/ui/ash/chrome_screenshot_grabber.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "url/gurl.h"

namespace policy {

class MockDlpContentManager : public DlpContentManager {
 public:
  MockDlpContentManager();
  ~MockDlpContentManager() override;

  MOCK_METHOD2(OnConfidentialityChanged,
               void(content::WebContents*, const DlpContentRestrictionSet&));
  MOCK_METHOD1(OnWebContentsDestroyed, void(content::WebContents*));
  MOCK_CONST_METHOD1(GetRestrictionSetForURL,
                     DlpContentRestrictionSet(const GURL&));
  MOCK_METHOD1(OnVisibilityChanged, void(content::WebContents*));
  MOCK_CONST_METHOD1(IsScreenshotRestricted, bool(const ScreenshotArea& area));
  MOCK_CONST_METHOD1(IsScreenCaptureRestricted,
                     bool(const content::DesktopMediaID& media_id));

 protected:
  void Init() override;
};

}  // namespace policy

#endif  // CHROME_BROWSER_CHROMEOS_POLICY_DLP_MOCK_DLP_CONTENT_MANAGER_H_