netwerk/test/TestCommon.h
author Carsten "Tomcat" Book <cbook@mozilla.com>
Fri, 09 Jan 2015 16:24:14 +0100
changeset 248850 548ff18c9fc6a54dced7165f22f00ab803c746d2
parent 211336 ef6cb0f76224810fdaf44cd4c5164ff552e0ade0
child 251016 3f77b996196eccae6a97d83abe1acb4a214da29c
permissions -rw-r--r--
Backed out changeset f61a4592da85 (bug 1117607) for Reftest Failures/Crashes on a CLOSED TREE

/* 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"

inline int test_common_init(int *argc, char ***argv)
{
  return 0;
}

//-----------------------------------------------------------------------------

static bool gKeepPumpingEvents = false;

class nsQuitPumpingEvent MOZ_FINAL : public nsIRunnable {
  ~nsQuitPumpingEvent() {}
public:
  NS_DECL_THREADSAFE_ISUPPORTS
  NS_IMETHOD Run() {
    gKeepPumpingEvents = false;
    return NS_OK;
  }
};
NS_IMPL_ISUPPORTS(nsQuitPumpingEvent, nsIRunnable)

static inline void PumpEvents()
{
  nsCOMPtr<nsIThread> 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<nsIRunnable> event = new nsQuitPumpingEvent();
  NS_DispatchToMainThread(event);
}

#endif