From adb0b8fb4a469105bb732934b69aec3b9888cdf6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jan=20Jakes=CC=8C?= <jan@jakes.pro>
Date: Wed, 30 Oct 2019 14:04:13 +0100
Subject: [PATCH] Cleanup EntityManager for data factories in acceptance tests

[MAILPOET-2436]
---
 tests/_support/CleanupExtension.php | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tests/_support/CleanupExtension.php b/tests/_support/CleanupExtension.php
index 25894b17ae..25b9f0408d 100644
--- a/tests/_support/CleanupExtension.php
+++ b/tests/_support/CleanupExtension.php
@@ -5,6 +5,8 @@ use Codeception\Event\TestEvent;
 use Codeception\Events;
 use Codeception\Extension;
 use MailPoet\Config\Env;
+use MailPoet\DI\ContainerWrapper;
+use MailPoetVendor\Doctrine\ORM\EntityManager;
 
 class CleanupExtension extends Extension { // phpcs:ignore PSR1.Classes.ClassDeclaration
   const DB_BACKUP_PATH = __DIR__ . '/../_data/acceptanceBackup.sql';
@@ -66,6 +68,9 @@ class CleanupExtension extends Extension { // phpcs:ignore PSR1.Classes.ClassDec
   function cleanupEnvironment(TestEvent $event) {
     $this->root_connection->exec(file_get_contents(self::DB_BACKUP_PATH));
     exec('rm -rf ' . self::MAILHOG_DATA_PATH . '/*', $output);
+
+    // cleanup EntityManager for data factories that are using it
+    ContainerWrapper::getInstance()->get(EntityManager::class)->clear();
   }
 
   private function createDsnConnectionString() {