зеркало из https://github.com/mozilla/gecko-dev.git
45 строки
1.0 KiB
C++
45 строки
1.0 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/. */
|
|
|
|
#ifndef TestCommon_h__
|
|
#define TestCommon_h__
|
|
|
|
#include <stdlib.h>
|
|
#include "nsThreadUtils.h"
|
|
#include "mozilla/Attributes.h"
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
class WaitForCondition final : public nsIRunnable {
|
|
public:
|
|
NS_DECL_THREADSAFE_ISUPPORTS
|
|
|
|
void Wait(int pending) {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
MOZ_ASSERT(mPending == 0);
|
|
|
|
mPending = pending;
|
|
mozilla::SpinEventLoopUntil([&]() { return !mPending; });
|
|
NS_ProcessPendingEvents(nullptr);
|
|
}
|
|
|
|
void Notify() { NS_DispatchToMainThread(this); }
|
|
|
|
private:
|
|
virtual ~WaitForCondition() {}
|
|
|
|
NS_IMETHOD Run() override {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
MOZ_ASSERT(mPending);
|
|
|
|
--mPending;
|
|
return NS_OK;
|
|
}
|
|
|
|
uint32_t mPending = 0;
|
|
};
|
|
NS_IMPL_ISUPPORTS(WaitForCondition, nsIRunnable)
|
|
|
|
#endif
|