Refactor our environment handler to decorate the original one

kollex · hbagherzadegan · Jul 24, 2023 · Jul 22, 2016 · Dec 19, 2018 · Jan 10, 2019
commit db0a1d1ba939c4884060e7a659c58018c47e61df
diff --git a/src/Context/Environment/ContextServiceEnvironment.php b/src/Context/Environment/ContextServiceEnvironment.php
diff --git a/.../InitialisedContextServiceEnvironment.php → ...nitialisedSymfonyExtensionEnvironment.php b/.../InitialisedContextServiceEnvironment.php → ...nitialisedSymfonyExtensionEnvironment.php
@@ -22,7 +22,7 @@
 /**
  * @see ContextServiceEnvironmentHandler
  */
-final class InitialisedContextServiceEnvironment implements ContextServiceEnvironment
+final class InitialisedSymfonyExtensionEnvironment implements SymfonyExtensionEnvironment
 {
     /** @var Suite */
     private $suite;

diff --git a/src/Context/Environment/UninitialisedContextServiceEnvironment.php b/src/Context/Environment/UninitialisedContextServiceEnvironment.php
diff --git a/src/Context/Environment/UninitialisedSymfonyExtensionEnvironment.php b/src/Context/Environment/UninitialisedSymfonyExtensionEnvironment.php
@@ -0,0 +1,64 @@
+<?php
+
+declare(strict_types=1);
+
+/*
+ * This file is part of the SymfonyExtension package.
+ *
+ * (c) Kamil Kokot <kamil@kokot.me>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace FriendsOfBehat\SymfonyExtension\Context\Environment;
+
+use Behat\Behat\Context\Environment\ContextEnvironment;
+use Behat\Testwork\Environment\StaticEnvironment;
+use Behat\Testwork\Suite\Suite;
+use FriendsOfBehat\SymfonyExtension\Context\Environment\Handler\ContextServiceEnvironmentHandler;
+
+/**
+ * @see ContextServiceEnvironmentHandler
+ */
+final class UninitialisedSymfonyExtensionEnvironment extends StaticEnvironment implements SymfonyExtensionEnvironment
+{
+    /** @var string[] */
+    private $contexts;
+
+    /** @var ContextEnvironment|null */
+    private $delegatedEnvironment;
+
+    public function __construct(Suite $suite, array $contexts, ContextEnvironment $delegatedEnvironment)
+    {
+        parent::__construct($suite);
+
+        $this->contexts = $contexts;
+        $this->delegatedEnvironment = $delegatedEnvironment;
+    }
+
+    public function getServices(): array
+    {
+        return array_keys($this->contexts);
+    }
+
+    public function hasContexts(): bool
+    {
+        return count($this->contexts) > 0 || $this->delegatedEnvironment->hasContexts();
+    }
+
+    public function getContextClasses(): array
+    {
+        return array_merge(array_values($this->contexts), $this->delegatedEnvironment->getContextClasses());
+    }
+
+    public function hasContextClass($class): bool
+    {
+        return in_array($class, $this->contexts, true) || $this->delegatedEnvironment->hasContextClass($class);
+    }
+
+    public function getDelegatedEnvironment(): ContextEnvironment
+    {
+        return $this->delegatedEnvironment;
+    }
+}