@@ -249,20 +249,23 @@ public function testWorkerWithMultipleReceivers()
249
249
250
250
public function testWorkerLimitQueues ()
251
251
{
252
- $ receiver1 = $ this ->createMock (QueueReceiverInterface::class);
253
- $ receiver1 ->expects ($ this ->once ())
252
+ $ envelope = [new Envelope (new DummyMessage ('message1 ' ))];
253
+ $ receiver = $ this ->createMock (QueueReceiverInterface::class);
254
+ $ receiver ->expects ($ this ->once ())
254
255
->method ('getFromQueues ' )
255
256
->with (['foo ' ])
256
- ->willReturn ([new Envelope (new DummyMessage ('message1 ' ))])
257
+ ->willReturn ($ envelope )
258
+ ;
259
+ $ receiver ->expects ($ this ->never ())
260
+ ->method ('get ' )
257
261
;
258
262
259
263
$ bus = $ this ->getMockBuilder (MessageBusInterface::class)->getMock ();
260
264
261
265
$ dispatcher = new EventDispatcher ();
262
266
$ dispatcher ->addSubscriber (new StopWorkerOnMessageLimitListener (1 ));
263
267
264
- $ worker = new Worker ([$ receiver1 ], $ bus , $ dispatcher );
265
- $ this ->expectException (RuntimeException::class);
268
+ $ worker = new Worker (['transport ' => $ receiver ], $ bus , $ dispatcher );
266
269
$ worker ->run (['queues ' => ['foo ' ]]);
267
270
}
268
271
@@ -273,8 +276,9 @@ public function testWorkerLimitQueuesUnsupported()
273
276
274
277
$ bus = $ this ->getMockBuilder (MessageBusInterface::class)->getMock ();
275
278
276
- $ worker = new Worker ([$ receiver1 , $ receiver2 ], $ bus );
279
+ $ worker = new Worker ([' transport1 ' => $ receiver1 , ' transport2 ' => $ receiver2 ], $ bus );
277
280
$ this ->expectException (RuntimeException::class);
281
+ $ this ->expectExceptionMessage (sprintf ('Receiver for transport2 does not implement %s ' , QueueReceiverInterface::class));
278
282
$ worker ->run (['queues ' => ['foo ' ]]);
279
283
}
280
284
0 commit comments