CustomerGroupsFixture.php

Go to the documentation of this file.

<?php
namespace Magento\Setup\Fixtures;

use Magento\Customer\Api\Data\GroupInterfaceFactory;
use Magento\Customer\Api\GroupRepositoryInterface;
use Magento\Customer\Model\ResourceModel\Group\CollectionFactory;

class CustomerGroupsFixture extends Fixture
{
    const DEFAULT_TAX_CLASS_ID = 3;

    protected $priority = 60;

    private $groupCollectionFactory;

    private $groupRepository;

    private $groupFactory;

    public function __construct(
        FixtureModel $fixtureModel,
        CollectionFactory $groupCollectionFactory,
        GroupRepositoryInterface $groupRepository,
        GroupInterfaceFactory $groupFactory
    ) {
        parent::__construct($fixtureModel);
        $this->groupCollectionFactory = $groupCollectionFactory;
        $this->groupRepository = $groupRepository;
        $this->groupFactory = $groupFactory;
    }

    public function execute()
    {
        $existingCustomerGroupsCount = $this->groupCollectionFactory->create()->getSize();
        $customerGroupsCount = $this->fixtureModel->getValue('customer_groups', 0);
        if ($customerGroupsCount < 1) {
            return;
        }

        for ($i = $existingCustomerGroupsCount; $i <  $customerGroupsCount; ++$i) {
            $groupDataObject = $this->groupFactory->create();
            $groupDataObject
                ->setCode('customer_group_' . $i)
                ->setTaxClassId(self::DEFAULT_TAX_CLASS_ID);
            $this->groupRepository->save($groupDataObject);
        }
    }

    public function getActionTitle()
    {
        return 'Generating customer groups';
    }

    public function introduceParamLabels()
    {
        return [
            'customer_groups' => 'Customer groups'
        ];
    }
}