More tests and method signatures for ClassPathScanner

Author: graemerocher

grails-core/src/main/groovy/grails/boot/config/tools/ClassPathScanner.groovy

@@ -28,6 +28,41 @@ import java.lang.annotation.Annotation
 class ClassPathScanner {
     private static final List DEFAULT_IGNORED_ROOT_PACKAGES = ['com', 'org', 'net', 'co', 'java', 'javax', 'groovy']
 
+
+    /**
+     * Scans for classes relative to the given class
+     *
+     * @param applicationClass The class, usually the Application class
+     *
+     * @return A set of classes
+     */
+    Set<Class> scan(Class applicationClass) {
+        return scan(applicationClass,[applicationClass.package.name])
+    }
+
+    /**
+     * Scans for classes relative to the given class
+     *
+     * @param applicationClass The class, usually the Application class
+     * @param annotationFilter The annotation to filter by
+     *
+     * @return A set of classes
+     */
+    Set<Class> scan(Class applicationClass, Class<? extends Annotation> annotationFilter) {
+        return scan(applicationClass,[applicationClass.package.name], annotationFilter)
+    }
+
+    /**
+     * Scans for classes relative to the given class
+     *
+     * @param applicationClass The class, usually the Application class
+     * @param annotationFilter The annotation to filter by
+     *
+     * @return A set of classes
+     */
+    Set<Class> scan(Class applicationClass, Closure<Boolean> annotationFilter ) {
+        return scan(applicationClass,[applicationClass.package.name], annotationFilter)
+    }
     /**
      * Scans for classes relative to the given class
      *

grails-core/src/test/groovy/grails/boot/config/tools/ClassPathScannerSpec.groovy

@@ -0,0 +1,36 @@
+package grails.boot.config.tools
+
+import grails.persistence.Entity
+import spock.lang.Specification
+
+/**
+ * Created by graemerocher on 06/10/2016.
+ */
+class ClassPathScannerSpec extends Specification {
+    void "Test classpath scanner with package names"() {
+        when:"the classpath is scanned"
+        ClassPathScanner scanner = new ClassPathScanner()
+        def results = scanner.scan(Application, ["grails.boot.config.tools"])
+
+        then:"the results are correct"
+        results.size() == 1
+        results.contains(Foo)
+    }
+
+    void "Test classpath scanner with application"() {
+        when:"the classpath is scanned"
+        ClassPathScanner scanner = new ClassPathScanner()
+        def results = scanner.scan(Application)
+
+        then:"the results are correct"
+        results.size() == 1
+        results.contains(Foo)
+    }
+}
+
+class Application {
+
+}
+
+@Entity
+class Foo {}