lib/railroad/rails_framework.rb in royw-railroad_xing-0.5.0.1 vs lib/railroad/rails_framework.rb in royw-railroad_xing-0.5.0.2

@@ -18,10 +18,11 @@
   end
   
   # get the controller's files returning the application controller first in returned array
   def get_controller_files(options)
     files = []
-    files << 'app/controllers/application.rb'
+    files << 'app/controllers/application_controller.rb' if File.exist?('app/controllers/application_controller.rb')
+    files << 'app/controllers/application.rb' if File.exist?('app/controllers/application.rb')
     files += Dir.glob("app/controllers/**/*_controller.rb") - options.exclude
     files.uniq
   end
   
   # Extract class name from filename