Merge pull request #230 from serhiyzhovnir/issue-198-cover-module-ViewModel-class-generator

#issue-198 Covered module View Model class generator with test

Author: Vitaliy

testData/actions/generation/generator/ModuleViewModelClassGenerator/generateViewModelFile/TestViewModel.php

@@ -0,0 +1,10 @@
+<?php
+
+namespace Foo\Bar\ViewModel;
+
+use Magento\Framework\View\Element\Block\ArgumentInterface;
+
+class TestViewModel implements ArgumentInterface
+{
+
+}

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

@@ -0,0 +1,37 @@
+/*
+ * 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.ViewModelFileData;
+
+public class ModuleViewModelClassGeneratorTest extends BaseGeneratorTestCase {
+
+    public void testGenerateViewModelFile() {
+        Project project = myFixture.getProject();
+
+        ViewModelFileData viewModelData = new ViewModelFileData(
+                "ViewModel",
+                "TestViewModel",
+                "Foo_Bar",
+                "Foo\\Bar\\ViewModel"
+        );
+        ModuleViewModelClassGenerator moduleViewModelClassGenerator = new ModuleViewModelClassGenerator(
+                viewModelData,
+                project
+        );
+        PsiFile viewModelFile = moduleViewModelClassGenerator.generate("test");
+
+        String filePath = this.getFixturePath("TestViewModel.php");
+        PsiFile expectedFile = myFixture.configureByFile(filePath);
+
+        assertGeneratedFileIsCorrect(
+                expectedFile,
+                "src/app/code/Foo/Bar/ViewModel",
+                viewModelFile
+        );
+    }
+}