vendor/zenstruck/messenger-test/src/ZenstruckMessengerTestBundle.php line 30

  1. <?php
  3. /*
  4.  * This file is part of the zenstruck/messenger-test package.
  5.  *
  6.  * (c) Kevin Bond <kevinbond@gmail.com>
  7.  *
  8.  * For the full copyright and license information, please view the LICENSE
  9.  * file that was distributed with this source code.
  10.  */
  12. namespace Zenstruck\Messenger\Test;
  14. use Psr\Clock\ClockInterface;
  15. use Symfony\Component\DependencyInjection\Compiler\CompilerPassInterface;
  16. use Symfony\Component\DependencyInjection\ContainerBuilder;
  17. use Symfony\Component\DependencyInjection\ContainerInterface;
  18. use Symfony\Component\DependencyInjection\Extension\ExtensionInterface;
  19. use Symfony\Component\DependencyInjection\Reference;
  20. use Symfony\Component\HttpKernel\Bundle\Bundle;
  21. use Symfony\Component\Messenger\Transport\TransportInterface;
  22. use Zenstruck\Messenger\Test\Bus\TestBus;
  23. use Zenstruck\Messenger\Test\Bus\TestBusRegistry;
  24. use Zenstruck\Messenger\Test\Transport\TestTransportFactory;
  25. use Zenstruck\Messenger\Test\Transport\TestTransportRegistry;
  27. /**
  28.  * @author Kevin Bond <kevinbond@gmail.com>
  29.  */
  30. final class ZenstruckMessengerTestBundle extends Bundle implements CompilerPassInterface
  31. {
  32.     public function build(ContainerBuilder $container): void
  33.     {
  34.         $container->register('zenstruck_messenger_test.transport_factory', TestTransportFactory::class)
  35.             ->setArguments([
  36.                 new Reference('messenger.routable_message_bus'),
  37.                 new Reference('event_dispatcher'),
  38.                 new Reference(ClockInterface::class, invalidBehavior: ContainerInterface::NULL_ON_INVALID_REFERENCE),
  39.             ])
  40.             ->addTag('messenger.transport_factory')
  41.         ;
  43.         $container->register('zenstruck_messenger_test.transport_registry', TestTransportRegistry::class)
  44.             ->setPublic(true)
  45.         ;
  47.         $container->register('zenstruck_messenger_test.bus_registry', TestBusRegistry::class)
  48.             ->setPublic(true)
  49.         ;
  51.         $container->addCompilerPass($this);
  52.     }
  54.     public function getContainerExtension(): ?ExtensionInterface
  55.     {
  56.         return null;
  57.     }
  59.     public function process(ContainerBuilder $container): void
  60.     {
  61.         $transportRegistry = $container->getDefinition('zenstruck_messenger_test.transport_registry');
  63.         foreach ($container->findTaggedServiceIds('messenger.receiver') as $id => $tags) {
  64.             $name = $id;
  66.             if (!$class = $container->getDefinition($name)->getClass()) {
  67.                 continue;
  68.             }
  70.             if (!\is_a($class, TransportInterface::class, true)) {
  71.                 continue;
  72.             }
  74.             foreach ($tags as $tag) {
  75.                 if (isset($tag['alias'])) {
  76.                     $name = $tag['alias'];
  77.                 }
  78.             }
  80.             $transportRegistry->addMethodCall('register', [$name, new Reference($id)]);
  81.         }
  83.         $busRegistry = $container->getDefinition('zenstruck_messenger_test.bus_registry');
  85.         foreach ($container->findTaggedServiceIds('messenger.bus') as $id => $tags) {
  86.             $name = "{$id}.test-bus";
  87.             $busRegistry->addMethodCall('register', [$id, new Reference($name)]);
  88.             $container->register($name, TestBus::class)
  89.                 ->setAutowired(true)
  90.                 ->setPublic(true)
  91.                 ->setArgument('$name', $id)
  92.                 ->setDecoratedService($id)
  93.             ;
  94.         }
  95.     }
  96. }