// 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 "base/command_line.h"
#include "base/test/task_environment.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/display/types/display_constants.h"
#include "ui/display/win/screen_win.h"
#include "ui/display/win/test/virtual_display_util_win.h"

// ScreenWin has a protected ctor so use a derived class to instantiate it.
class VirtualDisplayScreenWin : public display::win::ScreenWin {
 public:
  VirtualDisplayScreenWin() = default;

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

  ~VirtualDisplayScreenWin() override = default;
};

// This test suite requires the host to have a special driver installed.
// If the driver is not detected, this test will skip.
// See: //docs/ui/display/multiscreen_testing.md
class VirtualDisplayUtilWinInteractiveUitest : public testing::Test {
 public:
  VirtualDisplayUtilWinInteractiveUitest(
      const VirtualDisplayUtilWinInteractiveUitest&) = delete;
  VirtualDisplayUtilWinInteractiveUitest& operator=(
      const VirtualDisplayUtilWinInteractiveUitest&) = delete;

 protected:
  VirtualDisplayUtilWinInteractiveUitest()
      : virtual_display_util_win_(&screen_) {}

  void SetUp() override {
    if (!virtual_display_util_win_.IsAPIAvailable()) {
      GTEST_SKIP() << "Host does not support virtual displays (driver is not "
                      "detected).";
    }
    display::Screen::SetScreenInstance(&screen_);
    testing::Test::SetUp();
  }
  display::Screen* screen() { return &screen_; }
  base::test::TaskEnvironment task_environment_{
      base::test::TaskEnvironment::MainThreadType::UI};
  VirtualDisplayScreenWin screen_;
  display::test::VirtualDisplayUtilWin virtual_display_util_win_;
};

// TODO(crbug.com/371121282): Re-enable the test.
// TODO(crbug.com/365126887): Re-enable the test.
TEST_F(VirtualDisplayUtilWinInteractiveUitest, DISABLED_IsAPIAvailable) {
  EXPECT_TRUE(virtual_display_util_win_.IsAPIAvailable());
}

// TODO(crbug.com/371121282): Re-enable the test.
// TODO(crbug.com/365126887): Re-enable the test.
TEST_F(VirtualDisplayUtilWinInteractiveUitest, DISABLED_AddRemove) {
  int64_t display_id[3];
  int initial_display_count = screen()->GetNumDisplays();
  display_id[0] = virtual_display_util_win_.AddDisplay(
      display::test::VirtualDisplayUtilWin::k1920x1080);
  EXPECT_NE(display_id[0], display::kInvalidDisplayId);
  display::Display d;
  EXPECT_EQ(screen()->GetNumDisplays(), initial_display_count + 1);
  EXPECT_TRUE(screen()->GetDisplayWithDisplayId(display_id[0], &d));
  EXPECT_EQ(d.size(), gfx::Size(1920, 1080));

  display_id[1] = virtual_display_util_win_.AddDisplay(
      display::test::VirtualDisplayUtilWin::k1024x768);
  EXPECT_NE(display_id[1], display::kInvalidDisplayId);
  EXPECT_EQ(screen()->GetNumDisplays(), initial_display_count + 2);
  EXPECT_TRUE(screen()->GetDisplayWithDisplayId(display_id[1], &d));
  EXPECT_EQ(d.size(), gfx::Size(1024, 768));

  display_id[2] = virtual_display_util_win_.AddDisplay(
      display::test::VirtualDisplayUtilWin::k1920x1080);
  EXPECT_NE(display_id[2], display::kInvalidDisplayId);
  EXPECT_EQ(screen()->GetNumDisplays(), initial_display_count + 3);
  EXPECT_TRUE(screen()->GetDisplayWithDisplayId(display_id[2], &d));
  EXPECT_EQ(d.size(), gfx::Size(1920, 1080));

  virtual_display_util_win_.RemoveDisplay(display_id[1]);
  EXPECT_EQ(screen()->GetNumDisplays(), initial_display_count+2);
  // Only virtual display 2 should no longer exist.
  EXPECT_TRUE(screen()->GetDisplayWithDisplayId(display_id[0], &d));
  EXPECT_EQ(d.size(), gfx::Size(1920, 1080));
  EXPECT_FALSE(screen()->GetDisplayWithDisplayId(display_id[1], &d));
  EXPECT_TRUE(screen()->GetDisplayWithDisplayId(display_id[2], &d));
  EXPECT_EQ(d.size(), gfx::Size(1920, 1080));

  virtual_display_util_win_.ResetDisplays();
  EXPECT_FALSE(screen()->GetDisplayWithDisplayId(display_id[0], &d));
  EXPECT_FALSE(screen()->GetDisplayWithDisplayId(display_id[1], &d));
  EXPECT_FALSE(screen()->GetDisplayWithDisplayId(display_id[2], &d));
  EXPECT_EQ(screen()->GetNumDisplays(), initial_display_count);
}