OLD | NEW |
| (Empty) |
1 # Copyright 2014 The Chromium Authors. All rights reserved. | |
2 # Use of this source code is governed by a BSD-style license that can be | |
3 # found in the LICENSE file. | |
4 | |
5 import unittest | |
6 | |
7 # pylint: disable=F0401 | |
8 from mojo.bindings import promise | |
9 | |
10 | |
11 class PromiseTest(unittest.TestCase): | |
12 | |
13 def setUp(self): | |
14 self.accumulated = [] | |
15 | |
16 def _AddToAccumulated(self, res): | |
17 self.accumulated.append(res) | |
18 return res | |
19 | |
20 def testResolve(self): | |
21 p = promise.Promise.Resolve(0) | |
22 self.assertEquals(p.state, promise.Promise.STATE_FULLFILLED) | |
23 p.Then(self._AddToAccumulated) | |
24 self.assertEquals(self.accumulated, [0]) | |
25 | |
26 def testResolveToPromise(self): | |
27 p = promise.Promise.Resolve(0) | |
28 self.assertEquals(p.state, promise.Promise.STATE_FULLFILLED) | |
29 q = promise.Promise.Resolve(p) | |
30 self.assertEquals(p.state, promise.Promise.STATE_FULLFILLED) | |
31 q.Then(self._AddToAccumulated) | |
32 self.assertEquals(self.accumulated, [0]) | |
33 | |
34 def testReject(self): | |
35 p = promise.Promise.Reject(0) | |
36 self.assertEquals(p.state, promise.Promise.STATE_REJECTED) | |
37 p.Then(onRejected=self._AddToAccumulated) | |
38 self.assertEquals(self.accumulated, [0]) | |
39 | |
40 def testGeneratorFunctionResolve(self): | |
41 (p, resolve, _) = _GetPromiseAndFunctions() | |
42 self.assertEquals(p.state, promise.Promise.STATE_PENDING) | |
43 p.Then(self._AddToAccumulated) | |
44 resolve(0) | |
45 self.assertEquals(p.state, promise.Promise.STATE_FULLFILLED) | |
46 self.assertEquals(self.accumulated, [0]) | |
47 | |
48 def testGeneratorFunctionReject(self): | |
49 (p, _, reject) = _GetPromiseAndFunctions() | |
50 self.assertEquals(p.state, promise.Promise.STATE_PENDING) | |
51 p.Then(None, self._AddToAccumulated) | |
52 reject(0) | |
53 self.assertEquals(p.state, promise.Promise.STATE_REJECTED) | |
54 self.assertEquals(self.accumulated, [0]) | |
55 | |
56 def testGeneratorFunctionResolveToPromise(self): | |
57 (p1, resolve, _) = _GetPromiseAndFunctions() | |
58 p2 = promise.Promise(lambda x, y: x(p1)) | |
59 self.assertEquals(p2.state, promise.Promise.STATE_PENDING) | |
60 p2.Then(self._AddToAccumulated) | |
61 resolve(promise.Promise.Resolve(0)) | |
62 self.assertEquals(self.accumulated, [0]) | |
63 | |
64 def testComputation(self): | |
65 (p, resolve, _) = _GetPromiseAndFunctions() | |
66 p.Then(lambda x: x+1).Then(lambda x: x+2).Then(self._AddToAccumulated) | |
67 self.assertEquals(self.accumulated, []) | |
68 resolve(0) | |
69 self.assertEquals(self.accumulated, [3]) | |
70 | |
71 def testRecoverAfterException(self): | |
72 (p, resolve, _) = _GetPromiseAndFunctions() | |
73 p.Then(_ThrowException).Catch(self._AddToAccumulated) | |
74 self.assertEquals(self.accumulated, []) | |
75 resolve(0) | |
76 self.assertEquals(len(self.accumulated), 1) | |
77 self.assertIsInstance(self.accumulated[0], RuntimeError) | |
78 self.assertEquals(self.accumulated[0].message, 0) | |
79 | |
80 def testMultipleRejectResolve(self): | |
81 (p, resolve, reject) = _GetPromiseAndFunctions() | |
82 p.Then(self._AddToAccumulated, self._AddToAccumulated) | |
83 resolve(0) | |
84 self.assertEquals(self.accumulated, [0]) | |
85 resolve(0) | |
86 self.assertEquals(self.accumulated, [0]) | |
87 reject(0) | |
88 self.assertEquals(self.accumulated, [0]) | |
89 | |
90 self.accumulated = [] | |
91 (p, resolve, reject) = _GetPromiseAndFunctions() | |
92 p.Then(self._AddToAccumulated, self._AddToAccumulated) | |
93 reject(0) | |
94 self.assertEquals(self.accumulated, [0]) | |
95 resolve(0) | |
96 self.assertEquals(self.accumulated, [0]) | |
97 reject(0) | |
98 self.assertEquals(self.accumulated, [0]) | |
99 | |
100 def testAll(self): | |
101 promises_and_functions = [_GetPromiseAndFunctions() for x in xrange(10)] | |
102 promises = [x[0] for x in promises_and_functions] | |
103 all_promise = promise.Promise.All(*promises) | |
104 res = [] | |
105 def AddToRes(values): | |
106 res.append(values) | |
107 all_promise.Then(AddToRes, AddToRes) | |
108 for i, (_, resolve, _) in enumerate(promises_and_functions): | |
109 self.assertEquals(len(res), 0) | |
110 resolve(i) | |
111 self.assertEquals(len(res), 1) | |
112 self.assertEquals(res[0], [i for i in xrange(10)]) | |
113 self.assertEquals(all_promise.state, promise.Promise.STATE_FULLFILLED) | |
114 | |
115 def testAllFailure(self): | |
116 promises_and_functions = [_GetPromiseAndFunctions() for x in xrange(10)] | |
117 promises = [x[0] for x in promises_and_functions] | |
118 all_promise = promise.Promise.All(*promises) | |
119 res = [] | |
120 def AddToRes(values): | |
121 res.append(values) | |
122 all_promise.Then(AddToRes, AddToRes) | |
123 for i in xrange(10): | |
124 if i <= 5: | |
125 self.assertEquals(len(res), 0) | |
126 else: | |
127 self.assertEquals(len(res), 1) | |
128 if i != 5: | |
129 promises_and_functions[i][1](i) | |
130 else: | |
131 promises_and_functions[i][2]('error') | |
132 self.assertEquals(len(res), 1) | |
133 self.assertEquals(res[0], 'error') | |
134 self.assertEquals(all_promise.state, promise.Promise.STATE_REJECTED) | |
135 | |
136 def testRace(self): | |
137 promises_and_functions = [_GetPromiseAndFunctions() for x in xrange(10)] | |
138 promises = [x[0] for x in promises_and_functions] | |
139 race_promise = promise.Promise.Race(*promises) | |
140 res = [] | |
141 def AddToRes(values): | |
142 res.append(values) | |
143 race_promise.Then(AddToRes, AddToRes) | |
144 self.assertEquals(len(res), 0) | |
145 promises_and_functions[7][1]('success') | |
146 self.assertEquals(len(res), 1) | |
147 for i, (f) in enumerate(promises_and_functions): | |
148 f[1 + (i % 2)](i) | |
149 self.assertEquals(len(res), 1) | |
150 self.assertEquals(res[0], 'success') | |
151 self.assertEquals(race_promise.state, promise.Promise.STATE_FULLFILLED) | |
152 | |
153 def testRaceFailure(self): | |
154 promises_and_functions = [_GetPromiseAndFunctions() for x in xrange(10)] | |
155 promises = [x[0] for x in promises_and_functions] | |
156 race_promise = promise.Promise.Race(*promises) | |
157 res = [] | |
158 def AddToRes(values): | |
159 res.append(values) | |
160 race_promise.Then(AddToRes, AddToRes) | |
161 self.assertEquals(len(res), 0) | |
162 promises_and_functions[7][2]('error') | |
163 self.assertEquals(len(res), 1) | |
164 for i, (f) in enumerate(promises_and_functions): | |
165 f[1 + (i % 2)](i) | |
166 self.assertEquals(len(res), 1) | |
167 self.assertEquals(res[0], 'error') | |
168 self.assertEquals(race_promise.state, promise.Promise.STATE_REJECTED) | |
169 | |
170 | |
171 def _GetPromiseAndFunctions(): | |
172 functions = {} | |
173 def GeneratorFunction(resolve, reject): | |
174 functions['resolve'] = resolve | |
175 functions['reject'] = reject | |
176 p = promise.Promise(GeneratorFunction) | |
177 return (p, functions['resolve'], functions['reject']) | |
178 | |
179 | |
180 def _ThrowException(x): | |
181 raise RuntimeError(x) | |
OLD | NEW |