gecko-dev/netwerk/test/TestCommon.h
Benjamin Smedberg d5e468da6c Bug 1313752 - Port binary tests in netwerk/test to gtest, or remove the ones that we currently aren't running. r=michal.novotny. Includes the following followups:
* TestUDPSocket wasn't initializing it's members properly, fixup by mcmanus
* Scope the netwerk TestCommon waiting to be non-global, and add assertions so that waiting is deterministic. r=mcmanus

MozReview-Commit-ID: 7jLgNIujrbu

--HG--
extra : rebase_source : fabee29260f06686e874d0130cb00067c769ad6f
2016-11-29 13:53:48 -05:00

51 lines
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 : public nsIRunnable
{
public:
NS_DECL_THREADSAFE_ISUPPORTS
void Wait(int pending)
{
MOZ_ASSERT(NS_IsMainThread());
MOZ_ASSERT(mPending == 0);
mPending = pending;
while (mPending) {
NS_ProcessNextEvent();
}
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