offload/unittests/OffloadAPI/init/olInit.cpp

//===------- Offload API tests - olInit -----------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

// NOTE: For this test suite, the implicit olInit/olShutDown doesn't happen, so
// tests have to do it themselves

#include "../common/Fixtures.hpp"
#include <OffloadAPI.h>
#include <gtest/gtest.h>

struct olInitTest : ::testing::Test {};

TEST_F(olInitTest, Success) {
  ASSERT_SUCCESS(olInit(nullptr));
  ASSERT_SUCCESS(olShutDown());
}

TEST_F(olInitTest, Uninitialized) {
  ASSERT_ERROR(OL_ERRC_UNINITIALIZED,
               olIterateDevices(
                   [](ol_device_handle_t, void *) { return false; }, nullptr));
}

TEST_F(olInitTest, RepeatedInit) {
  for (size_t I = 0; I < 10; I++) {
    ASSERT_SUCCESS(olInit(nullptr));
    ASSERT_SUCCESS(olShutDown());
  }
}

TEST_F(olInitTest, RepeatedInitWithDevices) {
  size_t Expected = 0;
  for (size_t I = 0; I < 3; I++) {
    ASSERT_SUCCESS(olInit(nullptr));
    size_t Count = 0;

    // Force the platform to be initialized.
    ASSERT_SUCCESS(olIterateDevices(
        [](ol_device_handle_t, void *Data) {
          ++*static_cast<size_t *>(Data);
          return true;
        },
        &Count));
    ASSERT_SUCCESS(olShutDown());
    if (I == 0)
      Expected = Count;
    ASSERT_EQ(Count, Expected);
  }
}

TEST_F(olInitTest, WithInitArgs) {
  ol_init_args_t Args = OL_INIT_ARGS_INIT;
  ol_platform_backend_t Backends[] = {OL_PLATFORM_BACKEND_HOST};
  Args.NumPlatforms = 1;
  Args.Platforms = Backends;
  ASSERT_SUCCESS(olInit(&Args));
  ASSERT_SUCCESS(olShutDown());
}

TEST_F(olInitTest, InvalidSize) {
  ol_init_args_t Args = OL_INIT_ARGS_INIT;
  Args.Size = 0;
  ASSERT_ERROR(OL_ERRC_INVALID_SIZE, olInit(&Args));
}