src/graphics/lib/compute/tests/common/vk_app_state_unittest.cc

// Copyright 2019 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 "vk_app_state.h"

#include <gtest/gtest.h>

#include "vk_surface.h"

TEST(vkAppStateTest, DefautInit)
{
  vk_app_state_t app = {};

  const vk_app_state_config_t config = {
    // Empty configuration.
  };

  ASSERT_TRUE(vk_app_state_init(&app, &config)) << "Could not initialize vk_app_state_t instance";

  ASSERT_TRUE(app.instance);
  ASSERT_TRUE(app.d);
  ASSERT_TRUE(app.pd);

  ASSERT_FALSE(app.ac);  // no allocator by default.
  ASSERT_FALSE(app.pc);  // no pipeline cache by default.

  ASSERT_NE(app.qfi, UINT32_MAX);  // graphics queue is used by default
  ASSERT_EQ(app.compute_qfi, UINT32_MAX);

  ASSERT_FALSE(app.has_debug_report);
  ASSERT_FALSE(app.has_amd_statistics);
  ASSERT_FALSE(app.has_subgroup_size_control);

  vk_app_state_destroy(&app);
}

TEST(vkAppStateTest, EnableValidation)
{
  vk_app_state_t app = {};

  const vk_app_state_config_t config = {
    .enable_validation = true,
  };

  ASSERT_TRUE(vk_app_state_init(&app, &config)) << "Could not initialize vk_app_state_t instance";

  ASSERT_TRUE(app.instance);
  ASSERT_TRUE(app.d);
  ASSERT_TRUE(app.pd);

  ASSERT_FALSE(app.ac);  // no allocator by default.
  ASSERT_FALSE(app.pc);  // no pipeline cache by default.

  ASSERT_NE(app.qfi, UINT32_MAX);  // graphics queue is used by default
  ASSERT_EQ(app.compute_qfi, UINT32_MAX);

  ASSERT_FALSE(app.has_debug_report);
  ASSERT_FALSE(app.has_amd_statistics);
  ASSERT_FALSE(app.has_subgroup_size_control);

  vk_app_state_destroy(&app);
}

TEST(vkAppStateTest, EnableDebug)
{
  vk_app_state_t app = {};

  const vk_app_state_config_t config = {
    .enable_validation   = true,  // required by debug_report
    .enable_debug_report = true,
  };

  ASSERT_TRUE(vk_app_state_init(&app, &config)) << "Could not initialize vk_app_state_t instance";

  ASSERT_TRUE(app.instance);
  ASSERT_TRUE(app.d);
  ASSERT_TRUE(app.pd);

  ASSERT_FALSE(app.ac);  // no allocator by default.
  ASSERT_FALSE(app.pc);  // no pipeline cache by default.

  ASSERT_NE(app.qfi, UINT32_MAX);  // graphics queue is used by default
  ASSERT_EQ(app.compute_qfi, UINT32_MAX);

  ASSERT_TRUE(app.has_debug_report);  // debug report enabled!
  ASSERT_FALSE(app.has_amd_statistics);
  ASSERT_FALSE(app.has_subgroup_size_control);

  vk_app_state_destroy(&app);
}

TEST(vkAppStateTest, EnablePipelineCache)
{
  vk_app_state_t app = {};

  const vk_app_state_config_t config = {
    .enable_pipeline_cache = true,
  };

  ASSERT_TRUE(vk_app_state_init(&app, &config)) << "Could not initialize vk_app_state_t instance";

  ASSERT_TRUE(app.instance);
  ASSERT_TRUE(app.d);
  ASSERT_TRUE(app.pd);

  ASSERT_FALSE(app.ac);  // no allocator by default.
  ASSERT_TRUE(app.pc);   // pipeline cache should be created!

  ASSERT_NE(app.qfi, UINT32_MAX);  // graphics queue is used by default
  ASSERT_EQ(app.compute_qfi, UINT32_MAX);

  ASSERT_FALSE(app.has_debug_report);
  ASSERT_FALSE(app.has_amd_statistics);
  ASSERT_FALSE(app.has_subgroup_size_control);

  vk_app_state_destroy(&app);
}

TEST(vkAppStateTest, EnableComputeQueueOnly)
{
  vk_app_state_t app = {};

  const vk_app_state_config_t config = {
    .device_config = {
      .required_queues = VK_QUEUE_COMPUTE_BIT,
    },
  };

  ASSERT_TRUE(vk_app_state_init(&app, &config)) << "Could not initialize vk_app_state_t instance";

  ASSERT_TRUE(app.instance);
  ASSERT_TRUE(app.d);
  ASSERT_TRUE(app.pd);

  ASSERT_FALSE(app.ac);  // no allocator by default.
  ASSERT_FALSE(app.pc);  // no pipeline cache by default.

  ASSERT_EQ(app.qfi, UINT32_MAX);          // graphics queue should not be selected.
  ASSERT_NE(app.compute_qfi, UINT32_MAX);  // compute queue should be selected.

  ASSERT_FALSE(app.has_debug_report);
  ASSERT_FALSE(app.has_amd_statistics);
  ASSERT_FALSE(app.has_subgroup_size_control);
}

