// Copyright (C) 2023 Intel Corporation
// Part of the Unified-Runtime Project, under the Apache License v2.0 with LLVM
// Exceptions. See LICENSE.TXT
//
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

#include "uur/known_failure.h"
#include <uur/fixtures.h>

struct urProgramGetBuildInfoTest : uur::urProgramTest {
  void SetUp() override {
    UUR_RETURN_ON_FATAL_FAILURE(urProgramTest::SetUp());
    ASSERT_SUCCESS(urProgramBuild(this->context, program, nullptr));
  }
};
UUR_INSTANTIATE_DEVICE_TEST_SUITE(urProgramGetBuildInfoTest);

TEST_P(urProgramGetBuildInfoTest, SuccessStatus) {
  UUR_KNOWN_FAILURE_ON(uur::LevelZero{}, uur::LevelZeroV2{});

  size_t property_size = 0;
  const ur_program_build_info_t property_name = UR_PROGRAM_BUILD_INFO_STATUS;

  ASSERT_SUCCESS_OR_OPTIONAL_QUERY(
      urProgramGetBuildInfo(program, device, property_name, 0, nullptr,
                            &property_size),
      property_name);
  ASSERT_EQ(sizeof(ur_program_build_status_t), property_size);

  ur_program_build_status_t property_value =
      UR_PROGRAM_BUILD_STATUS_FORCE_UINT32;
  ASSERT_SUCCESS(urProgramGetBuildInfo(
      program, device, property_name, property_size, &property_value, nullptr));

  ASSERT_GE(property_value, UR_PROGRAM_BUILD_STATUS_NONE);
  ASSERT_LE(property_value, UR_PROGRAM_BUILD_STATUS_IN_PROGRESS);
}

TEST_P(urProgramGetBuildInfoTest, SuccessOptions) {
  size_t property_size = 0;
  const ur_program_build_info_t property_name = UR_PROGRAM_BUILD_INFO_OPTIONS;

  ASSERT_SUCCESS_OR_OPTIONAL_QUERY(
      urProgramGetBuildInfo(program, device, property_name, 0, nullptr,
                            &property_size),
      property_name);
  ASSERT_GT(property_size, 0);

  std::vector<char> property_value(property_size, '\0');
  ASSERT_SUCCESS(urProgramGetBuildInfo(program, device, property_name,
                                       property_size, property_value.data(),
                                       nullptr));

  ASSERT_TRUE(uur::stringPropertyIsValid(property_value.data(), property_size));
}

TEST_P(urProgramGetBuildInfoTest, SuccessLog) {
  UUR_KNOWN_FAILURE_ON(uur::CUDA{});

  size_t property_size = 0;
  const ur_program_build_info_t property_name = UR_PROGRAM_BUILD_INFO_LOG;

  ASSERT_SUCCESS_OR_OPTIONAL_QUERY(
      urProgramGetBuildInfo(program, device, property_name, 0, nullptr,
                            &property_size),
      property_name);
  ASSERT_GT(property_size, 0);

  std::vector<char> property_value(property_size, '\0');
  ASSERT_SUCCESS(urProgramGetBuildInfo(program, device, property_name,
                                       property_size, property_value.data(),
                                       nullptr));

  ASSERT_TRUE(uur::stringPropertyIsValid(property_value.data(), property_size));
}

TEST_P(urProgramGetBuildInfoTest, SuccessBinaryType) {
  size_t property_size = 0;
  const ur_program_build_info_t property_name =
      UR_PROGRAM_BUILD_INFO_BINARY_TYPE;

  ASSERT_SUCCESS_OR_OPTIONAL_QUERY(
      urProgramGetBuildInfo(program, device, property_name, 0, nullptr,
                            &property_size),
      property_name);
  ASSERT_EQ(sizeof(ur_program_binary_type_t), property_size);

  ur_program_binary_type_t property_value = UR_PROGRAM_BINARY_TYPE_FORCE_UINT32;
  ASSERT_SUCCESS(urProgramGetBuildInfo(
      program, device, property_name, property_size, &property_value, nullptr));

  ASSERT_GE(property_value, UR_PROGRAM_BINARY_TYPE_NONE);
  ASSERT_LE(property_value, UR_PROGRAM_BINARY_TYPE_EXECUTABLE);
}

TEST_P(urProgramGetBuildInfoTest, InvalidNullHandleProgram) {
  ur_program_build_status_t property_value = UR_PROGRAM_BUILD_STATUS_ERROR;
  ASSERT_EQ_RESULT(
      urProgramGetBuildInfo(nullptr, device, UR_PROGRAM_BUILD_INFO_STATUS,
                            sizeof(property_value), &property_value, nullptr),
      UR_RESULT_ERROR_INVALID_NULL_HANDLE);
}

TEST_P(urProgramGetBuildInfoTest, InvalidNullHandleDevice) {
  ur_program_build_status_t property_value = UR_PROGRAM_BUILD_STATUS_ERROR;
  ASSERT_EQ_RESULT(
      urProgramGetBuildInfo(program, nullptr, UR_PROGRAM_BUILD_INFO_STATUS,
                            sizeof(property_value), &property_value, nullptr),
      UR_RESULT_ERROR_INVALID_NULL_HANDLE);
}

TEST_P(urProgramGetBuildInfoTest, InvalidEnumeration) {
  size_t property_size = UR_PROGRAM_BUILD_STATUS_ERROR;
  ASSERT_EQ_RESULT(UR_RESULT_ERROR_INVALID_ENUMERATION,
                   urProgramGetBuildInfo(program, device,
                                         UR_PROGRAM_BUILD_INFO_FORCE_UINT32, 0,
                                         nullptr, &property_size));
}