Add OperationStatusPool and OperationStatusValidator

Author: sedonik

app/code/Magento/AsynchronousOperations/Model/Operation.php

@@ -6,13 +6,33 @@
 namespace Magento\AsynchronousOperations\Model;
 
 use Magento\AsynchronousOperations\Api\Data\OperationInterface;
+use Magento\AsynchronousOperations\Model\OperationStatusValidator;
 use Magento\Framework\DataObject;
 
 /**
  * Class Operation
  */
 class Operation extends DataObject implements OperationInterface
 {
+    /**
+     * @var OperationStatusValidator
+     */
+    protected $operationStatusValidator;
+
+    /**
+     * Operation constructor.
+     *
+     * @param array $data
+     * @param OperationStatusValidator $operationStatusValidator
+     */
+    public function __construct(
+        array $data = [],
+        OperationStatusValidator $operationStatusValidator
+    ) {
+        $this->operationStatusValidator = $operationStatusValidator;
+        parent::__construct($data);
+    }
+
     /**
      * @inheritDoc
      */
@@ -106,6 +126,7 @@ public function getStatus()
      */
     public function setStatus($status)
     {
+        $this->operationStatusValidator->validate($status);
         return $this->setData(self::STATUS, $status);
     }
 

app/code/Magento/AsynchronousOperations/Model/OperationStatusPool.php

@@ -0,0 +1,37 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+namespace Magento\AsynchronousOperations\Model;
+
+/**
+ * Class OperationStatusPool
+ *
+ * Pool of statuses that require validate
+ */
+class OperationStatusPool
+{
+    /**
+     * @var array
+     */
+    protected $statuses;
+
+    /**
+     * @param array $statuses
+     */
+    public function __construct(array $statuses = [])
+    {
+        $this->statuses = $statuses;
+    }
+
+    /**
+     * Retrieve statuses that require validate
+     *
+     * @return array
+     */
+    public function getStatuses()
+    {
+        return $this->statuses;
+    }
+}

app/code/Magento/AsynchronousOperations/Model/OperationStatusValidator.php

@@ -0,0 +1,47 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+namespace Magento\AsynchronousOperations\Model;
+
+use Magento\AsynchronousOperations\Model\OperationStatusPool;
+use Magento\Framework\Exception\NoSuchEntityException;
+use Doctrine\Instantiator\Exception\InvalidArgumentException;
+
+/**
+ * Class OperationStatusValidator to validate operation status
+ */
+class OperationStatusValidator
+{
+    /**
+     * @var OperationStatusPool
+     */
+    protected $operationStatusPool;
+
+    /**
+     * OperationStatusValidator constructor.
+     *
+     * @param OperationStatusPool $operationStatusPool
+     */
+    public function __construct(OperationStatusPool $operationStatusPool)
+    {
+        $this->operationStatusPool = $operationStatusPool;
+    }
+
+    /**
+     * Validate method
+     *
+     * @param $status
+     */
+    public function validate($status)
+    {
+        $statuses = $this->operationStatusPool->getStatuses();
+
+        if (!in_array($status, $statuses)) {
+            throw new \InvalidArgumentException('Invalid Operation Status.');
+        }
+
+        return;
+    }
+}

app/code/Magento/AsynchronousOperations/etc/di.xml

@@ -80,6 +80,17 @@
             </argument>
         </arguments>
     </type>
+    <type name="Magento\AsynchronousOperations\Model\OperationStatusPool">
+        <arguments>
+            <argument name="statuses" xsi:type="array">
+                <item name="complete" xsi:type="string">1</item>
+                <item name="retriablyFailed" xsi:type="string">2</item>
+                <item name="notRetriablyFailed" xsi:type="string">3</item>
+                <item name="open" xsi:type="string">4</item>
+                <item name="rejected" xsi:type="string">5</item>
+            </argument>
+        </arguments>
+    </type>
     <virtualType
             name="Magento\AsynchronousOperations\Ui\Component\DataProvider"
             type="Magento\Framework\View\Element\UiComponent\DataProvider\DataProvider"/>