Author: thimel Date: 2009-04-30 17:23:30 +0000 (Thu, 30 Apr 2009) New Revision: 489 Modified: eugene/trunk/changelog.txt eugene/trunk/src/main/java/org/nuiton/eugene/ImportsManager.java eugene/trunk/src/test/java/org/nuiton/eugene/ImportsManagerTest.java Log: ImportsManager can handle Class Modified: eugene/trunk/changelog.txt =================================================================== --- eugene/trunk/changelog.txt 2009-04-30 17:01:48 UTC (rev 488) +++ eugene/trunk/changelog.txt 2009-04-30 17:23:30 UTC (rev 489) @@ -1,4 +1,5 @@ 1.0.0 xxx xxx + * Add ImportsManager * Add useful methods in 'GeneratorUtil' * Improve ant task * Move to org.nuiton groupid Modified: eugene/trunk/src/main/java/org/nuiton/eugene/ImportsManager.java =================================================================== --- eugene/trunk/src/main/java/org/nuiton/eugene/ImportsManager.java 2009-04-30 17:01:48 UTC (rev 488) +++ eugene/trunk/src/main/java/org/nuiton/eugene/ImportsManager.java 2009-04-30 17:23:30 UTC (rev 489) @@ -8,6 +8,8 @@ import java.util.Map; import java.util.Set; +import org.jaxen.expr.AdditiveExpr; + /** * Class used in generators that allows to manage easily imports. A first-pass * allow to register imports, and in a second-pass, returns the type to use in @@ -48,6 +50,16 @@ private Map<String, String> imports = new HashMap<String, String>(); /** + * From the given class, add it to the imports list. + * @param clazz the class to import + * @return true if there is a conflict adding the import + * @see ImportsManager#addImport(String) + */ + public boolean addImport(Class<?> clazz) { + return addImport(clazz.getName()); + } + + /** * From the given fqn, add it to the imports list. If there is a confilct * adding this import, will return true * @param fqn the fully qualified name to import Modified: eugene/trunk/src/test/java/org/nuiton/eugene/ImportsManagerTest.java =================================================================== --- eugene/trunk/src/test/java/org/nuiton/eugene/ImportsManagerTest.java 2009-04-30 17:01:48 UTC (rev 488) +++ eugene/trunk/src/test/java/org/nuiton/eugene/ImportsManagerTest.java 2009-04-30 17:23:30 UTC (rev 489) @@ -1,5 +1,6 @@ package org.nuiton.eugene; +import java.io.Serializable; import java.util.List; import org.nuiton.eugene.ImportsManager; @@ -26,6 +27,9 @@ assertFalse(mgr.addImport("Byte")); assertFalse(mgr.addImport("java.lang.Toto")); assertTrue(mgr.addImport("java.lang.sub.Toto")); + + assertFalse(mgr.addImport(Serializable.class)); + assertTrue(mgr.addImport("org.nuiton.eugene.Serializable")); } public void testGetType() throws Exception {