Unreviewed, rolling out r235784.
[WebKit-https.git] / Tools / TestWebKitAPI / Tests / WTF / RunLoop.cpp
1 /*
2  * Copyright (C) 2015 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "config.h"
27
28 #include "Utilities.h"
29 #include <wtf/RunLoop.h>
30
31 namespace TestWebKitAPI {
32
33 static bool testFinished;
34 static int count = 100;
35
36 TEST(WTF_RunLoop, Deadlock)
37 {
38     RunLoop::initializeMainRunLoop();
39
40     struct DispatchFromDestructorTester {
41         ~DispatchFromDestructorTester() {
42             RunLoop::main().dispatch([] {
43                 if (!(--count))
44                     testFinished = true;
45             });
46         }
47     };
48
49     for (int i = 0; i < count; ++i) {
50         auto capture = std::make_shared<DispatchFromDestructorTester>();
51         RunLoop::main().dispatch([capture] { });
52     }
53
54     Util::run(&testFinished);
55 }
56
57 TEST(WTF_RunLoop, NestedRunLoop)
58 {
59     RunLoop::initializeMainRunLoop();
60
61     bool testFinished = false;
62     RunLoop::current().dispatch([&] {
63         RunLoop::current().dispatch([&] {
64             testFinished = true;
65         });
66         Util::run(&testFinished);
67     });
68
69     Util::run(&testFinished);
70 }
71
72 TEST(WTF_RunLoop, OneShotTimer)
73 {
74     RunLoop::initializeMainRunLoop();
75
76     bool testFinished = false;
77
78     class DerivedTimer : public RunLoop::Timer<DerivedTimer> {
79     public:
80         DerivedTimer(bool& testFinished)
81             : RunLoop::Timer<DerivedTimer>(RunLoop::current(), this, &DerivedTimer::fired)
82             , m_testFinished(testFinished)
83         {
84         }
85
86         void fired()
87         {
88             m_testFinished = true;
89             stop();
90         }
91
92     private:
93         bool& m_testFinished;
94     };
95
96     {
97         DerivedTimer timer(testFinished);
98         timer.startOneShot(100_ms);
99         Util::run(&testFinished);
100     }
101 }
102
103 TEST(WTF_RunLoop, RepeatingTimer)
104 {
105     RunLoop::initializeMainRunLoop();
106
107     bool testFinished = false;
108
109     class DerivedTimer : public RunLoop::Timer<DerivedTimer> {
110     public:
111         DerivedTimer(bool& testFinished)
112             : RunLoop::Timer<DerivedTimer>(RunLoop::current(), this, &DerivedTimer::fired)
113             , m_testFinished(testFinished)
114         {
115         }
116
117         void fired()
118         {
119             if (++m_count == 10) {
120                 m_testFinished = true;
121                 stop();
122             }
123         }
124
125     private:
126         unsigned m_count { 0 };
127         bool& m_testFinished;
128     };
129
130     {
131         DerivedTimer timer(testFinished);
132         timer.startRepeating(10_ms);
133         Util::run(&testFinished);
134     }
135 }
136
137 TEST(WTF_RunLoop, ManyTimes)
138 {
139     RunLoop::initializeMainRunLoop();
140
141     class Counter {
142     public:
143         void run()
144         {
145             if (++m_count == 100000) {
146                 RunLoop::current().stop();
147                 return;
148             }
149             RunLoop::current().dispatch([this] {
150                 run();
151             });
152         }
153
154     private:
155         unsigned m_count { 0 };
156     };
157
158     Counter counter;
159
160     RunLoop::current().dispatch([&counter] {
161         counter.run();
162     });
163     RunLoop::run();
164 }
165
166 } // namespace TestWebKitAPI