Chris@0: <?php
Chris@0: 
Chris@0: namespace Drupal\simpletest;
Chris@0: 
Chris@0: use Drupal\Core\DependencyInjection\ContainerBuilder;
Chris@0: use Drupal\Core\DependencyInjection\ServiceModifierInterface;
Chris@0: use Drupal\Core\DependencyInjection\ServiceProviderInterface;
Chris@0: use Symfony\Component\DependencyInjection\Definition;
Chris@0: 
Chris@0: class TestServiceProvider implements ServiceProviderInterface, ServiceModifierInterface {
Chris@0: 
Chris@0:   /**
Chris@0:    * @var \Drupal\simpletest\TestBase;
Chris@0:    */
Chris@0:   public static $currentTest;
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   public function register(ContainerBuilder $container) {
Chris@0:     if (static::$currentTest && method_exists(static::$currentTest, 'containerBuild')) {
Chris@0:       static::$currentTest->containerBuild($container);
Chris@0:     }
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   public function alter(ContainerBuilder $container) {
Chris@0:     if (static::$currentTest instanceof KernelTestBase) {
Chris@0:       static::addRouteProvider($container);
Chris@0:     }
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Add the on demand rebuild route provider service.
Chris@0:    *
Chris@0:    * @param \Drupal\Core\DependencyInjection\ContainerBuilder $container
Chris@0:    */
Chris@0:   public static function addRouteProvider(ContainerBuilder $container) {
Chris@0:     foreach (['router.route_provider' => 'RouteProvider'] as $original_id => $class) {
Chris@0:       // While $container->get() does a recursive resolve, getDefinition() does
Chris@0:       // not, so do it ourselves.
Chris@0:       // @todo Make the code more readable in
Chris@0:       //   https://www.drupal.org/node/2911498.
Chris@0:       for ($id = $original_id; $container->hasAlias($id); $id = (string) $container->getAlias($id)) {
Chris@0:       }
Chris@0:       $definition = $container->getDefinition($id);
Chris@0:       $definition->clearTag('needs_destruction');
Chris@0:       $container->setDefinition("simpletest.$original_id", $definition);
Chris@0:       $container->setDefinition($id, new Definition('Drupal\simpletest\\' . $class));
Chris@0:     }
Chris@0:   }
Chris@0: 
Chris@0: }