Added test coverage

Author: drpayyne

testData/actions/generation/generator/DataModelGenerator/generateDataModel/Sample.php

@@ -0,0 +1,25 @@
+<?php
+
+namespace Foo\Bar\Model\Data;
+
+use Foo\Bar\Api\Data\SampleInterface;
+use Magento\Framework\DataObject;
+
+class Sample extends DataObject implements SampleInterface
+{
+    /**
+     * @inheritDoc
+     */
+    public function getSampleProperty()
+    {
+        return $this->getData(self::SAMPLE_PROPERTY);
+    }
+
+    /**
+     * @inheritDoc
+     */
+    public function setSampleProperty($sampleProperty)
+    {
+        return $this->setData(self::SAMPLE_PROPERTY, $sampleProperty);
+    }
+}

testData/actions/generation/generator/DataModelInterfaceGenerator/generateDataModelInterface/SampleInterface.php

@@ -0,0 +1,22 @@
+<?php
+
+namespace Foo\Bar\Api\Data;
+
+interface SampleInterface
+{
+    /**
+     * String constants for property names
+     */
+    const SAMPLE_PROPERTY = "sample_property";
+
+    /**
+     * @return string
+     */
+    public function getSampleProperty();
+
+    /**
+     * @param string $sampleProperty
+     * @return $this
+     */
+    public function setSampleProperty($sampleProperty);
+}

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

@@ -0,0 +1,40 @@
+/*
+ * 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.NewDataModelAction;
+import com.magento.idea.magento2plugin.actions.generation.data.DataModelData;
+
+public class DataModelGeneratorTest extends BaseGeneratorTestCase {
+    /**
+     * Tests for generation of a Magento 2 Data Model.
+     */
+    public void testGenerateDataModel() {
+        final Project project = myFixture.getProject();
+        final DataModelData modelData = new DataModelData(
+                "Foo\\Bar\\Model\\Data",
+                "Sample",
+                "Foo_Bar",
+                "Foo\\Bar\\Model\\Data\\Sample",
+                "Foo\\Bar\\Api\\Data\\SampleInterface",
+                "SAMPLE_PROPERTY;sample_property;string;SampleProperty;sampleProperty"
+        );
+        final DataModelGenerator generator = new DataModelGenerator(
+                project, modelData
+        );
+        final PsiFile modelFile = generator.generate(NewDataModelAction.ACTION_NAME);
+        final PsiFile expectedFile
+                = myFixture.configureByFile(this.getFixturePath("Sample.php"));
+
+        assertGeneratedFileIsCorrect(
+                expectedFile,
+                "src/app/code/Foo/Bar/Model/Data",
+                modelFile
+        );
+    }
+}

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

@@ -0,0 +1,39 @@
+/*
+ * 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.NewDataModelAction;
+import com.magento.idea.magento2plugin.actions.generation.data.DataModelInterfaceData;
+
+public class DataModelInterfaceGeneratorTest extends BaseGeneratorTestCase {
+    /**
+     * Tests for generation of a Magento 2 Data Model Interface.
+     */
+    public void testGenerateDataModelInterface() {
+        final Project project = myFixture.getProject();
+        final DataModelInterfaceData interfaceData = new DataModelInterfaceData(
+                "Foo\\Bar\\Api\\Data",
+                "SampleInterface",
+                "Foo_Bar",
+                "Foo\\Bar\\Api\\Data\\SampleInterface",
+                "SAMPLE_PROPERTY;sample_property;string;SampleProperty;sampleProperty"
+        );
+        final DataModelInterfaceGenerator generator = new DataModelInterfaceGenerator(
+                project, interfaceData
+        );
+        final PsiFile interfaceFile = generator.generate(NewDataModelAction.ACTION_NAME);
+        final PsiFile expectedFile
+                = myFixture.configureByFile(this.getFixturePath("SampleInterface.php"));
+
+        assertGeneratedFileIsCorrect(
+                expectedFile,
+                "src/app/code/Foo/Bar/Api/Data",
+                interfaceFile
+        );
+    }
+}