ENGCOM-6509: [ImportExport] Cover Export Source Model by Unit Test #26140

Author: VladimirZaets

app/code/Magento/ImportExport/Test/Unit/Model/Source/Export/EntityTest.php

@@ -0,0 +1,95 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+
+declare(strict_types=1);
+
+namespace Magento\ImportExport\Test\Unit\Model\Source\Export;
+
+use Magento\Framework\TestFramework\Unit\Helper\ObjectManager;
+use Magento\ImportExport\Model\Export\ConfigInterface;
+use Magento\ImportExport\Model\Source\Export\Entity;
+use PHPUnit\Framework\MockObject\MockObject;
+use PHPUnit\Framework\TestCase;
+
+class EntityTest extends TestCase
+{
+    /**
+     * @var ConfigInterface|MockObject
+     */
+    private $exportConfigMock;
+
+    /**
+     * @var Entity
+     */
+    private $model;
+
+    /**
+     * Setup environment for test
+     */
+    protected function setUp()
+    {
+        $this->exportConfigMock = $this->createMock(ConfigInterface::class);
+
+        $objectManager = new ObjectManager($this);
+        $this->model = $objectManager->getObject(
+            Entity::class,
+            [
+                'exportConfig' => $this->exportConfigMock
+            ]
+        );
+    }
+
+    /**
+     * Test toOptionArray with data provider
+     *
+     * @param array $entities
+     * @param array $expected
+     * @dataProvider toOptionArrayDataProvider
+     */
+    public function testToOptionArray($entities, $expected)
+    {
+        $this->exportConfigMock->expects($this->any())->method('getEntities')->willReturn($entities);
+
+        $this->assertEquals($expected, $this->model->toOptionArray());
+    }
+
+    /**
+     * Data Provider for test toOptionArray
+     *
+     * @return array
+     */
+    public function toOptionArrayDataProvider()
+    {
+        return [
+            'Empty Entity' => [
+                [],
+                [
+                    [
+                        'label' => (string)__('-- Please Select --'),
+                        'value' => ''
+                    ]
+                ]
+            ],
+            'Has entities' => [
+                [
+                    'entity1' => [
+                        'label' => 'Entity 1'
+                    ]
+                ],
+                [
+                    [
+                        'label' => (string)__('-- Please Select --'),
+                        'value' => ''
+                    ],
+                    [
+                        'label' => (string)__('Entity 1'),
+                        'value' => 'entity1'
+                    ]
+                ]
+            ]
+        ];
+    }
+}

app/code/Magento/ImportExport/Test/Unit/Model/Source/Export/FormatTest.php

@@ -0,0 +1,86 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+
+declare(strict_types=1);
+
+namespace Magento\ImportExport\Test\Unit\Model\Source\Export;
+
+use Magento\Framework\TestFramework\Unit\Helper\ObjectManager;
+use Magento\ImportExport\Model\Export\ConfigInterface;
+use Magento\ImportExport\Model\Source\Export\Format;
+use PHPUnit\Framework\MockObject\MockObject;
+use PHPUnit\Framework\TestCase;
+
+class FormatTest extends TestCase
+{
+    /**
+     * @var ConfigInterface|MockObject
+     */
+    private $exportConfigMock;
+
+    /**
+     * @var Format
+     */
+    private $model;
+
+    /**
+     * Setup environment for test
+     */
+    protected function setUp()
+    {
+        $this->exportConfigMock = $this->createMock(ConfigInterface::class);
+
+        $objectManager = new ObjectManager($this);
+        $this->model = $objectManager->getObject(
+            Format::class,
+            [
+                'exportConfig' => $this->exportConfigMock
+            ]
+        );
+    }
+
+    /**
+     * Test toOptionArray with data provider
+     *
+     * @param array $fileFormats
+     * @param array $expected
+     * @dataProvider toOptionArrayDataProvider
+     */
+    public function testToOptionArray($fileFormats, $expected)
+    {
+        $this->exportConfigMock->expects($this->any())->method('getFileFormats')->willReturn($fileFormats);
+
+        $this->assertEquals($expected, $this->model->toOptionArray());
+    }
+
+    /**
+     * Data Provider for test toOptionArray
+     *
+     * @return array
+     */
+    public function toOptionArrayDataProvider()
+    {
+        return [
+            'Empty file format' => [
+                [],
+                []
+            ],
+            'Has file format' => [
+                [
+                    'fileFormat1' => [
+                        'label' => 'File Format 1'
+                    ]
+                ],
+                [
+                    [
+                        'label' => (string)__('File Format 1'),
+                        'value' => 'fileFormat1'
+                    ]
+                ]
+            ]
+        ];
+    }
+}