Created test for generator

Author: ProkopovVitaliy

testData/actions/generation/generator/EavAttributeSetupPatchGenerator/generateFile/AddTestAttribute.php

@@ -0,0 +1,100 @@
+<?php
+
+namespace Foo\Bar\Setup\Patch\Data;
+
+use Magento\Eav\Setup\EavSetup;
+use Magento\Eav\Setup\EavSetupFactory;
+use Magento\Framework\Setup\ModuleDataSetupInterface;
+use Magento\Framework\Setup\Patch\DataPatchInterface;
+
+class AddTestAttribute implements DataPatchInterface
+{
+
+    /**
+     * @var ModuleDataSetupInterface
+     */
+    private $moduleDataSetup;
+
+    /**
+     * @var EavSetupFactory
+     */
+    private $eavSetupFactory;
+
+    /**
+     * AddRecommendedAttribute constructor.
+     *
+     * @param ModuleDataSetupInterface $moduleDataSetup
+     * @param EavSetupFactory $eavSetupFactory
+     */
+    public function __construct(
+        ModuleDataSetupInterface $moduleDataSetup,
+        EavSetupFactory $eavSetupFactory
+    )
+    {
+        $this->moduleDataSetup = $moduleDataSetup;
+        $this->eavSetupFactory = $eavSetupFactory;
+    }
+
+    /**
+     * Get array of patches that have to be executed prior to this.
+     *
+     * Example of implementation:
+     *
+     * [
+     *      \Vendor_Name\Module_Name\Setup\Patch\Patch1::class,
+     *      \Vendor_Name\Module_Name\Setup\Patch\Patch2::class
+     * ]
+     *
+     * @return string[]
+     */
+    public static function getDependencies()
+    {
+        return [];
+    }
+
+    /**
+     * Get aliases (previous names) for the patch.
+     *
+     * @return string[]
+     */
+    public function getAliases()
+    {
+        return [];
+    }
+
+    /**
+     * Run code inside patch
+     * If code fails, patch must be reverted, in case when we are speaking about schema - then under revert
+     * means run PatchInterface::revert()
+     *
+     * If we speak about data, under revert means: $transaction->rollback()
+     *
+     * @return $this
+     */
+    public function apply()
+    {
+        /** @var EavSetup $eavSetup */
+        $eavSetup = $this->eavSetupFactory->create(['setup' => $this->moduleDataSetup]);
+
+        $eavSetup->addAttribute(
+            \Magento\Catalog\Model\Product::ENTITY,
+            'test',
+            [
+                'is_visible_in_grid' => false,
+                'is_html_allowed_on_front' => false,
+                'visible_on_front' => false,
+                'visible' => true,
+                'global' => \Magento\Eav\Model\Entity\Attribute\ScopedAttributeInterface::SCOPE_GLOBAL,
+                'label' => 'Test Label',
+                'source' => null,
+                'type' => 'static',
+                'is_used_in_grid' => false,
+                'required' => false,
+                'input' => 'boolean',
+                'is_filterable_in_grid' => false,
+                'sort_order' => 10,
+                'group' => 'General',
+            ]
+        );
+    }
+}

tests/com/magento/idea/magento2plugin/actions/generation/generator/EavAttributeSetupPatchGeneratorTest.java

@@ -0,0 +1,49 @@
+/*
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+package com.magento.idea.magento2plugin.actions.generation.generator;
+
+import com.intellij.openapi.project.Project;
+import com.intellij.psi.PsiFile;
+import com.magento.idea.magento2plugin.actions.generation.data.ProductEntityData;
+import com.magento.idea.magento2plugin.magento.packages.eav.AttributeScopes;
+
+public class EavAttributeSetupPatchGeneratorTest extends BaseGeneratorTestCase {
+
+    /**
+     * Test Data patch for product's eav attribute generator.
+     */
+    public void testGenerateFile() {
+        final Project project = myFixture.getProject();
+
+        final ProductEntityData productEntityData = new ProductEntityData();
+        productEntityData.setCode("test");
+        productEntityData.setVisibleInGrid(false);
+        productEntityData.setHtmlAllowedOnFront(false);
+        productEntityData.setVisibleOnFront(false);
+        productEntityData.setVisible(true);
+        productEntityData.setScope(AttributeScopes.GLOBAL.getScope());
+        productEntityData.setLabel("Test Label");
+        productEntityData.setType("static");
+        productEntityData.setUsedInGrid(false);
+        productEntityData.setRequired(false);
+        productEntityData.setInput("boolean");
+        productEntityData.setFilterableInGrid(false);
+        productEntityData.setSortOrder(10);
+        productEntityData.setGroup("General");
+
+        productEntityData.setDataPatchName("AddTestAttribute");
+        productEntityData.setNamespace("Foo\\Bar\\Setup\\Patch\\Data");
+        productEntityData.setModuleName("Foo_Bar");
+
+        final EavAttributeSetupPatchGenerator setupPatchGenerator =
+                new EavAttributeSetupPatchGenerator(productEntityData, project);
+        final PsiFile dataPatchFile = setupPatchGenerator.generate("test");
+
+        final String filePatch = this.getFixturePath("AddTestAttribute.php");
+        final PsiFile expectedFile = myFixture.configureByFile(filePatch);
+
+        assertGeneratedFileIsCorrect(expectedFile, "src/app/code/Foo/Bar/Setup/Patch/Data", dataPatchFile);
+    }
+}