lib/capy.rb in capy-1.1.0 vs lib/capy.rb in capy-1.1.2

@@ -10,19 +10,22 @@
 
     def run(args)
       @opts = Slop.parse!(args, :help => true) do
         banner "capy [script.capy]\n"
         on :b, :browser=, 'chrome, firefox', :default => 'chrome'
+        on :h, :host=, 'app host'
         on :n, :nonstop
       end
       exit if opts.help?
 
       Capybara.register_driver :selenium do |app|
         Capybara::Selenium::Driver.new(app, :browser => opts[:browser].to_sym)
       end
       Capybara.current_driver = :selenium
 
+      Capybara.app_host = opts[:host] if opts[:host]
+
       if args.empty?
         start_shell
       else
         args.each do |script_file|
           abort "No such file: #{script_file}".red unless File.exists?(script_file)
@@ -34,11 +37,11 @@
 
     def start_shell(evaluater = Evaluater.new)
       exit_commands = %w(exit quit)
 
       Readline.completion_proc = lambda do |text|
-        (Capybara::DSL.instance_methods + exit_commands).grep(/^#{Regexp.quote(text.strip)}/)
+        (Evaluater.instance_methods - Object.methods + exit_commands).grep(/^#{Regexp.quote(text.strip)}/)
       end
 
       history_file = File.expand_path('~/.capy_history')
       if File.exists?(history_file)
         File.read(history_file, :encoding => "BINARY").
@@ -91,7 +94,11 @@
     end
   end
 
   class Evaluater
     include Capybara::DSL
+
+    def host(app_host)
+      Capybara.app_host = app_host
+    end
   end
 end