From 116743c7c7fd58c737e032f06f85ee73355514a9 Mon Sep 17 00:00:00 2001
From: "Vladislav.Soroka" <Vladislav.Soroka@jetbrains.com>
Date: Fri, 24 Apr 2015 13:03:14 +0300
Subject: [PATCH] gradle: project wizard test fix (cherry picked from commit
 c047fa7)

---
 .../gradle/GradleProjectWizardTest.java       | 46 ++++++++++++++++++-
 1 file changed, 45 insertions(+), 1 deletion(-)

diff --git a/plugins/gradle/testSources/org/jetbrains/plugins/gradle/GradleProjectWizardTest.java b/plugins/gradle/testSources/org/jetbrains/plugins/gradle/GradleProjectWizardTest.java
index c64b66713484..fb2564203e4c 100644
--- a/plugins/gradle/testSources/org/jetbrains/plugins/gradle/GradleProjectWizardTest.java
+++ b/plugins/gradle/testSources/org/jetbrains/plugins/gradle/GradleProjectWizardTest.java
@@ -18,9 +18,18 @@ package org.jetbrains.plugins.gradle;
 import com.intellij.ide.projectWizard.NewProjectWizardTestCase;
 import com.intellij.ide.projectWizard.ProjectTypeStep;
 import com.intellij.ide.util.projectWizard.ModuleWizardStep;
+import com.intellij.ide.util.projectWizard.ProjectBuilder;
 import com.intellij.ide.wizard.Step;
+import com.intellij.openapi.module.Module;
+import com.intellij.openapi.module.ModuleManager;
 import com.intellij.openapi.project.Project;
+import com.intellij.openapi.roots.ModuleRootManager;
+import com.intellij.openapi.roots.ProjectRootManager;
+import com.intellij.openapi.util.text.StringUtil;
+import com.intellij.openapi.vfs.VfsUtilCore;
+import com.intellij.openapi.vfs.VirtualFile;
 import com.intellij.util.Consumer;
+import org.jetbrains.plugins.gradle.service.project.wizard.GradleModuleBuilder;
 
 import java.util.List;
 
@@ -30,16 +39,51 @@ import java.util.List;
 public class GradleProjectWizardTest extends NewProjectWizardTestCase {
 
   public void testGradleProject() throws Exception {
+    final String projectName = "testProject";
     Project project = createProject(new Consumer<Step>() {
       @Override
       public void consume(Step step) {
         if (step instanceof ProjectTypeStep) {
           assertTrue(((ProjectTypeStep)step).setSelectedTemplate("Gradle", null));
           List<ModuleWizardStep> steps = myWizard.getSequence().getSelectedSteps();
-          assertEquals(3, steps.size());
+          assertEquals(5, steps.size());
+          final ProjectBuilder projectBuilder = myWizard.getProjectBuilder();
+          assertInstanceOf(projectBuilder, GradleModuleBuilder.class);
+          ((GradleModuleBuilder)projectBuilder).setName(projectName);
         }
       }
     });
+
+    assertEquals(projectName, project.getName());
+    Module[] modules = ModuleManager.getInstance(project).getModules();
+    assertEquals(1, modules.length);
+    final Module module = modules[0];
+    assertTrue(ModuleRootManager.getInstance(module).isSdkInherited());
+    assertEquals(projectName, module.getName());
+
+    VirtualFile root = ProjectRootManager.getInstance(project).getContentRoots()[0];
+    VirtualFile settingsScript = VfsUtilCore.findRelativeFile("settings.gradle", root);
+    assertNotNull(settingsScript);
+    assertEquals(String.format("rootProject.name = '%s'\n\n", projectName),
+                 StringUtil.convertLineSeparators(VfsUtilCore.loadText(settingsScript)));
+
+    VirtualFile buildScript = VfsUtilCore.findRelativeFile("build.gradle", root);
+    assertNotNull(buildScript);
+    assertEquals("apply plugin: 'java'\n" +
+                 "\n" +
+                 "sourceCompatibility = 1.5\n" +
+                 "\n" +
+                 "group '" + projectName + "'\n" +
+                 "version '1.0-SNAPSHOT'\n" +
+                 "\n" +
+                 "repositories {\n" +
+                 "    mavenCentral()\n" +
+                 "}\n" +
+                 "\n" +
+                 "dependencies {\n" +
+                 "    testCompile group: 'junit', name: 'junit', version: '4.11'\n" +
+                 "}",
+                 StringUtil.convertLineSeparators(VfsUtilCore.loadText(buildScript)));
   }
 
   @Override