Merge branch 'master' of ssh://svjatoslav.eu/home/git/repositories/javainspect

[javainspect.git] / src / test / java / eu / svjatoslav / inspector / java / structure / example / RenderDemoClasses.java

diff --git a/src/test/java/eu/svjatoslav/inspector/java/structure/example/RenderDemoClasses.java b/src/test/java/eu/svjatoslav/inspector/java/structure/example/RenderDemoClasses.java
index 7e9e5a3..90aa480 100755 (executable)
--- a/src/test/java/eu/svjatoslav/inspector/java/structure/example/RenderDemoClasses.java
+++ b/src/test/java/eu/svjatoslav/inspector/java/structure/example/RenderDemoClasses.java
@@ -1,7 +1,7 @@
 /*
  * JavaInspect - Utility to visualize java software
- * Copyright (C) 2013-2014, Svjatoslav Agejenko, svjatoslav@svjatoslav.eu
- * 
+ * Copyright (C) 2013-2015, Svjatoslav Agejenko, svjatoslav@svjatoslav.eu
+ *
  * This program is free software; you can redistribute it and/or
  * modify it under the terms of version 3 of the GNU Lesser General Public License
  * or later as published by the Free Software Foundation.
@@ -11,15 +11,14 @@ package eu.svjatoslav.inspector.java.structure.example;
 
 import eu.svjatoslav.inspector.java.structure.ClassGraph;
 import eu.svjatoslav.inspector.java.structure.example.structure.SampleClass;
+import eu.svjatoslav.inspector.java.structure.example.structure.SampleClass2;
 
 public class RenderDemoClasses {
 
-       public static void main(final String[] args) {
-               final ClassGraph graph = new ClassGraph();
+    public static void main(final String[] args) {
 
-               graph.addClass(SampleClass.class);
-
-               graph.generateGraph("example");
-       }
+        new ClassGraph().add(SampleClass.class, SampleClass2.class)
+                .generateGraph("example");
+    }
 
 }