/*
 * Copyright (C) 2019 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#define LOG_TAG "HwlBufferAllocatorTests"
#include <log/log.h>

#include <gtest/gtest.h>
#include <hwl_buffer_allocator.h>
#include "mock_buffer_allocator_hwl.h"

namespace android {
namespace google_camera_hal {

static const uint32_t kBufferWidth = 4032;
static const uint32_t kBufferHeight = 3024;
static const uint32_t kMaxBufferDepth = 10;

// Test HwlBufferAllocator Create.
TEST(HwlBufferAllocatorTests, Create) {
  auto mock_allocator_hwl = MockBufferAllocatorHwl::Create();
  ASSERT_NE(mock_allocator_hwl, nullptr);

  auto allocator = HwlBufferAllocator::Create(mock_allocator_hwl.get());
  ASSERT_NE(allocator, nullptr) << "Create HwlBufferAllocator failed.";
}

// Test HwlBufferAllocator AllocateFreeBuffers.
TEST(HwlBufferAllocatorTests, AllocateFreeBuffers) {
  auto mock_allocator_hwl = MockBufferAllocatorHwl::Create();
  ASSERT_NE(mock_allocator_hwl, nullptr);

  auto allocator = HwlBufferAllocator::Create(mock_allocator_hwl.get());
  ASSERT_NE(allocator, nullptr) << "Create HwlBufferAllocator failed.";

  HalBufferDescriptor buffer_descriptor = {};
  buffer_descriptor.width = kBufferWidth;
  buffer_descriptor.height = kBufferHeight;
  buffer_descriptor.format = HAL_PIXEL_FORMAT_RAW10;
  buffer_descriptor.producer_flags = GRALLOC1_PRODUCER_USAGE_CAMERA;
  buffer_descriptor.consumer_flags = GRALLOC1_CONSUMER_USAGE_CAMERA;
  buffer_descriptor.immediate_num_buffers = kMaxBufferDepth;
  buffer_descriptor.max_num_buffers = kMaxBufferDepth;

  std::vector<buffer_handle_t> buffers_;
  status_t res = allocator->AllocateBuffers(buffer_descriptor, &buffers_);
  ASSERT_EQ(res, OK) << "AllocateBuffers failed: " << strerror(res);
  ASSERT_EQ(buffers_.size(), (uint32_t)kMaxBufferDepth)
      << "AllocateBuffers failed with wrong buffer number " << buffers_.size();

  allocator->FreeBuffers(&buffers_);
  ASSERT_EQ(buffers_.size(), (uint32_t)0)
      << "AllocateBuffers failed with wrong buffer number " << buffers_.size();
}
}  // namespace google_camera_hal
}  // namespace android