AbstractCollectorPositionsTest.php

Go to the documentation of this file.

<?php
namespace Magento\Sales\Model;

abstract class AbstractCollectorPositionsTest extends \PHPUnit\Framework\TestCase
{
    public function testCollectorPosition($collectorCode, $configType, array $before, array $after)
    {
        $allCollectors = self::_getConfigCollectors($configType);
        $collectorCodes = array_keys($allCollectors);
        $collectorPos = array_search($collectorCode, $collectorCodes);
        $this->assertNotSame(false, $collectorPos, "'{$collectorCode}' total collector is not found");

        foreach ($before as $compareWithCode) {
            $compareWithPos = array_search($compareWithCode, $collectorCodes);
            if ($compareWithPos === false) {
                continue;
            }
            $this->assertLessThan(
                $compareWithPos,
                $collectorPos,
                "The '{$collectorCode}' collector must go before '{$compareWithCode}'"
            );
        }

        foreach ($after as $compareWithCode) {
            $compareWithPos = array_search($compareWithCode, $collectorCodes);
            if ($compareWithPos === false) {
                continue;
            }
            $this->assertGreaterThan(
                $compareWithPos,
                $collectorPos,
                "The '{$collectorCode}' collector must go after '{$compareWithCode}'"
            );
        }
    }

    protected static function _getConfigCollectors($configType)
    {
        switch ($configType) {
            case 'quote':
                $configClass = \Magento\Quote\Model\Quote\Address\Total\Collector::class;
                $methodGetCollectors = 'getCollectors';
                break;
            case 'invoice':
                $configClass = \Magento\Sales\Model\Order\Invoice\Config::class;
                $methodGetCollectors = 'getTotalModels';
                break;
            case 'creditmemo':
                $configClass = \Magento\Sales\Model\Order\Creditmemo\Config::class;
                $methodGetCollectors = 'getTotalModels';
                break;
            default:
                throw new \InvalidArgumentException('Unknown config type: ' . $configType);
        }
        $config = \Magento\TestFramework\Helper\Bootstrap::getObjectManager()->create($configClass);
        return $config->{$methodGetCollectors}();
    }

    abstract public function collectorPositionDataProvider();
}