Open Autotest class to add class methods. Use matcher for gets.

Author: justinko

lib/rspec/core/command_line_configuration.rb

@@ -15,36 +15,38 @@ def run
       end
 
       class Autotest
-        def self.generate
-          create_autotest_directory
-          create_discover_file
-          puts "autotest/discover.rb has been added"
-        end
+        class << self
+          def generate
+            create_autotest_directory
+            create_discover_file
+            puts "autotest/discover.rb has been added"
+          end
 
-        def self.create_autotest_directory
-          Dir.mkdir('autotest') unless File.exist?('autotest')
-        end
+          def create_autotest_directory
+            Dir.mkdir('autotest') unless File.exist?('autotest')
+          end
 
-        def self.create_discover_file
-          optionally_remove_discover_file if discover_file_exists?
+          def create_discover_file
+            optionally_remove_discover_file if discover_file_exists?
 
-          File.open(discover_file_path, 'w') do |file|
-            file << 'Autotest.add_discovery { "rspec2" }'
+            File.open(discover_file_path, 'w') do |file|
+              file << 'Autotest.add_discovery { "rspec2" }'
+            end
           end
-        end
 
-        def optionally_remove_discover_file
-          print "Discover file already exists, overwrite [Y/n]? "
-          exit if gets != 'Y'
-          FileUtils.rm_rf(discover_file_path)
-        end
+          def optionally_remove_discover_file
+            print "Discover file already exists, overwrite [y/N]? "
+            exit if gets !~ /y/i
+            FileUtils.rm_rf(discover_file_path)
+          end
 
-        def self.discover_file_exists?
-          File.exist?(discover_file_path)
-        end
+          def discover_file_exists?
+            File.exist?(discover_file_path)
+          end
 
-        def self.discover_file_path
-          File.join('autotest', 'discover.rb')
+          def discover_file_path
+            File.join('autotest', 'discover.rb')
+          end
         end
       end
     end