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

#include "src/graphics/display/lib/fake-display-stack/fake-sysmem-device-hierarchy.h"

#include <fidl/fuchsia.io/cpp/fidl.h>
#include <fidl/fuchsia.sysmem/cpp/fidl.h>
#include <fidl/fuchsia.sysmem2/cpp/fidl.h>
#include <lib/component/incoming/cpp/service.h>
#include <lib/device-watcher/cpp/device-watcher.h>
#include <lib/fdio/directory.h>
#include <zircon/syscalls/object.h>

#include <gtest/gtest.h>

#include "src/lib/testing/predicates/status.h"

namespace fake_display {

namespace {

TEST(FakeSysmemDeviceHierarchy, OpenAllocatorV2) {
  zx::result<std::unique_ptr<FakeSysmemDeviceHierarchy>> create_result =
      FakeSysmemDeviceHierarchy::Create();
  ASSERT_OK(create_result);

  std::unique_ptr<FakeSysmemDeviceHierarchy> fake_sysmem_device_hierarchy =
      std::move(create_result).value();

  zx::result<fidl::ClientEnd<fuchsia_sysmem2::Allocator>> allocator_v2_result =
      fake_sysmem_device_hierarchy->ConnectAllocator2();
  ASSERT_OK(allocator_v2_result);

  fidl::SyncClient allocator_v2(std::move(allocator_v2_result).value());
  ASSERT_TRUE(allocator_v2.is_valid());

  // Make FIDL calls to make sure the Allocator V2 client is correctly connected to the fake
  // sysmem service.
  auto [collection_client, collection_server] =
      fidl::Endpoints<fuchsia_sysmem2::BufferCollection>::Create();
  fit::result<fidl::OneWayStatus> allocate_collection_result =
      allocator_v2->AllocateNonSharedCollection(
          {{.collection_request = std::move(collection_server)}});
  ASSERT_TRUE(allocate_collection_result.is_ok())
      << allocate_collection_result.error_value().FormatDescription();

  fit::result<fidl::OneWayError> set_constraints_result =
      fidl::Call(collection_client)
          ->SetConstraints({{
              .constraints = fuchsia_sysmem2::BufferCollectionConstraints{{
                  .usage = fuchsia_sysmem2::BufferUsage{{
                      .cpu = fuchsia_sysmem2::kCpuUsageRead,
                  }},
                  .min_buffer_count = 1,
                  .buffer_memory_constraints = fuchsia_sysmem2::BufferMemoryConstraints{{
                      .min_size_bytes = 4096,
                      .cpu_domain_supported = true,
                  }},
              }},
          }});
  ASSERT_TRUE(set_constraints_result.is_ok())
      << set_constraints_result.error_value().FormatDescription();

  fidl::Result allocation_result = fidl::Call(collection_client)->WaitForAllBuffersAllocated();
  ASSERT_TRUE(allocation_result.is_ok()) << allocation_result.error_value().FormatDescription();

  fit::result<fidl::OneWayError> release_result = fidl::Call(collection_client)->Release();
  ASSERT_TRUE(release_result.is_ok()) << allocation_result.error_value().FormatDescription();
}

}  // namespace

}  // namespace fake_display