gecko-dev/netwerk/test/TestCommon.h

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