From 260c0ba3a62cbb4e24eb1f33557f615202639ec4 Mon Sep 17 00:00:00 2001
From: Serhii Donii <serhii.donii@macpaw.com>
Date: Tue, 26 Mar 2024 13:39:11 +0200
Subject: [PATCH] ORC-816: Update container definition

Signed-off-by: Serhii Donii <serhii.donii@macpaw.com>
---
 src/DependencyInjection/BehatApiContextExtension.php | 12 +++---------
 src/Resources/config/api_context.xml                 |  1 +
 2 files changed, 4 insertions(+), 9 deletions(-)

diff --git a/src/DependencyInjection/BehatApiContextExtension.php b/src/DependencyInjection/BehatApiContextExtension.php
index 717610b..bc58d10 100644
--- a/src/DependencyInjection/BehatApiContextExtension.php
+++ b/src/DependencyInjection/BehatApiContextExtension.php
@@ -60,18 +60,12 @@ private function enableOrmContext(array $config, ContainerBuilder $container): v
     {
         $config['use_orm_context'] = $config['use_orm_context'] ?? true;
 
-        if (!$config['use_orm_context']) {
+        if ($config['use_orm_context']) {
             return;
         }
 
-        if (!$container->has(EntityManagerInterface::class)) {
-            throw new RuntimeException('Entity manager does not exists');
+        if ($container->getDefinition(ORMContext::class)) {
+            $container->removeDefinition(ORMContext::class);
         }
-
-        $entityManagerDef = $container->get(EntityManagerInterface::class);
-
-        $ormContextDef = new Definition(ORMContext::class);
-        $ormContextDef->setArgument('$manager', $entityManagerDef);
-        $container->setDefinition(ORMContext::class, $ormContextDef);
     }
 }
diff --git a/src/Resources/config/api_context.xml b/src/Resources/config/api_context.xml
index c86c7da..9689b34 100644
--- a/src/Resources/config/api_context.xml
+++ b/src/Resources/config/api_context.xml
@@ -5,6 +5,7 @@
 
     <services>
         <service public="true" autowire="true" id="BehatApiContext\Context\ApiContext" class="BehatApiContext\Context\ApiContext"/>
+        <service public="true" autowire="true" id="BehatApiContext\Context\ORMContext" class="BehatApiContext\Context\ORMContext"/>
 
         <service public="false" id="BehatApiContext\Service\ResetManager\DoctrineResetManager" class="BehatApiContext\Service\ResetManager\DoctrineResetManager"/>
     </services>