Emilio Cobos Álvarez
2223a8bee2
Automatically generated with:
cp dom/.clang-format xpcom/ && mach format xpcom/**.{cpp,h}
Manual changes:
* nsTDependentSubstring.cpp was relying on include order, prevent that
from being changed by clang-format.
* Missing nsAtom.h include from nsGkAtoms.h
* nsWindowsHelpers.h needed to keep the include order
* MemoryPressureLevelMac.h needed mozilla/Attributes.h
Differential Revision: https://phabricator.services.mozilla.com/D311687
146 lines
3.9 KiB
C++
146 lines
3.9 KiB
C++
/* 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 "gtest/gtest.h"
|
|
#include "mozilla/Atomics.h"
|
|
#include "mozilla/gtest/MozAssertions.h"
|
|
#include "nsCOMPtr.h"
|
|
#include "nsIThread.h"
|
|
#include "nsIThreadManager.h"
|
|
#include "nsServiceManagerUtils.h"
|
|
#include "nsThreadUtils.h"
|
|
#include "nsXPCOM.h"
|
|
|
|
using mozilla::Atomic;
|
|
using mozilla::Runnable;
|
|
|
|
class WaitCondition final : public nsINestedEventLoopCondition {
|
|
public:
|
|
NS_DECL_THREADSAFE_ISUPPORTS
|
|
|
|
WaitCondition(Atomic<uint32_t>& aCounter, uint32_t aMaxCount)
|
|
: mCounter(aCounter), mMaxCount(aMaxCount) {}
|
|
|
|
NS_IMETHODIMP IsDone(bool* aDone) override {
|
|
*aDone = (mCounter == mMaxCount);
|
|
return NS_OK;
|
|
}
|
|
|
|
private:
|
|
~WaitCondition() = default;
|
|
|
|
Atomic<uint32_t>& mCounter;
|
|
const uint32_t mMaxCount;
|
|
};
|
|
|
|
NS_IMPL_ISUPPORTS(WaitCondition, nsINestedEventLoopCondition)
|
|
|
|
class SpinRunnable final : public Runnable {
|
|
public:
|
|
explicit SpinRunnable(nsINestedEventLoopCondition* aCondition)
|
|
: Runnable("SpinRunnable"), mCondition(aCondition), mResult(NS_OK) {}
|
|
|
|
NS_IMETHODIMP Run() {
|
|
nsCOMPtr<nsIThreadManager> threadMan =
|
|
do_GetService("@mozilla.org/thread-manager;1");
|
|
|
|
mResult = threadMan->SpinEventLoopUntil(
|
|
"xpcom:TestThreadManager.cpp:SpinRunnable->Run()"_ns, mCondition);
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult SpinLoopResult() { return mResult; }
|
|
|
|
private:
|
|
~SpinRunnable() = default;
|
|
|
|
nsCOMPtr<nsINestedEventLoopCondition> mCondition;
|
|
Atomic<nsresult> mResult;
|
|
};
|
|
|
|
class CountRunnable final : public Runnable {
|
|
public:
|
|
explicit CountRunnable(Atomic<uint32_t>& aCounter)
|
|
: Runnable("CountRunnable"), mCounter(aCounter) {}
|
|
|
|
NS_IMETHODIMP Run() {
|
|
mCounter++;
|
|
return NS_OK;
|
|
}
|
|
|
|
private:
|
|
Atomic<uint32_t>& mCounter;
|
|
};
|
|
|
|
TEST(ThreadManager, SpinEventLoopUntilSuccess)
|
|
{
|
|
const uint32_t kRunnablesToDispatch = 100;
|
|
nsresult rv;
|
|
mozilla::Atomic<uint32_t> count(0);
|
|
|
|
RefPtr condition =
|
|
mozilla::MakeRefPtr<WaitCondition>(count, kRunnablesToDispatch);
|
|
RefPtr spinner = mozilla::MakeRefPtr<SpinRunnable>(condition);
|
|
nsCOMPtr<nsIThread> thread;
|
|
rv = NS_NewNamedThread("SpinEventLoop", getter_AddRefs(thread), spinner);
|
|
ASSERT_NS_SUCCEEDED(rv);
|
|
|
|
RefPtr counter = mozilla::MakeRefPtr<CountRunnable>(count);
|
|
for (uint32_t i = 0; i < kRunnablesToDispatch; ++i) {
|
|
rv = thread->Dispatch(counter, NS_DISPATCH_NORMAL);
|
|
ASSERT_NS_SUCCEEDED(rv);
|
|
}
|
|
|
|
rv = thread->Shutdown();
|
|
ASSERT_NS_SUCCEEDED(rv);
|
|
ASSERT_NS_SUCCEEDED(spinner->SpinLoopResult());
|
|
}
|
|
|
|
class ErrorCondition final : public nsINestedEventLoopCondition {
|
|
public:
|
|
NS_DECL_THREADSAFE_ISUPPORTS
|
|
|
|
ErrorCondition(Atomic<uint32_t>& aCounter, uint32_t aMaxCount)
|
|
: mCounter(aCounter), mMaxCount(aMaxCount) {}
|
|
|
|
NS_IMETHODIMP IsDone(bool* aDone) override {
|
|
if (mCounter == mMaxCount) {
|
|
return NS_ERROR_ILLEGAL_VALUE;
|
|
}
|
|
return NS_OK;
|
|
}
|
|
|
|
private:
|
|
~ErrorCondition() = default;
|
|
|
|
Atomic<uint32_t>& mCounter;
|
|
const uint32_t mMaxCount;
|
|
};
|
|
|
|
NS_IMPL_ISUPPORTS(ErrorCondition, nsINestedEventLoopCondition)
|
|
|
|
TEST(ThreadManager, SpinEventLoopUntilError)
|
|
{
|
|
const uint32_t kRunnablesToDispatch = 100;
|
|
nsresult rv;
|
|
mozilla::Atomic<uint32_t> count(0);
|
|
|
|
RefPtr condition =
|
|
mozilla::MakeRefPtr<ErrorCondition>(count, kRunnablesToDispatch);
|
|
RefPtr spinner = mozilla::MakeRefPtr<SpinRunnable>(condition);
|
|
nsCOMPtr<nsIThread> thread;
|
|
rv = NS_NewNamedThread("SpinEventLoop", getter_AddRefs(thread), spinner);
|
|
ASSERT_NS_SUCCEEDED(rv);
|
|
|
|
RefPtr counter = mozilla::MakeRefPtr<CountRunnable>(count);
|
|
for (uint32_t i = 0; i < kRunnablesToDispatch; ++i) {
|
|
rv = thread->Dispatch(counter, NS_DISPATCH_NORMAL);
|
|
ASSERT_NS_SUCCEEDED(rv);
|
|
}
|
|
|
|
rv = thread->Shutdown();
|
|
ASSERT_NS_SUCCEEDED(rv);
|
|
ASSERT_NS_FAILED(spinner->SpinLoopResult());
|
|
}
|