Uses the test container if exists

Behat · Apr 8, 2018 · 2eef810 · 2eef810
1 parent 022ece0
commit 2eef810
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/Behat/Symfony2Extension/Context/Argument/ServiceArgumentResolver.php b/src/Behat/Symfony2Extension/Context/Argument/ServiceArgumentResolver.php
@@ -70,6 +70,7 @@ private function resolveArgument($argument)
         }
 
         $container = $this->kernel->getContainer();
+        $container = $container->has('test.service_container') ? $container->get('test.service_container') : $container;
 
         if ($service = $this->getService($container, $argument)) {
             return $service;