/* -*- Mode: C++; tab-width: 12; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* 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 "nsIThread.h"
#include "nsThreadUtils.h"
#include "mozilla/SyncRunnable.h"

#include "gtest/gtest.h"

using namespace mozilla;

nsIThread* gThread = nullptr;

class TestRunnable : public Runnable {
 public:
  TestRunnable() : Runnable("TestRunnable"), ran_(false) {}

  NS_IMETHOD Run() override {
    ran_ = true;

    return NS_OK;
  }

  bool ran() const { return ran_; }

 private:
  bool ran_;
};

class TestSyncRunnable : public ::testing::Test {
 public:
  static void SetUpTestCase() {
    nsresult rv = NS_NewNamedThread("thread", &gThread);
    ASSERT_TRUE(NS_SUCCEEDED(rv));
  }

  static void TearDownTestCase() {
    if (gThread) gThread->Shutdown();
  }
};

TEST_F(TestSyncRunnable, TestDispatch) {
  RefPtr<TestRunnable> r(new TestRunnable());
  RefPtr<SyncRunnable> s(new SyncRunnable(r));
  s->DispatchToThread(gThread);

  ASSERT_TRUE(r->ran());
}

TEST_F(TestSyncRunnable, TestDispatchStatic) {
  RefPtr<TestRunnable> r(new TestRunnable());
  SyncRunnable::DispatchToThread(gThread, r);
  ASSERT_TRUE(r->ran());
}