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
|