lib/ui/scenic/tests/mocks.h

// Copyright 2017 The Fuchsia 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 GARNET_LIB_UI_SCENIC_TESTS_MOCKS_H_
#define GARNET_LIB_UI_SCENIC_TESTS_MOCKS_H_

#include "garnet/lib/ui/scenic/displays/display_manager.h"
#include "garnet/lib/ui/scenic/engine/engine.h"
#include "garnet/lib/ui/scenic/engine/session.h"
#include "garnet/lib/ui/scenic/engine/session_handler.h"
#include "lib/escher/flib/release_fence_signaller.h"

namespace scene_manager {
namespace test {

class SessionForTest : public Session {
 public:
  SessionForTest(SessionId id,
                 Engine* engine,
                 scene_manager::EventReporter* event_reporter,
                 mz::ErrorReporter* error_reporter);

  virtual void TearDown() override;
};

class SessionHandlerForTest : public SessionHandler {
 public:
  SessionHandlerForTest(mz::CommandDispatcherContext context,
                        Engine* engine,
                        SessionId session_id,
                        mz::EventReporter* event_reporter,
                        mz::ErrorReporter* error_reporter);

  // scenic::Session interface methods.
  void Enqueue(::f1dl::Array<ui_mozart::CommandPtr> ops) override;
  void Present(uint64_t presentation_time,
               ::f1dl::Array<zx::event> acquire_fences,
               ::f1dl::Array<zx::event> release_fences,
               const ui_mozart::Session::PresentCallback& callback) override;

  // Return the number of Enqueue()/Present()/Connect() messages that have
  // been processed.
  uint32_t enqueue_count() const { return enqueue_count_; }
  uint32_t present_count() const { return present_count_; }

 private:
  std::atomic<uint32_t> enqueue_count_;
  std::atomic<uint32_t> present_count_;
};

class ReleaseFenceSignallerForTest : public escher::ReleaseFenceSignaller {
 public:
  ReleaseFenceSignallerForTest(
      escher::impl::CommandBufferSequencer* command_buffer_sequencer);

  void AddCPUReleaseFence(zx::event fence) override;

  uint32_t num_calls_to_add_cpu_release_fence() {
    return num_calls_to_add_cpu_release_fence_;
  }

 private:
  uint32_t num_calls_to_add_cpu_release_fence_ = 0;
};

class EngineForTest : public Engine {
 public:
  EngineForTest(DisplayManager* display_manager,
                std::unique_ptr<escher::ReleaseFenceSignaller> r,
                escher::Escher* escher = nullptr);

 private:
  std::unique_ptr<SessionHandler> CreateSessionHandler(
      mz::CommandDispatcherContext context,
      SessionId id,
      mz::EventReporter* event_reporter,
      mz::ErrorReporter* error_reporter) override;
};

}  // namespace test
}  // namespace scene_manager

#endif  // GARNET_LIB_UI_SCENIC_TESTS_MOCKS_H_