TEST(vkAppStateTest, EnableComputeQueueAndSwapchain)
{
  vk_app_state_t app = {};

  const vk_app_state_config_t config = {
    .device_config = {
      .required_queues = VK_QUEUE_COMPUTE_BIT,
    },
    .require_swapchain = true,
  };

  ASSERT_TRUE(vk_app_state_init(&app, &config)) << "Could not initialize vk_app_state_t instance";

  ASSERT_TRUE(app.instance);
  ASSERT_TRUE(app.d);
  ASSERT_TRUE(app.pd);

  ASSERT_FALSE(app.ac);  // no allocator by default.
  ASSERT_FALSE(app.pc);  // no pipeline cache by default.

  ASSERT_NE(app.qfi, UINT32_MAX);          // graphics queue should be selected.
  ASSERT_NE(app.compute_qfi, UINT32_MAX);  // compute queue should be selected.

  ASSERT_FALSE(app.has_debug_report);
  ASSERT_FALSE(app.has_amd_statistics);
  ASSERT_FALSE(app.has_subgroup_size_control);
}

TEST(vkAppStateTest, RequireSwapchain)
{
  vk_app_state_t app = {};

  const vk_app_state_config_t config = {
    .require_swapchain = true,
  };

  ASSERT_TRUE(vk_app_state_init(&app, &config)) << "Could not initialize vk_app_state_t instance";

  ASSERT_TRUE(app.instance);
  ASSERT_TRUE(app.d);
  ASSERT_TRUE(app.pd);

  ASSERT_FALSE(app.ac);  // no allocator by default.
  ASSERT_FALSE(app.pc);  // no pipeline cache by default.

  ASSERT_NE(app.qfi, UINT32_MAX);  // graphics queue is used by default
  ASSERT_EQ(app.compute_qfi, UINT32_MAX);

  ASSERT_FALSE(app.has_debug_report);
  ASSERT_FALSE(app.has_amd_statistics);
  ASSERT_FALSE(app.has_subgroup_size_control);

  vk_app_state_destroy(&app);
}

TEST(vkAppStateTest, RequireSwapchainDisablePresent)
{
  vk_app_state_t app = {};

  const vk_app_state_config_t config = {
    .require_swapchain         = true,
    .disable_swapchain_present = true,
  };

  ASSERT_TRUE(vk_app_state_init(&app, &config)) << "Could not initialize vk_app_state_t instance";

  vk_app_state_destroy(&app);
}

TEST(vkAppStateTest, GetQueueFamiliesCombined)
{
  vk_app_state_t app = {};

  const vk_app_state_config_t config = {
    .device_config = {
      .required_combined_queues = VK_QUEUE_GRAPHICS_BIT | VK_QUEUE_COMPUTE_BIT,
    },
  };

  ASSERT_TRUE(vk_app_state_init(&app, &config));
  ASSERT_NE(app.qfi, UINT32_MAX);
  ASSERT_NE(app.compute_qfi, UINT32_MAX);
  ASSERT_EQ(app.qfi, app.compute_qfi);

  vk_queue_families_t families = vk_app_state_get_queue_families(&app);
  ASSERT_EQ(families.count, 1u);
  ASSERT_EQ(families.indices[0], app.qfi);
}

TEST(vkAppStateTest, GetQueueFamiliesSingle)
{
  vk_app_state_t app = {};

  const vk_app_state_config_t config = {
    .device_config = {
      .required_queues = VK_QUEUE_GRAPHICS_BIT | VK_QUEUE_COMPUTE_BIT,
    },
  };

  ASSERT_TRUE(vk_app_state_init(&app, &config));
  ASSERT_NE(app.qfi, UINT32_MAX);
  ASSERT_NE(app.compute_qfi, UINT32_MAX);

  vk_queue_families_t families = vk_app_state_get_queue_families(&app);
  if (app.qfi == app.compute_qfi)
    {
      ASSERT_EQ(families.count, 1u);
      ASSERT_EQ(families.indices[0], app.qfi);
    }
  else
    {
      ASSERT_EQ(families.count, 2u);
      ASSERT_EQ(families.indices[0], app.qfi);
      ASSERT_EQ(families.indices[1], app.compute_qfi);
    }
}