ArchitectureNodeTagger { public static void tagArchitectureNodes(DependencyGraph graph, Architecture architecture, String tag) { graph.getAllNodes() .forEach(node -> tagNode(graph.getBaseGraph(), node, architecture, tag)); } private  ArchitectureNodeTagger(); static void tagArchitectureNodes(DependencyGraph graph, Architecture architecture, String tag);  }
@Test public void testTagArchitectureNodes() { Architecture packageArchitecture = new DefaultPackageArchitectureFactory(dependencyGraph).createArchitecture(); DependencyGraph inputGraph = dependencyGraph.filter(new NodePropertyInFilter("scope", "input")); new ArchitectureNodeCreator(dependencyGraph, packageArchitecture).createAllArchitectureNodes(); ArchitectureNodeTagger.tagArchitectureNodes(inputGraph, packageArchitecture, "my-tag"); Node x = assertNode("X"); assertThat(x.getProperty("my-tag"), notNullValue()); assertThat(x.getProperty("my-tag"), is(true)); Node y = assertNode("Y"); assertThat(y.getProperty("my-tag"), nullValue()); }