From 339cba3c4265f3b44ef0f9d061d37bee4f2988e9 Mon Sep 17 00:00:00 2001
From: David Remer <webmaster@websupporter.net>
Date: Wed, 14 Sep 2022 14:19:18 +0300
Subject: [PATCH] Throw InvalidStateException when workflow is not found

[MAILPOET-4533]
---
 .../Engine/Builder/CreateWorkflowFromTemplateController.php    | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mailpoet/lib/Automation/Engine/Builder/CreateWorkflowFromTemplateController.php b/mailpoet/lib/Automation/Engine/Builder/CreateWorkflowFromTemplateController.php
index 178ac7c5ce..6e681770ad 100644
--- a/mailpoet/lib/Automation/Engine/Builder/CreateWorkflowFromTemplateController.php
+++ b/mailpoet/lib/Automation/Engine/Builder/CreateWorkflowFromTemplateController.php
@@ -3,6 +3,7 @@
 namespace MailPoet\Automation\Engine\Builder;
 
 use MailPoet\Automation\Engine\Data\Workflow;
+use MailPoet\Automation\Engine\Exceptions\InvalidStateException;
 use MailPoet\Automation\Engine\Storage\WorkflowStorage;
 use MailPoet\Automation\Engine\Storage\WorkflowTemplateStorage;
 use MailPoet\UnexpectedValueException;
@@ -32,7 +33,7 @@ class CreateWorkflowFromTemplateController {
     $workflowId = $this->storage->createWorkflow($template->getWorkflow());
     $workflow = $this->storage->getWorkflow($workflowId);
     if (!$workflow) {
-      throw UnexpectedValueException::create()->withMessage('Workflow not found.');
+      throw new InvalidStateException('Workflow not found.');
     }
     return $workflow;
   }