lib/appium_lib/driver.rb in appium_lib-0.6.4 vs lib/appium_lib/driver.rb in appium_lib-0.6.5

@@ -89,12 +89,30 @@
     # ensure files are absolute
     r.map! do |file|
       file = file.include?(File::Separator) ? file :
              File.join(parent_dir, file)
       file = File.expand_path file
+
       File.exists?(file) ? file : nil
     end
-    r.compact # remove nils
+    r.compact! # remove nils
+
+    files = []
+
+    # now expand dirs
+    r.each do |item|
+      unless File.directory? item
+        # save file
+        files << item
+        next # only look inside folders
+      end
+      Dir.glob(File.join(item, '**/*.rb')) do |file|
+        # do not add folders to the file list
+        files << File.expand_path(file) unless File.directory? file
+      end
+    end
+
+    files
   end
 end
 
 # Fix uninitialized constant Minitest (NameError)
 module Minitest