gecko-dev/netwerk/test/TestCommon.h

49 строки
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