TheRock/rocm-libraries/projects/hipthreads/test/std/thread/thread.condition/thread.condition.condvarany/helpers.h

//===----------------------------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//

#ifndef TEST_STD_THREAD_CONDITION_CONDVARANY_HELPERS_H
#define TEST_STD_THREAD_CONDITION_CONDVARANY_HELPERS_H

#include <hip/std/chrono>
#include <cassert>

#include "test_macros.h"

#if TEST_STD_VER >= 17

// wait_for and wait_until function can exit via
// - predicate is true
// - timeout
// - stop_requested
// The return value only tells if the predicate is true
// when the function exits, but it does not tell whether
// it is timeout or stop_requested.
//
// ElapsedTimeCheck would fail the test if a test takes
// longer than a duration. This is useful because we can
// ensure that the wait_{for, until} function does not
// wait until the timeout
struct ElapsedTimeCheck {
  ElapsedTimeCheck(cuda::std::chrono::steady_clock::duration timeoutDuration)
      : timeout_(cuda::std::chrono::steady_clock::now() + timeoutDuration) {}

  ElapsedTimeCheck(ElapsedTimeCheck&&)            = delete;
  ElapsedTimeCheck& operator=(ElapsedTimeCheck&&) = delete;

  ~ElapsedTimeCheck() { assert(cuda::std::chrono::steady_clock::now() < timeout_); }

  cuda::std::chrono::time_point<cuda::std::chrono::steady_clock> timeout_;
};

#endif

#endif // TEST_STD_THREAD_CONDITION_CONDVARANY_HELPERS_H