summaryrefslogtreecommitdiffstats
path: root/netwerk/test/gtest/TestCommon.h
blob: 76620fbb447cb940c5eabe11d9b67b6ca3e8698a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
/* 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"
#include "mozilla/SpinEventLoopUntil.h"

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

class WaitForCondition final : public nsIRunnable {
 public:
  NS_DECL_THREADSAFE_ISUPPORTS

  void Wait(int pending) {
    MOZ_RELEASE_ASSERT(NS_IsMainThread());
    MOZ_RELEASE_ASSERT(mPending == 0);

    mPending = pending;
    mozilla::SpinEventLoopUntil("TestCommon.h:WaitForCondition::Wait"_ns,
                                [&]() { return !mPending; });
    NS_ProcessPendingEvents(nullptr);
  }

  void Notify() { NS_DispatchToMainThread(this); }

 private:
  virtual ~WaitForCondition() = default;

  NS_IMETHOD Run() override {
    MOZ_RELEASE_ASSERT(NS_IsMainThread());
    MOZ_RELEASE_ASSERT(mPending);

    --mPending;
    return NS_OK;
  }

  uint32_t mPending = 0;
};

#endif