chrome/browser/headless/headless_mode_browsertest.h

// Copyright 2022 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_HEADLESS_HEADLESS_MODE_BROWSERTEST_H_
#define CHROME_BROWSER_HEADLESS_HEADLESS_MODE_BROWSERTEST_H_

#include <memory>

#include "base/command_line.h"
#include "base/files/file_path.h"
#include "base/files/scoped_temp_dir.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "content/public/test/browser_test.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/gfx/geometry/size.h"

namespace content {
class WebContents;
}

namespace headless {
class HeadlessModeHandle;

class HeadlessModeBrowserTest : public InProcessBrowserTest {
 public:
  HeadlessModeBrowserTest();

  HeadlessModeBrowserTest(const HeadlessModeBrowserTest&) = delete;
  HeadlessModeBrowserTest& operator=(const HeadlessModeBrowserTest&) = delete;

  ~HeadlessModeBrowserTest() override;

  void SetUpCommandLine(base::CommandLine* command_line) override;
  void SetUpOnMainThread() override;

 protected:
  virtual bool IsIncognito();

  bool headful_mode() const { return headful_mode_; }

  void AppendHeadlessCommandLineSwitches(base::CommandLine* command_line);

  content::WebContents* GetActiveWebContents();

 private:
  bool headful_mode_ = false;
  std::unique_ptr<HeadlessModeHandle> headless_mode_handle_;
};

enum StartWindowMode {
  kStartWindowNormal,
  kStartWindowMaximized,
  kStartWindowFullscreen,
};

class HeadlessModeBrowserTestWithStartWindowMode
    : public HeadlessModeBrowserTest,
      public testing::WithParamInterface<StartWindowMode> {
 public:
  HeadlessModeBrowserTestWithStartWindowMode() = default;
  ~HeadlessModeBrowserTestWithStartWindowMode() override = default;

  void SetUpCommandLine(base::CommandLine* command_line) override;

  StartWindowMode start_window_mode() const { return GetParam(); }
};

// Toggles browser fullscreen mode synchronously.
void ToggleFullscreenModeSync(Browser* browser);

}  // namespace headless

#endif  // CHROME_BROWSER_HEADLESS_HEADLESS_MODE_BROWSERTEST_H_