CouponManagementServiceTest.php

Go to the documentation of this file.

<?php
namespace Magento\SalesRule\Test\Unit\Model\Service;

class CouponManagementServiceTest extends \PHPUnit\Framework\TestCase
{
    protected $model;

    protected $couponFactory;

    protected $ruleFactory;

    protected $collectionFactory;

    protected $couponGenerator;

    protected $resourceModel;

    protected $couponMassDeleteResultFactory;

    protected $couponMassDeleteResult;

    protected $objectManager;

    protected function setUp()
    {
        $this->objectManager = new \Magento\Framework\TestFramework\Unit\Helper\ObjectManager($this);

        $className = \Magento\SalesRule\Model\CouponFactory::class;
        $this->couponFactory = $this->createMock($className);

        $className = \Magento\SalesRule\Model\RuleFactory::class;
        $this->ruleFactory = $this->createPartialMock($className, ['create']);

        $className = \Magento\SalesRule\Model\ResourceModel\Coupon\CollectionFactory::class;
        $this->collectionFactory = $this->createPartialMock($className, ['create']);

        $className = \Magento\SalesRule\Model\Coupon\Massgenerator::class;
        $this->couponGenerator = $this->createMock($className);

        $className = \Magento\SalesRule\Model\Spi\CouponResourceInterface::class;
        $this->resourceModel = $this->createMock($className);

        $className = \Magento\SalesRule\Api\Data\CouponMassDeleteResultInterface::class;
        $this->couponMassDeleteResult = $this->createMock($className);

        $className = \Magento\SalesRule\Api\Data\CouponMassDeleteResultInterfaceFactory::class;
        $this->couponMassDeleteResultFactory = $this->createPartialMock($className, ['create']);
        $this->couponMassDeleteResultFactory
            ->expects($this->any())
            ->method('create')
            ->willReturn($this->couponMassDeleteResult);

        $this->model = $this->objectManager->getObject(
            \Magento\SalesRule\Model\Service\CouponManagementService::class,
            [
                'couponFactory' => $this->couponFactory,
                'ruleFactory' => $this->ruleFactory,
                'collectionFactory' => $this->collectionFactory,
                'couponGenerator' => $this->couponGenerator,
                'resourceModel' => $this->resourceModel,
                'couponMassDeleteResultFactory' => $this->couponMassDeleteResultFactory,
            ]
        );
    }

    public function testGenerate()
    {
        $className = \Magento\SalesRule\Model\Data\CouponGenerationSpec::class;
        $couponSpec = $this->createPartialMock(
            $className,
            ['getRuleId', 'getQuantity', 'getFormat', 'getLength', 'setData']
        );

        $couponSpec->expects($this->atLeastOnce())->method('getRuleId')->willReturn(1);
        $couponSpec->expects($this->once())->method('getQuantity')->willReturn(1);
        $couponSpec->expects($this->once())->method('getFormat')->willReturn('num');
        $couponSpec->expects($this->once())->method('getLength')->willReturn(1);

        $this->couponGenerator->expects($this->atLeastOnce())->method('setData');
        $this->couponGenerator->expects($this->once())->method('validateData')->willReturn(true);
        $this->couponGenerator->expects($this->once())->method('generatePool');
        $this->couponGenerator->expects($this->once())->method('getGeneratedCodes')->willReturn([]);

        $rule = $this->createPartialMock(
            \Magento\SalesRule\Model\Rule::class,
            ['load', 'getRuleId', 'getToDate', 'getUsesPerCoupon', 'getUsesPerCustomer', 'getUseAutoGeneration']
        );

        $rule->expects($this->any())->method('load')->willReturnSelf();
        $rule->expects($this->any())->method('getRuleId')->willReturn(1);
        $rule->expects($this->any())->method('getToDate')->willReturn('2015-07-31 00:00:00');
        $rule->expects($this->any())->method('getUsesPerCoupon')->willReturn(20);
        $rule->expects($this->any())->method('getUsesPerCustomer')->willReturn(5);
        $rule->expects($this->any())->method('getUseAutoGeneration')->willReturn(true);

        $this->ruleFactory->expects($this->any())->method('create')->willReturn($rule);

        $result =  $this->model->generate($couponSpec);
        $this->assertEquals([], $result);
    }

    public function testGenerateValidationException()
    {
        $className = \Magento\SalesRule\Api\Data\CouponGenerationSpecInterface::class;
        $couponSpec = $this->createMock($className);

        $rule = $this->createPartialMock(\Magento\SalesRule\Model\Rule::class, ['load', 'getRuleId']);

        $rule->expects($this->any())->method('load')->willReturnSelf();
        $rule->expects($this->any())->method('getRuleId')->willReturn(1);

        $this->ruleFactory->expects($this->any())->method('create')->willReturn($rule);

        $this->couponGenerator->expects($this->once())->method('validateData')
            ->willThrowException(new \Magento\Framework\Exception\InputException());
        $this->expectException(\Magento\Framework\Exception\InputException::class);

        $this->model->generate($couponSpec);
    }

