Asynchronous operation validate

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 encapsulates methods for Operation Model Object
  */
 class Operation extends DataObject implements OperationInterface
 {
+    /**
+     * @var OperationStatusValidator
+     */
+    private $operationStatusValidator;
+
+    /**
+     * Operation constructor.
+     *
+     * @param OperationStatusValidator $operationStatusValidator
+     * @param array $data
+     */
+    public function __construct(
+        OperationStatusValidator $operationStatusValidator,
+        array $data = []
+    ) {
+        $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
+     */
+    private $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
+     */
+    private $operationStatusPool;
+
+    /**
+     * OperationStatusValidator constructor.
+     *
+     * @param OperationStatusPool $operationStatusPool
+     */
+    public function __construct(OperationStatusPool $operationStatusPool)
+    {
+        $this->operationStatusPool = $operationStatusPool;
+    }
+
+    /**
+     * Validate method
+     *
+     * @param int $status
+     * @throws \InvalidArgumentException
+     * @return void
+     */
+    public function validate($status)
+    {
+        $statuses = $this->operationStatusPool->getStatuses();
+
+        if (!in_array($status, $statuses)) {
+            throw new \InvalidArgumentException('Invalid Operation Status.');
+        }
+    }
+}

app/code/Magento/AsynchronousOperations/Test/Unit/Model/OperationStatusValidatorTest.php

@@ -0,0 +1,155 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+namespace Magento\AsynchronousOperations\Test\Unit\Model;
+
+use Magento\AsynchronousOperations\Model\OperationStatusValidator;
+use Magento\AsynchronousOperations\Model\Operation;
+use Magento\AsynchronousOperations\Model\OperationStatusPool;
+use Magento\Framework\TestFramework\Unit\Helper\ObjectManager;
+use PHPUnit\Framework\TestCase;
+
+/**
+ * Class OperationStatusValidatorTest implements logic for testing Operation::setStatus() method
+ */
+class OperationStatusValidatorTest extends TestCase
+{
+    /**
+     * @var OperationStatusPool
+     */
+    private $operationStatusPool;
+
+    /**
+     * @var OperationStatusValidator
+     */
+    private $operationStatusValidator;
+
+    /**
+     * @var Operation
+     */
+    private $operation;
+
+    protected function setUp()
+    {
+        $this->operationStatusPool = $this->getMockBuilder(OperationStatusPool::class)
+            ->disableOriginalConstructor()
+            ->getMock();
+
+        $objectManager = new ObjectManager($this);
+
+        $this->operationStatusValidator = $objectManager->getObject(
+            OperationStatusValidator::class,
+            [
+                'operationStatusPool' => $this->operationStatusPool
+            ]
+        );
+
+        $this->operation = $objectManager->getObject(
+            Operation::class,
+            [
+                'operationStatusValidator' => $this->operationStatusValidator
+            ]
+        );
+    }
+
+    /**
+     * @param string $status
+     * @param array $statusPool
+     * @param string $expectedResult
+     * @dataProvider dataProviderForTestSetStatus
+     */
+    public function testSetStatus(
+        string $status,
+        array $statusPool,
+        string $expectedResult
+    ) {
+        $this->operationStatusPool
+            ->expects($this->any())
+            ->method('getStatuses')
+            ->willReturn($statusPool);
+
+        try {
+            $this->operation->setStatus($status);
+            $this->assertEquals($expectedResult, $this->operation->getStatus());
+        } catch (\Exception $exception) {
+            $this->assertEquals($expectedResult, $exception->getMessage());
+        }
+    }
+
+    /**
+     * @SuppressWarnings(PHPMD.ExcessiveMethodLength)
+     */
+    public function dataProviderForTestSetStatus()
+    {
+        return [
+            [
+                'status' => 0,
+                'statusPool' => [
+                    'complete' => 1,
+                    'retriablyFailed' => 2,
+                    'notRetriablyFailed' => 3,
+                    'open' => 4,
+                    'rejected' => 5
+                ],
+                'expectedResult' => 'Invalid Operation Status.'
+            ],
+            [
+                'status' => 1,
+                'statusPool' => [
+                    'complete' => 1,
+                    'retriablyFailed' => 2,
+                    'notRetriablyFailed' => 3,
+                    'open' => 4,
+                    'rejected' => 5
+                ],
+                'expectedResult' => 1
+            ],
+            [
+                'status' => 2,
+                'statusPool' => [
+                    'complete' => 1,
+                    'retriablyFailed' => 2,
+                    'notRetriablyFailed' => 3,
+                    'open' => 4,
+                    'rejected' => 5
+                ],
+                'expectedResult' => 2
+            ],
+            [
+                'status' => 3,
+                'statusPool' => [
+                    'complete' => 1,
+                    'retriablyFailed' => 2,
+                    'notRetriablyFailed' => 3,
+                    'open' => 4,
+                    'rejected' => 5
+                ],
+                'expectedResult' => 3
+            ],
+            [
+                'status' => 4,
+                'statusPool' => [
+                    'complete' => 1,
+                    'retriablyFailed' => 2,
+                    'notRetriablyFailed' => 3,
+                    'open' => 4,
+                    'rejected' => 5
+                ],
+                'expectedResult' => 4
+            ],
+            [
+                'status' => 5,
+                'statusPool' => [
+                    'complete' => 1,
+                    'retriablyFailed' => 2,
+                    'notRetriablyFailed' => 3,
+                    'open' => 4,
+                    'rejected' => 5
+                ],
+                'expectedResult' => 5
+            ]
+        ];
+    }
+}

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"/>