private void setParentProject(FileModel fileModel, MavenProjectModel projectModel) {
if (fileModel == null) {
return;
} else if (fileModel.getProjectModel() != null) {
projectModel.setParentProject(fileModel.getProjectModel());
} else {
setParentProject(fileModel.getParentArchive(), projectModel);
}
}
/**
* This method just attaches the {@link OrganizationModel} to the {@link FileModel}. It will only
* do so if this link is not already present.
*/
public OrganizationModel attachOrganization(
OrganizationModel organizationModel, ArchiveModel archiveModel) {
// check for duplicates
for (FileModel existingFileModel : organizationModel.getArchiveModels()) {
if (existingFileModel.equals(archiveModel)) {
return organizationModel;
}
}
organizationModel.addArchiveModel(archiveModel);
return organizationModel;
}
private void setMavenParentProject(MavenProjectModel projectModel) {
FileModel fileModel = projectModel.getRootFileModel();
if (fileModel == null) {
// skip if no file was discovered for it
return;
} else if (fileModel instanceof ArchiveModel) {
ArchiveModel archiveModel = (ArchiveModel) fileModel;
// look at the parent archive first
setParentProject(archiveModel.getParentArchive(), projectModel);
} else {
FileModel parentFile = fileModel.getParentFile();
setParentProject(parentFile, projectModel);
}
}
@Test
public void testNestedCondition() throws IOException {
try (GraphContext context = factory.create()) {
ProjectModel pm = context.getFramed().addVertex(null, ProjectModel.class);
pm.setName("Main Project");
FileModel inputPath = context.getFramed().addVertex(null, FileModel.class);
inputPath.setFilePath("src/test/resources/");
Path outputPath =
Paths.get(
FileUtils.getTempDirectory().toString(), "windup_" + UUID.randomUUID().toString());
FileUtils.deleteDirectory(outputPath.toFile());
Files.createDirectories(outputPath);
inputPath.setProjectModel(pm);
pm.setRootFileModel(inputPath);
WindupConfiguration windupConfiguration =
new WindupConfiguration()
.setRuleProviderFilter(
new NotPredicate(
new RuleProviderPhasePredicate(
MigrationRulesPhase.class, ReportGenerationPhase.class)))
.setGraphContext(context);
windupConfiguration.setInputPath(Paths.get(inputPath.getFilePath()));
windupConfiguration.setOutputDirectory(outputPath);
processor.execute(windupConfiguration);
GraphService<ClassificationModel> classificationService =
new GraphService<>(context, ClassificationModel.class);
Assert.assertEquals(1, provider.getXmlFileMatches().size());
List<ClassificationModel> classifications = Iterators.asList(classificationService.findAll());
for (ClassificationModel model : classifications) {
String classification = model.getClassification();
String classificationString = classification.toString();
Assert.assertEquals("Spring File", classificationString);
}
Assert.assertEquals(1, classifications.size());
Iterator<FileModel> iterator = classifications.get(0).getFileModels().iterator();
Assert.assertNotNull(iterator.next());
Assert.assertNotNull(iterator.next());
Assert.assertFalse(iterator.hasNext());
}
}
public void perform(GraphRewrite event, EvaluationContext context, FileModel payload) {
GraphService<PropertiesModel> service = getPropertiesService(event.getGraphContext());
PropertiesModel properties = service.create();
properties.setFileResource(payload);
try (InputStream is = payload.asInputStream()) {
Properties props = new Properties();
props.load(is);
for (Object key : props.keySet()) {
String property = StringUtils.trim(key.toString());
String propertyValue = StringUtils.trim(props.get(key).toString());
properties.setProperty(property, propertyValue);
}
} catch (IOException e) {
throw new WindupException(
"Failed to load properties file: "
+ payload.getFilePath()
+ " due to: "
+ e.getMessage());
}
}
/** Recurses the given folder and adds references to these files to the graph as FileModels */
private void recurseAndAddFiles(FileService fileService, FileModel file) {
String filePath = file.getFilePath();
File fileReference = new File(filePath);
if (fileReference.isDirectory()) {
File[] subFiles = fileReference.listFiles();
if (subFiles != null) {
for (File reference : subFiles) {
FileModel subFile = fileService.createByFilePath(file, reference.getAbsolutePath());
recurseAndAddFiles(fileService, subFile);
}
}
}
}
@Test
public void testQuerying() throws Exception {
FileModel f1 = context.getFramed().addVertex(null, FileModel.class);
f1.setFilePath("/f1");
FileModel f2 = context.getFramed().addVertex(null, FileModel.class);
f2.setFilePath("/f2");
FileModel f3 = context.getFramed().addVertex(null, FileModel.class);
f3.setFilePath("/f3");
FileModel f4 = context.getFramed().addVertex(null, FileModel.class);
f4.setFilePath("/f4");
FileModel f5 = context.getFramed().addVertex(null, FileModel.class);
f5.setFilePath("/f5");
FileModel f6 = context.getFramed().addVertex(null, FileModel.class);
f6.setFilePath("/f6");
FileModel f7 = context.getFramed().addVertex(null, FileModel.class);
f7.setFilePath("/f7");
BlackListModel b1 = context.getFramed().addVertex(null, BlackListModel.class);
BlackListModel b1b = context.getFramed().addVertex(null, BlackListModel.class);
b1.setFileModel(f1);
b1b.setFileModel(f1);
BlackListModel b2 = context.getFramed().addVertex(null, BlackListModel.class);
b2.setFileModel(f2);
ClassificationModel c1 = context.getFramed().addVertex(null, ClassificationModel.class);
ClassificationModel c1b = context.getFramed().addVertex(null, ClassificationModel.class);
c1.setFileModel(f1);
c1b.setFileModel(f1);
ClassificationModel c2 = context.getFramed().addVertex(null, ClassificationModel.class);
c2.setFileModel(f3);
List<Vertex> vertexList = new ArrayList<>();
for (Vertex v :
context
.getFramed()
.query()
.has(WindupVertexFrame.TYPE_FIELD, Text.CONTAINS, "FileResource")
.vertices()) {
vertexList.add(v);
}
GremlinPipeline<Vertex, Vertex> pipeline =
new GremlinPipeline<>(
context
.getFramed()
.query()
.has(WindupVertexFrame.TYPE_FIELD, Text.CONTAINS, "FileResource")
.vertices());
GraphRewrite event = new GraphRewrite(context);
// manually execute this criterion (this just adds things to the pipeline)
new FindClassifiedFilesGremlinCriterion().query(event, pipeline);
List<FileModel> fileModels = new ArrayList<>();
for (Vertex v : pipeline) {
// Explicit cast here insures that the frame returned was actually a FileModel. If it is not,
// a
// ClassCastException will
// occur and the test will fail.
//
// If we called frame(v, FileModel.class) directly, frames would happily force it to be a
// FileModel
// even if the underlying query were returning invalid results.
FileModel fm = (FileModel) context.getFramed().frame(v, WindupVertexFrame.class);
fileModels.add(fm);
}
boolean foundF1 = false;
boolean foundF2 = false;
boolean foundF3 = false;
Assert.assertEquals(3, fileModels.size());
for (FileModel fm : fileModels) {
if (fm.getFilePath().equals(f1.getFilePath())) {
foundF1 = true;
} else if (fm.getFilePath().equals(f2.getFilePath())) {
foundF2 = true;
} else if (fm.getFilePath().equals(f3.getFilePath())) {
foundF3 = true;
}
}
Assert.assertTrue(foundF1);
Assert.assertTrue(foundF2);
Assert.assertTrue(foundF3);
}
@Test
public void testFindingNonClassifiedFiles() throws Exception {
FileModel f1 = context.getFramed().addVertex(null, FileModel.class);
f1.setFilePath("/f1");
FileModel f2 = context.getFramed().addVertex(null, FileModel.class);
f2.setFilePath("/f2");
FileModel f3 = context.getFramed().addVertex(null, FileModel.class);
f3.setFilePath("/f3");
FileModel f4 = context.getFramed().addVertex(null, FileModel.class);
f4.setFilePath("/f4");
FileModel f5 = context.getFramed().addVertex(null, FileModel.class);
f5.setFilePath("/f5");
FileModel f6 = context.getFramed().addVertex(null, FileModel.class);
f6.setFilePath("/f6");
FileModel f7 = context.getFramed().addVertex(null, FileModel.class);
f7.setFilePath("/f7");
InlineHintModel b1 = context.getFramed().addVertex(null, InlineHintModel.class);
InlineHintModel b1b = context.getFramed().addVertex(null, InlineHintModel.class);
b1.setFile(f1);
b1b.setFile(f1);
InlineHintModel b2 = context.getFramed().addVertex(null, InlineHintModel.class);
b2.setFile(f2);
ClassificationModel c1 = context.getFramed().addVertex(null, ClassificationModel.class);
ClassificationModel c1b = context.getFramed().addVertex(null, ClassificationModel.class);
c1.addFileModel(f1);
c1b.addFileModel(f1);
ClassificationModel c2 = context.getFramed().addVertex(null, ClassificationModel.class);
c2.addFileModel(f3);
List<Vertex> vertexList = new ArrayList<>();
for (Vertex v : context.getQuery().type(FileModel.class).vertices()) {
vertexList.add(v);
}
// manually execute this criterion (this just adds things to the pipeline)
Iterable<Vertex> allFMVertices = context.getQuery().type(FileModel.class).vertices();
Iterable<Vertex> fileModelIterable =
new FindFilesNotClassifiedOrHintedGremlinCriterion().query(context, allFMVertices);
List<FileModel> fileModels = new ArrayList<>();
for (Vertex v : fileModelIterable) {
// Explicit cast here insures that the frame returned was actually a FileModel. If it is not,
// a
// ClassCastException will
// occur and the test will fail.
//
// If we called frame(v, FileModel.class) directly, frames would happily force it to be a
// FileModel
// even if the underlying query were returning invalid results.
FileModel fm = (FileModel) context.getFramed().frame(v, WindupVertexFrame.class);
fileModels.add(fm);
}
boolean foundF4 = false;
boolean foundF5 = false;
boolean foundF6 = false;
boolean foundF7 = false;
Assert.assertEquals(4, fileModels.size());
for (FileModel fm : fileModels) {
if (fm.getFilePath().equals(f4.getFilePath())) {
foundF4 = true;
} else if (fm.getFilePath().equals(f5.getFilePath())) {
foundF5 = true;
} else if (fm.getFilePath().equals(f6.getFilePath())) {
foundF6 = true;
} else if (fm.getFilePath().equals(f7.getFilePath())) {
foundF7 = true;
}
}
Assert.assertTrue(foundF4);
Assert.assertTrue(foundF5);
Assert.assertTrue(foundF6);
Assert.assertTrue(foundF7);
}
@Test
public void testJavaClassCondition()
throws IOException, InstantiationException, IllegalAccessException {
try (GraphContext context = factory.create(getDefaultPath())) {
final String inputDir = "src/test/resources/org/jboss/windup/rules/java";
final Path outputPath =
Paths.get(
FileUtils.getTempDirectory().toString(),
"windup_" + RandomStringUtils.randomAlphanumeric(6));
FileUtils.deleteDirectory(outputPath.toFile());
Files.createDirectories(outputPath);
// Fill the graph with test data.
ProjectModel pm = context.getFramed().addVertex(null, ProjectModel.class);
pm.setName("Main Project");
// Create FileModel for $inputDir
FileModel inputPathFrame = context.getFramed().addVertex(null, FileModel.class);
inputPathFrame.setFilePath(inputDir);
inputPathFrame.setProjectModel(pm);
pm.addFileModel(inputPathFrame);
// Set project.rootFileModel to inputPath
pm.setRootFileModel(inputPathFrame);
// Create FileModel for $inputDir/HintsClassificationsTest.java
FileModel fileModel = context.getFramed().addVertex(null, FileModel.class);
fileModel.setFilePath(inputDir + "/JavaHintsClassificationsTest.java");
fileModel.setProjectModel(pm);
pm.addFileModel(fileModel);
// Create FileModel for $inputDir/JavaClassTest.java
fileModel = context.getFramed().addVertex(null, FileModel.class);
fileModel.setFilePath(inputDir + "/JavaClassTest.java");
fileModel.setProjectModel(pm);
pm.addFileModel(fileModel);
context.getGraph().getBaseGraph().commit();
final WindupConfiguration processorConfig =
new WindupConfiguration().setOutputDirectory(outputPath);
processorConfig.setRuleProviderFilter(
new RuleProviderWithDependenciesPredicate(TestJavaClassTestRuleProvider.class));
processorConfig
.setGraphContext(context)
.setRuleProviderFilter(
new RuleProviderWithDependenciesPredicate(TestJavaClassTestRuleProvider.class));
processorConfig.setInputPath(Paths.get(inputDir));
processorConfig.setOutputDirectory(outputPath);
processorConfig.setOptionValue(ScanPackagesOption.NAME, Collections.singletonList(""));
processor.execute(processorConfig);
GraphService<JavaTypeReferenceModel> typeRefService =
new GraphService<>(context, JavaTypeReferenceModel.class);
Iterable<JavaTypeReferenceModel> typeReferences = typeRefService.findAll();
Assert.assertTrue(typeReferences.iterator().hasNext());
Assert.assertEquals(3, provider.getFirstRuleMatchCount());
Assert.assertEquals(1, provider.getSecondRuleMatchCount());
}
}
