/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "jsapi-tests/tests.h"
#include "threading/ConditionVariable.h"
#include "threading/Thread.h"
#include "vm/MutexIDs.h"

struct TestState {
  js::Mutex mutex MOZ_UNANNOTATED;
  js::ConditionVariable condition;
  bool flag;
  js::Thread testThread;

  explicit TestState(bool createThread = true)
      : mutex(js::mutexid::TestMutex), flag(false) {
    if (createThread) {
      MOZ_RELEASE_ASSERT(testThread.init(setFlag, this));
    }
  }

  static void setFlag(TestState* state) {
    js::UniqueLock<js::Mutex> lock(state->mutex);
    state->flag = true;
    state->condition.notify_one();
  }

  void join() { testThread.join(); }
};

BEGIN_TEST(testThreadingConditionVariable) {
  auto state = mozilla::MakeUnique<TestState>();
  {
    js::UniqueLock<js::Mutex> lock(state->mutex);
    while (!state->flag) {
      state->condition.wait(lock);
    }
  }
  state->join();

  CHECK(state->flag);

  return true;
}
END_TEST(testThreadingConditionVariable)

BEGIN_TEST(testThreadingConditionVariablePredicate) {
  auto state = mozilla::MakeUnique<TestState>();
  {
    js::UniqueLock<js::Mutex> lock(state->mutex);
    state->condition.wait(lock, [&state]() { return state->flag; });
  }
  state->join();

  CHECK(state->flag);

  return true;
}
END_TEST(testThreadingConditionVariablePredicate)

BEGIN_TEST(testThreadingConditionVariableUntilOkay) {
  auto state = mozilla::MakeUnique<TestState>();
  {
    js::UniqueLock<js::Mutex> lock(state->mutex);
    while (!state->flag) {
      auto to =
          mozilla::TimeStamp::Now() + mozilla::TimeDuration::FromSeconds(600);
      js::CVStatus res = state->condition.wait_until(lock, to);
      CHECK(res == js::CVStatus::NoTimeout);
    }
  }
  state->join();

  CHECK(state->flag);

  return true;
}
END_TEST(testThreadingConditionVariableUntilOkay)

BEGIN_TEST(testThreadingConditionVariableUntilTimeout) {
  auto state = mozilla::MakeUnique<TestState>(false);
  {
    js::UniqueLock<js::Mutex> lock(state->mutex);
    while (!state->flag) {
      auto to = mozilla::TimeStamp::Now() +
                mozilla::TimeDuration::FromMilliseconds(10);
      js::CVStatus res = state->condition.wait_until(lock, to);
      if (res == js::CVStatus::Timeout) {
        break;
      }
    }
  }
  CHECK(!state->flag);

  // Timeout in the past should return with timeout immediately.
  {
    js::UniqueLock<js::Mutex> lock(state->mutex);
    auto to =
        mozilla::TimeStamp::Now() - mozilla::TimeDuration::FromMilliseconds(10);
    js::CVStatus res = state->condition.wait_until(lock, to);
    CHECK(res == js::CVStatus::Timeout);
  }

  return true;
}
END_TEST(testThreadingConditionVariableUntilTimeout)

BEGIN_TEST(testThreadingConditionVariableUntilOkayPredicate) {
  auto state = mozilla::MakeUnique<TestState>();
  {
    js::UniqueLock<js::Mutex> lock(state->mutex);
    auto to =
        mozilla::TimeStamp::Now() + mozilla::TimeDuration::FromSeconds(600);
    bool res = state->condition.wait_until(lock, to,
                                           [&state]() { return state->flag; });
    CHECK(res);
  }
  state->join();

  CHECK(state->flag);

  return true;
}
END_TEST(testThreadingConditionVariableUntilOkayPredicate)

BEGIN_TEST(testThreadingConditionVariableUntilTimeoutPredicate) {
  auto state = mozilla::MakeUnique<TestState>(false);
  {
    js::UniqueLock<js::Mutex> lock(state->mutex);
    auto to =
        mozilla::TimeStamp::Now() + mozilla::TimeDuration::FromMilliseconds(10);
    bool res = state->condition.wait_until(lock, to,
                                           [&state]() { return state->flag; });
    CHECK(!res);
  }
  CHECK(!state->flag);

  return true;
}
END_TEST(testThreadingConditionVariableUntilTimeoutPredicate)

BEGIN_TEST(testThreadingConditionVariableForOkay) {
  auto state = mozilla::MakeUnique<TestState>();
  {
    js::UniqueLock<js::Mutex> lock(state->mutex);
    while (!state->flag) {
      auto duration = mozilla::TimeDuration::FromSeconds(600);
      js::CVStatus res = state->condition.wait_for(lock, duration);
      CHECK(res == js::CVStatus::NoTimeout);
    }
  }
  state->join();

  CHECK(state->flag);

  return true;
}
END_TEST(testThreadingConditionVariableForOkay)

BEGIN_TEST(testThreadingConditionVariableForTimeout) {
  auto state = mozilla::MakeUnique<TestState>(false);
  {
    js::UniqueLock<js::Mutex> lock(state->mutex);
    while (!state->flag) {
      auto duration = mozilla::TimeDuration::FromMilliseconds(10);
      js::CVStatus res = state->condition.wait_for(lock, duration);
      if (res == js::CVStatus::Timeout) {
        break;
      }
    }
  }
  CHECK(!state->flag);

  // Timeout in the past should return with timeout immediately.
  {
    js::UniqueLock<js::Mutex> lock(state->mutex);
    auto duration = mozilla::TimeDuration::FromMilliseconds(-10);
    js::CVStatus res = state->condition.wait_for(lock, duration);
    CHECK(res == js::CVStatus::Timeout);
  }

  return true;
}
END_TEST(testThreadingConditionVariableForTimeout)

BEGIN_TEST(testThreadingConditionVariableForOkayPredicate) {
  auto state = mozilla::MakeUnique<TestState>();
  {
    js::UniqueLock<js::Mutex> lock(state->mutex);
    auto duration = mozilla::TimeDuration::FromSeconds(600);
    bool res = state->condition.wait_for(lock, duration,
                                         [&state]() { return state->flag; });
    CHECK(res);
  }
  state->join();

  CHECK(state->flag);

  return true;
}
END_TEST(testThreadingConditionVariableForOkayPredicate)

BEGIN_TEST(testThreadingConditionVariableForTimeoutPredicate) {
  auto state = mozilla::MakeUnique<TestState>(false);
  {
    js::UniqueLock<js::Mutex> lock(state->mutex);
    auto duration = mozilla::TimeDuration::FromMilliseconds(10);
    bool res = state->condition.wait_for(lock, duration,
                                         [&state]() { return state->flag; });
    CHECK(!res);
  }
  CHECK(!state->flag);

  return true;
}
END_TEST(testThreadingConditionVariableForTimeoutPredicate)