diff -r 000000000000 -r 6474c204b198 netwerk/test/TestCommon.h --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/netwerk/test/TestCommon.h Wed Dec 31 06:09:35 2014 +0100 @@ -0,0 +1,50 @@ +/* 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 +#include "nsThreadUtils.h" +#include "mozilla/Attributes.h" + +inline int test_common_init(int *argc, char ***argv) +{ + return 0; +} + +//----------------------------------------------------------------------------- + +static bool gKeepPumpingEvents = false; + +class nsQuitPumpingEvent MOZ_FINAL : public nsIRunnable { +public: + NS_DECL_THREADSAFE_ISUPPORTS + NS_IMETHOD Run() { + gKeepPumpingEvents = false; + return NS_OK; + } +}; +NS_IMPL_ISUPPORTS(nsQuitPumpingEvent, nsIRunnable) + +static inline void PumpEvents() +{ + nsCOMPtr thread = do_GetCurrentThread(); + + gKeepPumpingEvents = true; + while (gKeepPumpingEvents) + NS_ProcessNextEvent(thread); + + NS_ProcessPendingEvents(thread); +} + +static inline void QuitPumpingEvents() +{ + // Dispatch a task that toggles gKeepPumpingEvents so that we flush all + // of the pending tasks before exiting from PumpEvents. + nsCOMPtr event = new nsQuitPumpingEvent(); + NS_DispatchToMainThread(event); +} + +#endif