tddwizard / magento2-fixtures

use Magento\Framework\ObjectManagerInterface;

use Magento\Framework\Registry;

class ProductFixtureRollback

{

    /**

     * @var Registry

     */

    private $registry;

    /**

     * @var ProductRepositoryInterface

     */

    private $productRepository;

    public function __construct(Registry $registry, ProductRepositoryInterface $productRepository)

    {

        $this->registry = $registry;

        $this->productRepository = $productRepository;

    }

    public static function create(ObjectManagerInterface $objectManager = null)

    {

        if ($objectManager === null) {

            $objectManager = \Magento\TestFramework\Helper\Bootstrap::getObjectManager();

        }

        return new self(

            $objectManager->get(Registry::class),

            $objectManager->get(ProductRepositoryInterface::class)

        );

    }

    public function execute(ProductFixture ...$productFixtures)

    {

        $this->registry->unregister('isSecureArea');

        $this->registry->register('isSecureArea', true);

        foreach ($productFixtures as $productFixture) {

            $this->productRepository->deleteById($productFixture->getSku());

        }

        $this->registry->unregister('isSecureArea');

    }

}

use Magento\Framework\ObjectManagerInterface;

use Magento\Framework\Registry;

class CustomerFixtureRollback

{

    /**

     * @var Registry

     */

    private $registry;

    /**

     * @var CustomerRepositoryInterface

     */

    private $customerRepository;

    public function __construct(Registry $registry, CustomerRepositoryInterface $customerRepository)

    {

        $this->registry = $registry;

        $this->customerRepository = $customerRepository;

    }

    public static function create(ObjectManagerInterface $objectManager = null)

    {

        if ($objectManager === null) {

            $objectManager = \Magento\TestFramework\Helper\Bootstrap::getObjectManager();

        }

        return new self(

            $objectManager->get(Registry::class),

            $objectManager->get(CustomerRepositoryInterface::class)

        );

    }

    public function execute(CustomerFixture ...$customerFixtures)

    {

        $this->registry->unregister('isSecureArea');

        $this->registry->register('isSecureArea', true);

        foreach ($customerFixtures as $customerFixture) {

            $this->customerRepository->deleteById($customerFixture->getId());

        }

        $this->registry->unregister('isSecureArea');

    }

}

use Magento\Framework\ObjectManagerInterface;

use Magento\Framework\Registry;

class CategoryFixtureRollback

{

    /**

     * @var Registry

     */

    private $registry;

    /**

     * @var CategoryRepositoryInterface

     */

    private $categoryRepository;

    public function __construct(Registry $registry, CategoryRepositoryInterface $categoryRepository)

    {

        $this->registry = $registry;

        $this->categoryRepository = $categoryRepository;

    }

    public static function create(ObjectManagerInterface $objectManager = null)

    {

        if ($objectManager === null) {

            $objectManager = \Magento\TestFramework\Helper\Bootstrap::getObjectManager();

        }

        return new self(

            $objectManager->get(Registry::class),

            $objectManager->get(CategoryRepositoryInterface::class)

        );

    }

    public function execute(CategoryFixture ...$categoryFixtures)

    {

        $this->registry->unregister('isSecureArea');

        $this->registry->register('isSecureArea', true);

        foreach ($categoryFixtures as $categoryFixture) {

            $this->categoryRepository->deleteByIdentifier($categoryFixture->getId());

        }

        $this->registry->unregister('isSecureArea');

    }

}