gecko-dev/netwerk/test/TestCommon.h

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