MSI: Unit test example

app/code/Magento/Inventory/Test/Unit/Model/SourceRepositoryTest.php

@@ -0,0 +1,150 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+namespace Magento\Inventory\Test\Unit\Model;
+
+use Magento\Framework\Api\SearchCriteriaInterface;
+use Magento\Framework\Exception\CouldNotSaveException;
+use Magento\Framework\Exception\NoSuchEntityException;
+use Magento\Framework\TestFramework\Unit\Helper\ObjectManager;
+use Magento\Inventory\Model\Source\Command\GetInterface;
+use Magento\Inventory\Model\Source\Command\GetListInterface;
+use Magento\Inventory\Model\Source\Command\SaveInterface;
+use Magento\Inventory\Model\SourceRepository;
+use Magento\InventoryApi\Api\Data\SourceInterface;
+use Magento\InventoryApi\Api\Data\SourceSearchResultsInterface;
+
+class SourceRepositoryTest extends \PHPUnit_Framework_TestCase
+{
+    /**
+     * @var SaveInterface|\PHPUnit_Framework_MockObject_MockObject
+     */
+    private $commandSave;
+
+    /**
+     * @var GetInterface|\PHPUnit_Framework_MockObject_MockObject
+     */
+    private $commandGet;
+
+    /**
+     * @var GetListInterface|\PHPUnit_Framework_MockObject_MockObject
+     */
+    private $commandGetList;
+
+    /**
+     * @var SourceInterface|\PHPUnit_Framework_MockObject_MockObject
+     */
+    private $source;
+
+    /**
+     * @var SourceSearchResultsInterface|\PHPUnit_Framework_MockObject_MockObject
+     */
+    private $searchResult;
+
+    /**
+     * @var SourceRepository
+     */
+    private $sourceRepository;
+
+    protected function setUp()
+    {
+        $this->commandSave = $this->getMockBuilder(SaveInterface::class)->getMock();
+        $this->commandGet = $this->getMockBuilder(GetInterface::class)->getMock();
+        $this->commandGetList = $this->getMockBuilder(GetListInterface::class)->getMock();
+        $this->source = $this->getMockBuilder(SourceInterface::class)->getMock();
+        $this->searchResult = $this->getMockBuilder(SourceSearchResultsInterface::class)->getMock();
+
+        $this->sourceRepository = (new ObjectManager($this))->getObject(
+            SourceRepository::class,
+            [
+                'commandSave' => $this->commandSave,
+                'commandGet' => $this->commandGet,
+                'commandGetList' => $this->commandGetList,
+            ]
+        );
+    }
+
+    public function testSave()
+    {
+        $sourceId = 42;
+
+        $this->commandSave
+            ->expects($this->once())
+            ->method('execute')
+            ->with($this->source)
+            ->willReturn($sourceId);
+
+        self::assertEquals($sourceId, $this->sourceRepository->save($this->source));
+    }
+
+    /**
+     * @expectedException \Magento\Framework\Exception\CouldNotSaveException
+     * @expectedExceptionMessage Some error
+     */
+    public function testSaveWithCouldNotSaveException()
+    {
+        $this->commandSave
+            ->expects($this->once())
+            ->method('execute')
+            ->with($this->source)
+            ->willThrowException(new CouldNotSaveException(__('Some error')));
+
+        $this->sourceRepository->save($this->source);
+    }
+
+    public function testGet()
+    {
+        $sourceId = 42;
+
+        $this->commandGet
+            ->expects($this->once())
+            ->method('execute')
+            ->with($sourceId)
+            ->willReturn($this->source);
+
+        self::assertEquals($this->source, $this->sourceRepository->get($sourceId));
+    }
+
+    /**
+     * @expectedException \Magento\Framework\Exception\NoSuchEntityException
+     * @expectedExceptionMessage Some error
+     */
+    public function testGetWithNoSuchEntityException()
+    {
+        $sourceId = 42;
+
+        $this->commandGet
+            ->expects($this->once())
+            ->method('execute')
+            ->with($sourceId)
+            ->willThrowException(new NoSuchEntityException(__('Some error')));
+
+        $this->sourceRepository->get($sourceId);
+    }
+
+    public function testGetListWithoutSearchCriteria()
+    {
+        $this->commandGetList
+            ->expects($this->once())
+            ->method('execute')
+            ->with(null)
+            ->willReturn($this->searchResult);
+
+        self::assertEquals($this->searchResult, $this->sourceRepository->getList());
+    }
+
+    public function testGetListWithSearchCriteria()
+    {
+        $searchCriteria = $this->getMockBuilder(SearchCriteriaInterface::class)->getMock();
+
+        $this->commandGetList
+            ->expects($this->once())
+            ->method('execute')
+            ->with($searchCriteria)
+            ->willReturn($this->searchResult);
+
+        self::assertEquals($this->searchResult, $this->sourceRepository->getList($searchCriteria));
+    }
+}