@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()); } }
/** Add report data as associations to this report instance */ @SuppressWarnings("unchecked") public static void addAssociatedReportData( GraphContext context, ReportModel reportModel, Map<String, Object> reportData) { Map<String, WindupVertexFrame> relatedResources = new HashMap<>(); for (Map.Entry<String, Object> varEntry : reportData.entrySet()) { Object value = varEntry.getValue(); if (value instanceof WindupVertexFrame) { relatedResources.put(varEntry.getKey(), (WindupVertexFrame) varEntry.getValue()); } else if (value instanceof Iterable) { WindupVertexListModel list = context.getFramed().addVertex(null, WindupVertexListModel.class); for (WindupVertexFrame frame : (Iterable<? extends WindupVertexFrame>) value) { list.addItem(frame); } relatedResources.put(varEntry.getKey(), list); } else { throw new WindupException( "Unrecognized variable type: " + value.getClass().getCanonicalName() + " encountered!"); } } }
@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); }