Merge pull request #17 from olexp/add-factory

Add addFactoryToContainer method
Codeception · Oct 16, 2021 · 7423b3b · 7423b3b
2 parents 3cfcc47 + 1fd4b05
commit 7423b3b
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/src/Codeception/Lib/Connector/Laminas.php b/src/Codeception/Lib/Connector/Laminas.php
@@ -47,6 +47,11 @@ class Laminas extends AbstractBrowser
      */
     private $persistentServices = [];
 
+    /**
+     * @var array
+     */
+    private $persistentFactories = [];
+
     public function setApplicationConfig(array $applicationConfig): void
     {
         $this->applicationConfig = $applicationConfig;
@@ -148,6 +153,15 @@ public function addServiceToContainer(string $name, $service): void
         $this->persistentServices[$name] = $service;
     }
 
+    public function addFactoryToContainer(string $name, $factory): void
+    {
+        $this->application->getServiceManager()->setAllowOverride(true);
+        $this->application->getServiceManager()->setFactory($name, $factory);
+        $this->application->getServiceManager()->setAllowOverride(false);
+
+        $this->persistentFactories[$name] = $factory;
+    }
+
     private function extractHeaders(BrowserKitRequest $browserKitRequest): Headers
     {
         $headers        = [];
@@ -200,5 +214,9 @@ private function createApplication(): void
         $eventManager               = $this->application->getEventManager();
 
         $eventManager->detach([$sendResponseListener, 'sendResponse']);
+
+        $serviceManager->setAllowOverride(true);
+        $serviceManager->configure(['factories' => $this->persistentFactories]);
+        $serviceManager->setAllowOverride(false);
     }
 }
diff --git a/src/Codeception/Module/Laminas.php b/src/Codeception/Module/Laminas.php
@@ -18,6 +18,7 @@
 use Laminas\Router\Http\Hostname;
 use Laminas\Router\Http\Part;
 use Laminas\Router\Http\TreeRouteStack;
+use Laminas\ServiceManager\Factory\FactoryInterface;
 use Traversable;
 use function array_unique;
 use function class_exists;
@@ -181,6 +182,19 @@ public function addServiceToContainer(string $name, object $service): void
         $this->client->addServiceToContainer($name, $service);
     }
 
+    /**
+     * Adds factory to a Laminas container
+     *
+     * @param string $name
+     * @param string|callable|FactoryInterface $factory
+     * @return void
+     * @part services
+     */
+    public function addFactoryToContainer(string $name, $factory): void
+    {
+        $this->client->addFactoryToContainer($name, $factory);
+    }
+
     /**
      * Opens web page using route name and parameters.
      *