/*
* Copyright 2000-2014 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.jetbrains.plugins.gradle.tooling.builder;
import com.intellij.openapi.util.io.FileUtil;
import com.intellij.util.Function;
import com.intellij.util.containers.ContainerUtil;
import org.gradle.tooling.model.DomainObjectSet;
import org.gradle.tooling.model.idea.IdeaModule;
import org.gradle.tooling.model.idea.IdeaSourceDirectory;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.plugins.gradle.model.ExtIdeaContentRoot;
import org.jetbrains.plugins.gradle.model.ModuleExtendedModel;
import org.junit.Assert;
import org.junit.Test;
import java.io.File;
import java.util.*;
import static org.junit.Assert.*;
/**
* @author Vladislav.Soroka
* @since 1/16/14
*/
public class ModuleExtendedModelBuilderImplTest extends AbstractModelBuilderTest {
public ModuleExtendedModelBuilderImplTest(@NotNull String gradleVersion) {
super(gradleVersion);
}
@Test
public void testGradleSourcesSetsInterpretation() throws Exception {
final int modulesSize = 9;
DomainObjectSet<? extends IdeaModule> ideaModules = allModels.getIdeaProject().getModules();
List<ModuleExtendedModel> models =
ContainerUtil.mapNotNull(ideaModules, (Function<IdeaModule, ModuleExtendedModel>)module -> {
ModuleExtendedModel moduleExtendedModel = allModels.getExtraProject(module, ModuleExtendedModel.class);
assertNotNull(moduleExtendedModel);
List<String> sourceDirectories = ContainerUtil.newArrayList();
List<String> resourceDirectories = ContainerUtil.newArrayList();
List<String> testResourceDirectories = ContainerUtil.newArrayList();
List<String> testDirectories = ContainerUtil.newArrayList();
List<String> excludeDirectories = ContainerUtil.newArrayList();
fillDirectories(moduleExtendedModel,
sourceDirectories, resourceDirectories,
testDirectories, testResourceDirectories,
excludeDirectories);
if (module.getName().equals("defaultJavaModule") || module.getName().equals("moduleWithSourceSetDirBothAsResourceAndJava")) {
assertEquals(ContainerUtil.newArrayList("src/main/java"), sourceDirectories);
assertEquals(ContainerUtil.newArrayList("src/main/resources"), resourceDirectories);
assertEquals(ContainerUtil.newArrayList("src/test/java"), testDirectories);
assertEquals(ContainerUtil.newArrayList("src/test/resources"), testResourceDirectories);
assertEquals(ContainerUtil.newArrayList(".gradle", "build"), excludeDirectories);
}
else if (module.getName().equals("moduleWithSourceSetDirBothAsResourceAndGroovy")) {
assertEquals(ContainerUtil.newArrayList("src/main/groovy", "src/main/java"), sourceDirectories);
assertEquals(ContainerUtil.newArrayList("src/main/resources"), resourceDirectories);
assertEquals(ContainerUtil.newArrayList("src/test/groovy", "src/test/java"), testDirectories);
assertEquals(ContainerUtil.newArrayList("src/test/resources"), testResourceDirectories);
assertEquals(ContainerUtil.newArrayList(".gradle", "build"), excludeDirectories);
}
else if (module.getName().equals("moduleWithCustomSourceSet")) {
assertEquals(ContainerUtil.newArrayList("src/custom/java", "src/main/java"), sourceDirectories);
assertEquals(ContainerUtil.newArrayList("src/custom/resources", "src/main/resources"), resourceDirectories);
assertEquals(ContainerUtil.newArrayList("src/test/java"), testDirectories);
assertEquals(ContainerUtil.newArrayList("src/test/resources"), testResourceDirectories);
assertEquals(ContainerUtil.newArrayList(".gradle", "build"), excludeDirectories);
}
else if (module.getName().equals("withIntegrationTests")) {
assertEquals(ContainerUtil.newArrayList("src/main/java"), sourceDirectories);
assertEquals(ContainerUtil.newArrayList("src/main/resources"), resourceDirectories);
assertEquals(ContainerUtil.newArrayList(
"src/integration-test/java", "src/integrationTest/java", "src/test/java"), testDirectories);
assertEquals(ContainerUtil.newArrayList(
"src/integration-test/resources",
"src/integrationTest/resources",
"src/test/resources"), testResourceDirectories);
assertEquals(ContainerUtil.newArrayList(".gradle", "build"), excludeDirectories);
}
else if (module.getName().equals("testGradleSourcesSetsInterpretation")) {
assertTrue(sourceDirectories.isEmpty());
assertTrue(resourceDirectories.isEmpty());
assertTrue(testDirectories.isEmpty());
assertTrue(testResourceDirectories.isEmpty());
assertEquals(ContainerUtil.newArrayList(".gradle", "build"), excludeDirectories);
}
else if (module.getName().equals("withIdeaPluginCustomization1")) {
assertEquals(ContainerUtil.newArrayList("src/main/java"), sourceDirectories);
assertEquals(ContainerUtil.newArrayList("src/main/resources"), resourceDirectories);
assertEquals(ContainerUtil.newArrayList("src/intTest/java", "src/intTest/resources", "src/test/java"), testDirectories);
assertEquals(ContainerUtil.newArrayList("src/test/resources"), testResourceDirectories);
assertEquals(ContainerUtil.newArrayList(".gradle", "build", "some-extra-exclude-folder"), excludeDirectories);
}
else if (module.getName().equals("withIdeaPluginCustomization2")) {
assertEquals(ContainerUtil.newArrayList("src/main/java", "src/test/java", "src/test/resources"), sourceDirectories);
assertEquals(ContainerUtil.newArrayList("src/main/resources"), resourceDirectories);
assertTrue(testDirectories.isEmpty());
assertTrue(testResourceDirectories.isEmpty());
assertEquals(ContainerUtil.newArrayList(".gradle", "build"), excludeDirectories);
}
else if (module.getName().equals("withIdeaPluginCustomization3")) {
assertEquals(ContainerUtil.newArrayList("src/main/java"), sourceDirectories);
assertEquals(ContainerUtil.newArrayList("src/awesome-test/resources", "src/main/resources"), resourceDirectories);
assertEquals(ContainerUtil.newArrayList("src/awesome-test/java", "src/test/java"), testDirectories);
assertEquals(ContainerUtil.newArrayList("src/test/resources"), testResourceDirectories);
assertEquals(ContainerUtil.newArrayList(".gradle", "build"), excludeDirectories);
}
else {
fail();
}
return moduleExtendedModel;
});
assertEquals(modulesSize, models.size());
}
@Test
public void testJavaExtendedModel() throws Exception {
final Map<String, ModuleExtendedModel> modulesMap = getModulesMap(ModuleExtendedModel.class);
assertEquals(3, modulesMap.size());
final ModuleExtendedModel rootModule = modulesMap.get(":");
assertNotNull(rootModule);
assertNull(rootModule.getJavaSourceCompatibility());
final ModuleExtendedModel defaultJavaModule = modulesMap.get(":defaultJavaModule");
assertNotNull(defaultJavaModule);
assertNotNull(defaultJavaModule.getJavaSourceCompatibility());
final ModuleExtendedModel javaModule = modulesMap.get(":javaModule");
assertNotNull(javaModule);
assertEquals("1.6", javaModule.getJavaSourceCompatibility());
}
private void fillDirectories(final ModuleExtendedModel model,
List<String> sourceDirectories, List<String> resourceDirectories,
List<String> testDirectories, List<String> resourceTestDirectories,
List<String> excludeDirectories) {
for (ExtIdeaContentRoot contentRoot : model.getContentRoots()) {
sourceDirectories.addAll(getAllPaths(contentRoot.getSourceDirectories(), model.getName()));
resourceDirectories.addAll(getAllPaths(contentRoot.getResourceDirectories(), model.getName()));
testDirectories.addAll(getAllPaths(contentRoot.getTestDirectories(), model.getName()));
resourceTestDirectories.addAll(getAllPaths(contentRoot.getTestResourceDirectories(), model.getName()));
excludeDirectories.addAll(getAllPaths(contentRoot.getExcludeDirectories(), model.getName()));
}
}
private Collection<? extends String> getAllPaths(Collection<? extends File> directories, final String moduleName) {
List<String> list = ContainerUtil.map2List(directories, (Function<File, String>)sourceDirectory -> {
String path =
FileUtil.toCanonicalPath(FileUtil.getRelativePath(new File(testDir, moduleName), sourceDirectory));
Assert.assertNotNull(path);
return path.substring(path.indexOf("/") + 1);
});
Collections.sort(list);
return list;
}
private Collection<? extends String> getAllPaths(DomainObjectSet<? extends IdeaSourceDirectory> directories, final String moduleName) {
List<String> list = ContainerUtil.map2List(directories, (Function<IdeaSourceDirectory, String>)sourceDirectory -> {
String path =
FileUtil.toCanonicalPath(FileUtil.getRelativePath(new File(testDir, moduleName), sourceDirectory.getDirectory()));
Assert.assertNotNull(path);
return path.substring(path.indexOf("/") + 1);
});
Collections.sort(list);
return list;
}
@Override
protected Set<Class> getModels() {
return ContainerUtil.set(ModuleExtendedModel.class);
}
}