    public function testGenerateLocalizedException()
    {
        $className = \Magento\SalesRule\Api\Data\CouponGenerationSpecInterface::class;
        $couponSpec = $this->createMock($className);

        $rule = $this->createPartialMock(
            \Magento\SalesRule\Model\Rule::class,
            ['load', 'getRuleId', 'getUseAutoGeneration']
        );
        $rule->expects($this->any())->method('load')->willReturnSelf();
        $rule->expects($this->any())->method('getRuleId')->willReturn(1);
        $rule->expects($this->once())->method('getUseAutoGeneration')
            ->willThrowException(
                new \Magento\Framework\Exception\LocalizedException(
                    __('Error occurred when generating coupons: %1', '1')
                )
            );
        $this->ruleFactory->expects($this->any())->method('create')->willReturn($rule);

        $this->couponGenerator->expects($this->once())->method('validateData')->willReturn(true);

        $this->expectException(\Magento\Framework\Exception\LocalizedException::class);

        $this->model->generate($couponSpec);
    }

    public function testGenerateNoSuchEntity()
    {
        $className = \Magento\SalesRule\Api\Data\CouponGenerationSpecInterface::class;
        $couponSpec = $this->createMock($className);

        $rule = $this->createPartialMock(\Magento\SalesRule\Model\Rule::class, ['load', 'getRuleId']);

        $rule->expects($this->any())->method('load')->willReturnSelf();
        $rule->expects($this->any())->method('getRuleId')->willReturn(false);

        $this->ruleFactory->expects($this->any())->method('create')->willReturn($rule);

        $this->couponGenerator->expects($this->once())->method('validateData')->willReturn(true);

        $this->expectException(\Magento\Framework\Exception\LocalizedException::class);

        $this->model->generate($couponSpec);
    }

    public function testDeleteByIdsIgnore()
    {
        $ids = [1, 2, 3];

        $className = \Magento\SalesRule\Model\Coupon::class;
        $coupon = $this->createMock($className);
        $coupon->expects($this->exactly(3))->method('delete');

        $className = \Magento\SalesRule\Model\ResourceModel\Coupon\Collection::class;
        $couponCollection = $this->createMock($className);

        $couponCollection->expects($this->once())->method('addFieldToFilter')->willReturnSelf();
        $couponCollection->expects($this->once())->method('getItems')->willReturn([$coupon, $coupon, $coupon]);
        $this->collectionFactory->expects($this->once())->method('create')->willReturn($couponCollection);

        $this->couponMassDeleteResult->expects($this->once())->method('setFailedItems')->willReturnSelf();
        $this->couponMassDeleteResult->expects($this->once())->method('setMissingItems')->willReturnSelf();

        $this->model->deleteByIds($ids, true);
    }

    public function testDeleteByAnyNoIgnore()
    {
        $ids = [1, 2, 3];

        $className = \Magento\SalesRule\Model\ResourceModel\Coupon\Collection::class;
        $couponCollection = $this->createMock($className);
        $couponCollection->expects($this->once())->method('addFieldToFilter')->willReturnSelf();
        $this->collectionFactory->expects($this->once())->method('create')->willReturn($couponCollection);

        $this->expectException(\Magento\Framework\Exception\LocalizedException::class);

        $this->model->deleteByIds($ids, false);
    }

    public function testDeleteByAnyExceptionOnDelete()
    {
        $ids = [1, 2, 3];

        $className = \Magento\SalesRule\Model\Coupon::class;
        $coupon = $this->createMock($className);
        $coupon->expects($this->any())->method('delete')->willThrowException(new \Exception());

        $className = \Magento\SalesRule\Model\ResourceModel\Coupon\Collection::class;
        $couponCollection = $this->createMock($className);
        $couponCollection->expects($this->once())->method('addFieldToFilter')->willReturnSelf();
        $couponCollection->expects($this->once())->method('getItems')->willReturn([$coupon, $coupon, $coupon]);
        $this->collectionFactory->expects($this->once())->method('create')->willReturn($couponCollection);

        $this->couponMassDeleteResult->expects($this->once())->method('setFailedItems')->willReturnSelf();
        $this->couponMassDeleteResult->expects($this->once())->method('setMissingItems')->willReturnSelf();

        $this->model->deleteByIds($ids, true);
    }

    public function testDeleteByCodes()
    {
        $ids = [1, 2, 3];

        $className = \Magento\SalesRule\Model\Coupon::class;
        $coupon = $this->createMock($className);
        $coupon->expects($this->exactly(3))->method('delete');

        $className = \Magento\SalesRule\Model\ResourceModel\Coupon\Collection::class;
        $couponCollection = $this->createMock($className);

        $couponCollection->expects($this->once())->method('addFieldToFilter')->willReturnSelf();
        $couponCollection->expects($this->once())->method('getItems')->willReturn([$coupon, $coupon, $coupon]);
        $this->collectionFactory->expects($this->once())->method('create')->willReturn($couponCollection);

        $this->couponMassDeleteResult->expects($this->once())->method('setFailedItems')->willReturnSelf();
        $this->couponMassDeleteResult->expects($this->once())->method('setMissingItems')->willReturnSelf();

        $this->model->deleteByCodes($ids, true);
    }
}