--- /dev/null
+<?php
+
+/*
+ * This file is part of the Behat.
+ * (c) Konstantin Kudryashov <ever.zet@gmail.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Behat\Behat\Hook\ServiceContainer;
+
+use Behat\Behat\Context\ServiceContainer\ContextExtension;
+use Behat\Behat\Tester\ServiceContainer\TesterExtension;
+use Behat\Testwork\Hook\ServiceContainer\HookExtension as BaseExtension;
+use Symfony\Component\DependencyInjection\ContainerBuilder;
+use Symfony\Component\DependencyInjection\Definition;
+use Symfony\Component\DependencyInjection\Reference;
+
+/**
+ * Extends Testwork HookExtension with additional behat services.
+ *
+ * @author Konstantin Kudryashov <ever.zet@gmail.com>
+ */
+final class HookExtension extends BaseExtension
+{
+ /**
+ * {@inheritdoc}
+ */
+ public function load(ContainerBuilder $container, array $config)
+ {
+ parent::load($container, $config);
+
+ $this->loadAnnotationReader($container);
+ }
+
+ /**
+ * Loads hookable testers.
+ *
+ * @param ContainerBuilder $container
+ */
+ protected function loadHookableTesters(ContainerBuilder $container)
+ {
+ parent::loadHookableTesters($container);
+
+ $definition = new Definition('Behat\Behat\Hook\Tester\HookableFeatureTester', array(
+ new Reference(TesterExtension::SPECIFICATION_TESTER_ID),
+ new Reference(self::DISPATCHER_ID)
+ ));
+ $definition->addTag(TesterExtension::SPECIFICATION_TESTER_WRAPPER_TAG, array('priority' => 9999));
+ $container->setDefinition(TesterExtension::SPECIFICATION_TESTER_WRAPPER_TAG . '.hookable', $definition);
+
+ $definition = new Definition('Behat\Behat\Hook\Tester\HookableScenarioTester', array(
+ new Reference(TesterExtension::SCENARIO_TESTER_ID),
+ new Reference(self::DISPATCHER_ID)
+ )
+ );
+ $definition->addTag(TesterExtension::SCENARIO_TESTER_WRAPPER_TAG, array('priority' => 9999));
+ $container->setDefinition(TesterExtension::SCENARIO_TESTER_WRAPPER_TAG . '.hookable', $definition);
+
+ $definition = new Definition('Behat\Behat\Hook\Tester\HookableScenarioTester', array(
+ new Reference(TesterExtension::EXAMPLE_TESTER_ID),
+ new Reference(self::DISPATCHER_ID)
+ )
+ );
+ $definition->addTag(TesterExtension::EXAMPLE_TESTER_WRAPPER_TAG, array('priority' => 9999));
+ $container->setDefinition(TesterExtension::EXAMPLE_TESTER_WRAPPER_TAG . '.hookable', $definition);
+
+ $definition = new Definition('Behat\Behat\Hook\Tester\HookableStepTester', array(
+ new Reference(TesterExtension::STEP_TESTER_ID),
+ new Reference(self::DISPATCHER_ID)
+ ));
+ $definition->addTag(TesterExtension::STEP_TESTER_WRAPPER_TAG, array('priority' => 9999));
+ $container->setDefinition(TesterExtension::STEP_TESTER_WRAPPER_TAG . '.hookable', $definition);
+ }
+
+ /**
+ * Loads hook annotation reader.
+ *
+ * @param ContainerBuilder $container
+ */
+ private function loadAnnotationReader(ContainerBuilder $container)
+ {
+ $definition = new Definition('Behat\Behat\Hook\Context\Annotation\HookAnnotationReader');
+ $definition->addTag(ContextExtension::ANNOTATION_READER_TAG, array('priority' => 50));
+ $container->setDefinition(ContextExtension::ANNOTATION_READER_TAG . '.hook', $definition);
+ }
+}