<?php
/**
 * Copyright 2024 Adobe
 * All rights reserved.
 * See COPYING.txt for license details.
 */
declare(strict_types=1);

namespace Magento\InventorySourceSelectionApi\Test\Integration;

use Magento\InventoryApi\Api\Data\SourceItemInterface;
use Magento\InventorySourceSelectionApi\Model\GetInStockSourceItemsBySkusAndSortedSource;
use Magento\TestFramework\Helper\Bootstrap;
use PHPUnit\Framework\TestCase;

class GetSourceItemsBySkusAndSortedSourceTest extends TestCase
{
    /**
     * @var GetInStockSourceItemsBySkusAndSortedSource
     */
    private $subject;

    protected function setUp(): void
    {
        parent::setUp();

        $this->subject = Bootstrap::getObjectManager()->get(GetInStockSourceItemsBySkusAndSortedSource::class);
    }

    /**
     * @return array
     */
    public static function shouldReturnSortedSourceItemsDataProvider(): array
    {
        return [
            [
                ['SKU-1', 'SKU-2', 'SKU-3'],
                ['eu-1', 'eu-2', 'eu-3'],
                [
                    'eu-1/SKU-1' => 5.5,
                    'eu-2/SKU-1' => 3.0,
                ]
            ],
            [
                ['SKU-1', 'SKU-2', 'SKU-3'],
                ['eu-3', 'eu-2', 'eu-1'],
                [
                    'eu-2/SKU-1' => 3.0,
                    'eu-1/SKU-1' => 5.5,
                ]
            ],
            [
                ['SKU-1', 'SKU-2', 'SKU-3', 'SKU-6'],
                ['eu-3', 'eu-2', 'eu-1'],
                [
                    'eu-2/SKU-1' => 3.0,
                    'eu-1/SKU-1' => 5.5,
                    'eu-1/SKU-6' => 0.0
                ]
            ]
        ];
    }

    /**
     * @magentoDataFixture Magento_InventoryApi::Test/_files/sources.php
     * @magentoDataFixture Magento_InventoryApi::Test/_files/source_items.php
     * @magentoDataFixture Magento_InventoryApi::Test/_files/stocks.php
     * @magentoDataFixture Magento_InventoryApi::Test/_files/stock_source_links.php
     * @dataProvider shouldReturnSortedSourceItemsDataProvider
     * @param array $skus
     * @param array $sortedSourceCodes
     * @param array $expected
     */
    public function testShouldReturnSortedSourceItems(array $skus, array $sortedSourceCodes, array $expected): void
    {
        $sourceItems = $this->subject->execute($skus, $sortedSourceCodes);

        self::assertCount(count($expected), $sourceItems);

        $keys = [];
        foreach ($sourceItems as $sourceItem) {
            $key = $sourceItem->getSourceCode() . '/' . $sourceItem->getSku();
            $keys[] = $key;
            self::assertSame($expected[$key], $sourceItem->getQuantity());
            self::assertSame(SourceItemInterface::STATUS_IN_STOCK, $sourceItem->getStatus());
        }

        self::assertSame(array_keys($expected), $keys, 'Sources sorting is not preserved');
    }

    /**
     * @return array
     */
    public static function numericOnlySkusDataProvider(): array
    {
        return [
            [
                [123, 456],
                ['eu-1'],
                [
                    'eu-1/123',
                    'eu-1/456',
                ]
            ],
            [
                [123, '456DEF'],
                ['eu-1'],
                [
                    'eu-1/123',
                    'eu-1/456DEF',
                ]
            ]
        ];
    }

    /**
     * @magentoDataFixture Magento_InventoryApi::Test/_files/sources.php
     * @magentoDataFixture Magento_InventoryApi::Test/_files/source_items_numeric_skus.php
     * @magentoDataFixture Magento_InventoryApi::Test/_files/stocks.php
     * @magentoDataFixture Magento_InventoryApi::Test/_files/stock_source_links.php
     * @dataProvider numericOnlySkusDataProvider
     * @param array $skus
     * @param array $sortedSourceCodes
     * @param array $expected
     */
    public function testNumericOnlySkus(array $skus, array $sortedSourceCodes, array $expected): void
    {
        $sourceItems = $this->subject->execute($skus, $sortedSourceCodes);

        self::assertCount(count($expected), $sourceItems, 'Expected and actual product count mismatch');

        $keys = [];
        foreach ($sourceItems as $sourceItem) {
            $keys[] = $sourceItem->getSourceCode() . '/' . $sourceItem->getSku();
        }
        self::assertSame($expected, $keys, 'Expected and actual product SKUs mismatch');
    }
}