components/exo/wayland/test/wayland_server_test_base.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 COMPONENTS_EXO_WAYLAND_TEST_WAYLAND_SERVER_TEST_BASE_H_
#define COMPONENTS_EXO_WAYLAND_TEST_WAYLAND_SERVER_TEST_BASE_H_

#include <memory>

#include "base/files/scoped_temp_dir.h"
#include "components/exo/display.h"
#include "components/exo/test/exo_test_base.h"

namespace exo {
class SecurityDelegate;
class Display;

namespace wayland {
class Server;

namespace test {

// Use ExoTestBase because Server starts to depends on ash::Shell.
using TestBase = exo::test::ExoTestBase;

// Base class for tests that create an exo's wayland server.
class WaylandServerTestBase : public TestBase {
 public:
  static std::string GetUniqueSocketName();

  WaylandServerTestBase();

  // Constructs a WaylandServerTestBase with |traits| being forwarded to its
  // TaskEnvironment. See the corresponding |AshTestBase| constructor.
  template <typename... TaskEnvironmentTraits>
  explicit WaylandServerTestBase(TaskEnvironmentTraits&&... traits)
      : TestBase(std::forward<TaskEnvironmentTraits>(traits)...) {}

  WaylandServerTestBase(const WaylandServerTestBase&) = delete;
  WaylandServerTestBase& operator=(const WaylandServerTestBase&) = delete;
  ~WaylandServerTestBase() override;

  void SetUp() override;
  void TearDown() override;

  std::unique_ptr<Server> CreateServer(
      std::unique_ptr<SecurityDelegate> security_delegate);
  std::unique_ptr<Server> CreateServer();

 protected:
  std::unique_ptr<Display> display_;
  base::ScopedTempDir xdg_temp_dir_;
};

}  // namespace test
}  // namespace wayland
}  // namespace exo

#endif  // COMPONENTS_EXO_WAYLAND_TEST_WAYLAND_SERVER_TEST_BASE_H_