From e79b0e6b25e2afbe568da545af4ec12f8a38a8d3 Mon Sep 17 00:00:00 2001
From: Jan Jakes <jan.jakes@automattic.com>
Date: Thu, 31 Mar 2022 15:32:37 +0200
Subject: [PATCH] Activate automation for integration tests

[MAILPOET-4207]
---
 mailpoet/tests/integration/_bootstrap.php | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/mailpoet/tests/integration/_bootstrap.php b/mailpoet/tests/integration/_bootstrap.php
index cafb7d8362..6a9ffd0898 100644
--- a/mailpoet/tests/integration/_bootstrap.php
+++ b/mailpoet/tests/integration/_bootstrap.php
@@ -1,8 +1,13 @@
 <?php
 
+use MailPoet\Automation\Engine\Engine;
+use MailPoet\Automation\Engine\Hooks;
+use MailPoet\Automation\Engine\Migrations\Migrator;
+use MailPoet\Automation\Integrations\MailPoet\MailPoetIntegration;
 use MailPoet\Cache\TransientCache;
 use MailPoet\DI\ContainerWrapper;
 use MailPoet\Entities\ScheduledTaskEntity;
+use MailPoet\Features\FeatureFlagsController;
 use MailPoet\Features\FeaturesController;
 use MailPoet\Settings\SettingsController;
 use MailPoetVendor\Doctrine\DBAL\Connection;
@@ -85,6 +90,21 @@ if (is_dir((string)getenv('WP_TEST_CACHE_PATH'))) {
 // the action is called in ConflictResolverTest
 remove_filter('admin_print_styles', 'wp_resource_hints', 1);
 
+// enable & initialize automation (this is needed only when behind a feature flag)
+$_SERVER['SERVER_NAME'] = '';
+$container = ContainerWrapper::getInstance();
+$migrator = $container->get(Migrator::class);
+$container->get(FeatureFlagsController::class)->set(FeaturesController::AUTOMATION, true);
+if ($migrator->hasSchema()) {
+  $migrator->deleteSchema();
+}
+$migrator->createSchema();
+$action = [$container->get(MailPoetIntegration::class), 'register'];
+if (!has_action(Hooks::INITIALIZE, $action) && is_callable($action)) {
+  add_action(Hooks::INITIALIZE, $action);
+  $container->get(Engine::class)->initialize();
+}
+
 /**
  * @property IntegrationTester $tester
  */