@@ -314,16 +314,14 @@ public function testSuccessMessageGoesToStdout()
314314 $ receiver = $ this ->createMock (ReceiverInterface::class);
315315 $ receiver ->expects ($ this ->once ())->method ('get ' )->willReturn ([$ envelope ]);
316316
317- $ receiverLocator = $ this ->createMock (ContainerInterface::class);
318- $ receiverLocator ->expects ($ this ->once ())->method ('has ' )->with ('dummy-receiver ' )->willReturn (true );
319- $ receiverLocator ->expects ($ this ->once ())->method ('get ' )->with ('dummy-receiver ' )->willReturn ($ receiver );
317+ $ receiverLocator = new Container ();
318+ $ receiverLocator ->set ('dummy-receiver ' , $ receiver );
320319
321320 $ bus = $ this ->createMock (MessageBusInterface::class);
322321 $ bus ->expects ($ this ->once ())->method ('dispatch ' );
323322
324- $ busLocator = $ this ->createMock (ContainerInterface::class);
325- $ busLocator ->expects ($ this ->once ())->method ('has ' )->with ('dummy-bus ' )->willReturn (true );
326- $ busLocator ->expects ($ this ->once ())->method ('get ' )->with ('dummy-bus ' )->willReturn ($ bus );
323+ $ busLocator = new Container ();
324+ $ busLocator ->set ('dummy-bus ' , $ bus );
327325
328326 $ command = new ConsumeMessagesCommand (new RoutableMessageBus ($ busLocator ), $ receiverLocator , new EventDispatcher ());
329327
@@ -349,16 +347,14 @@ public function testCommentsGoToStderr()
349347 $ receiver = $ this ->createMock (ReceiverInterface::class);
350348 $ receiver ->expects ($ this ->once ())->method ('get ' )->willReturn ([$ envelope ]);
351349
352- $ receiverLocator = $ this ->createMock (ContainerInterface::class);
353- $ receiverLocator ->expects ($ this ->once ())->method ('has ' )->with ('dummy-receiver ' )->willReturn (true );
354- $ receiverLocator ->expects ($ this ->once ())->method ('get ' )->with ('dummy-receiver ' )->willReturn ($ receiver );
350+ $ receiverLocator = new Container ();
351+ $ receiverLocator ->set ('dummy-receiver ' , $ receiver );
355352
356353 $ bus = $ this ->createMock (MessageBusInterface::class);
357354 $ bus ->expects ($ this ->once ())->method ('dispatch ' );
358355
359- $ busLocator = $ this ->createMock (ContainerInterface::class);
360- $ busLocator ->expects ($ this ->once ())->method ('has ' )->with ('dummy-bus ' )->willReturn (true );
361- $ busLocator ->expects ($ this ->once ())->method ('get ' )->with ('dummy-bus ' )->willReturn ($ bus );
356+ $ busLocator = new Container ();
357+ $ busLocator ->set ('dummy-bus ' , $ bus );
362358
363359 $ command = new ConsumeMessagesCommand (new RoutableMessageBus ($ busLocator ), $ receiverLocator , new EventDispatcher ());
364360
0 commit comments