22
33namespace Tests ;
44
5+ use Google \Cloud \PubSub \Message ;
56use Google \Cloud \PubSub \PubSubClient ;
67use Google \Cloud \PubSub \Subscription ;
78use Google \Cloud \PubSub \Topic ;
@@ -121,27 +122,16 @@ public function testPublishWhenAutoTopicCreationIsDisabled()
121122
122123 public function testSubscribeWhenSubscriptionMustBeCreated ()
123124 {
125+ $ message1 = new Message (['data ' => 'a:1:{s:5:"hello";s:5:"world";} ' ], ['ackId ' => 1 ]);
126+ $ message2 = new Message (['data ' => 'this is a string ' ], ['ackId ' => 2 ]);
127+ $ message3 = new Message (['data ' => 'unsubscribe ' ], ['ackId ' => 3 ]);
128+
124129 $ messageBatch1 = [
125- [
126- 'ackId ' => 1 ,
127- 'message ' => [
128- 'data ' => base64_encode ('a:1:{s:5:"hello";s:5:"world";} ' )
129- ],
130- ],
131- [
132- 'ackId ' => 2 ,
133- 'message ' => [
134- 'data ' => base64_encode ('this is a string ' )
135- ],
136- ],
130+ $ message1 ,
131+ $ message2 ,
137132 ];
138133 $ messageBatch2 = [
139- [
140- 'ackId ' => 3 ,
141- 'message ' => [
142- 'data ' => base64_encode ('unsubscribe ' )
143- ],
144- ],
134+ $ message3 ,
145135 ];
146136
147137 $ subscription = Mockery::mock (Subscription::class);
@@ -154,16 +144,16 @@ public function testSubscribeWhenSubscriptionMustBeCreated()
154144 ->once ()
155145 ->andReturn ($ messageBatch1 );
156146 $ subscription ->shouldReceive ('acknowledge ' )
157- ->with (1 )
147+ ->with ($ message1 )
158148 ->once ();
159149 $ subscription ->shouldReceive ('acknowledge ' )
160- ->with (2 )
150+ ->with ($ message2 )
161151 ->once ();
162152 $ subscription ->shouldReceive ('pull ' )
163153 ->once ()
164154 ->andReturn ($ messageBatch2 );
165155 $ subscription ->shouldReceive ('acknowledge ' )
166- ->with (3 )
156+ ->with ($ message3 )
167157 ->once ();
168158
169159 $ topic = Mockery::mock (Topic::class);
@@ -197,27 +187,16 @@ public function testSubscribeWhenSubscriptionMustBeCreated()
197187
198188 public function testSubscribeWhenSubscriptionExists ()
199189 {
190+ $ message1 = new Message (['data ' => 'a:1:{s:5:"hello";s:5:"world";} ' ], ['ackId ' => 1 ]);
191+ $ message2 = new Message (['data ' => 'this is a string ' ], ['ackId ' => 2 ]);
192+ $ message3 = new Message (['data ' => 'unsubscribe ' ], ['ackId ' => 3 ]);
193+
200194 $ messageBatch1 = [
201- [
202- 'ackId ' => 1 ,
203- 'message ' => [
204- 'data ' => base64_encode ('a:1:{s:5:"hello";s:5:"world";} ' )
205- ],
206- ],
207- [
208- 'ackId ' => 2 ,
209- 'message ' => [
210- 'data ' => base64_encode ('this is a string ' )
211- ],
212- ],
195+ $ message1 ,
196+ $ message2 ,
213197 ];
214198 $ messageBatch2 = [
215- [
216- 'ackId ' => 3 ,
217- 'message ' => [
218- 'data ' => base64_encode ('unsubscribe ' )
219- ],
220- ],
199+ $ message3 ,
221200 ];
222201
223202 $ subscription = Mockery::mock (Subscription::class);
@@ -229,16 +208,16 @@ public function testSubscribeWhenSubscriptionExists()
229208 ->once ()
230209 ->andReturn ($ messageBatch1 );
231210 $ subscription ->shouldReceive ('acknowledge ' )
232- ->with (1 )
211+ ->with ($ message1 )
233212 ->once ();
234213 $ subscription ->shouldReceive ('acknowledge ' )
235- ->with (2 )
214+ ->with ($ message2 )
236215 ->once ();
237216 $ subscription ->shouldReceive ('pull ' )
238217 ->once ()
239218 ->andReturn ($ messageBatch2 );
240219 $ subscription ->shouldReceive ('acknowledge ' )
241- ->with (3 )
220+ ->with ($ message3 )
242221 ->once ();
243222
244223 $ topic = Mockery::mock (Topic::class);
@@ -272,27 +251,16 @@ public function testSubscribeWhenSubscriptionExists()
272251
273252 public function testSubscribeWhenAutoTopicCreationIsDisabled ()
274253 {
254+ $ message1 = new Message (['data ' => 'a:1:{s:5:"hello";s:5:"world";} ' ], ['ackId ' => 1 ]);
255+ $ message2 = new Message (['data ' => 'this is a string ' ], ['ackId ' => 2 ]);
256+ $ message3 = new Message (['data ' => 'unsubscribe ' ], ['ackId ' => 3 ]);
257+
275258 $ messageBatch1 = [
276- [
277- 'ackId ' => 1 ,
278- 'message ' => [
279- 'data ' => base64_encode ('a:1:{s:5:"hello";s:5:"world";} ' )
280- ],
281- ],
282- [
283- 'ackId ' => 2 ,
284- 'message ' => [
285- 'data ' => base64_encode ('this is a string ' )
286- ],
287- ],
259+ $ message1 ,
260+ $ message2 ,
288261 ];
289262 $ messageBatch2 = [
290- [
291- 'ackId ' => 3 ,
292- 'message ' => [
293- 'data ' => base64_encode ('unsubscribe ' )
294- ],
295- ],
263+ $ message3 ,
296264 ];
297265
298266 $ subscription = Mockery::mock (Subscription::class);
@@ -302,16 +270,16 @@ public function testSubscribeWhenAutoTopicCreationIsDisabled()
302270 ->once ()
303271 ->andReturn ($ messageBatch1 );
304272 $ subscription ->shouldReceive ('acknowledge ' )
305- ->with (1 )
273+ ->with ($ message1 )
306274 ->once ();
307275 $ subscription ->shouldReceive ('acknowledge ' )
308- ->with (2 )
276+ ->with ($ message2 )
309277 ->once ();
310278 $ subscription ->shouldReceive ('pull ' )
311279 ->once ()
312280 ->andReturn ($ messageBatch2 );
313281 $ subscription ->shouldReceive ('acknowledge ' )
314- ->with (3 )
282+ ->with ($ message3 )
315283 ->once ();
316284
317285 $ topic = Mockery::mock (Topic::class);
0 commit comments