pry 0.9.7.4 → 0.9.8pre2

data/man/pry.1.ronn

@@ -0,0 +1,141 @@
+PRY(1) -- A Reference to the PRY repl.
+======================================
+
+##Synopsis
+
+
+`pry` [`--version`] [`--exec`] [`--no-pager`] [`--no-history`] [`--no-color`] [`-f`] [`--no-plugins`] [`--installed-plugins`] [`--simple-prompt`] [`--require` _file_] [`-I`] [`--context`] [`--help`]
+
+## DESCRIPTION
+
+
+Pry is a powerful alternative to the standard IRB shell for Ruby. It is written from scratch to provide a number of advanced features.
+
+## HOMEPAGE
+
+
+http://pry.github.com/
+
+##OPTIONS
+
+
+* `-v --version`:
+  Prints the version of Pry.
+
+* `-e --exec`:
+  Executes argument in context before the session starts.
+
+* `--no-pager`:
+  Disable pager for long output.
+
+* `--no-history`:
+  Disable history loading.
+
+* `--no-color`:
+  Disable syntax highlighting for session.
+
+* `-f`:
+  Prevent loading of ~/.pryrc for session.
+
+* `--no-plugins`:
+  Supress loading of plugins.
+
+* `--installed-plugins`:
+  List installed plugins.
+
+* `--simple-prompt`:
+  Enable simple prompt mode (eg, >>).
+
+* `-r --require`:
+  Require a ruby script at startup.
+
+* `-I`:
+  Add a path to the $LOAD_PATH
+
+* `-c --context`:
+  Start the session in the specified context.  Equivalent to `context.pry` in a session.
+
+##FILES
+
+
+~/.pryrc Personal pry initialization
+
+##EXAMPLES
+
+###Basic Usage
+
+
+    $ pry
+    [1] pry(main)>4 + 5
+    => 9
+    [2] pry(main)> def hello_world
+    [2] pry(main)*   puts "Hello, World!"
+    [2] pry(main)* end  
+    => nil
+    [3] pry(main)> hello_world
+    Hello, World!
+    => nil
+
+###Command Line Interaction
+
+
+Prefix any command you want your shell to execute with a period and pry will return the results from your shell.
+
+    [1] pry(main)> .date
+    Fri Nov 11 09:52:07 EST 2011
+
+On the command line enter `shell-mode` to incorporate the current working directory into the Pry prompt.
+
+    pry(main)> shell-mode
+    pry main:/Users/john/ruby/projects/pry $ .cd ..
+    pry main:/Users/john/ruby/projects $ .cd ~
+    pry main:/Users/john $ .pwd
+    /Users/john
+    pry main:/Users/john $ shell-mode
+    pry(main)>
+
+###State Navigation
+
+
+The cd command is used to move into a new object (or scope) inside a Pry session.  When inside the new scope it becomes the self for the session and all commands and methods will operate on this new self.
+
+    pry(main)> self
+    => main
+    pry(main)> cd Pry
+    pry(Pry):1> self
+    => Pry
+    pry(Pry):1> cd ..
+    pry(main)>
+
+The ls command is essentially a unified wrapper to a number of Ruby's introspection mechanisms, including (but not limited to) the following methods: methods, instance\_variables, constants, local\_variables, instance\_methods, class_variables and all the various permutations thereof.
+
+By default typing ls will return a list of just the local and instance variables available in the current context.
+
+* The -M option selects public instance methods (if available).
+* The -m option selects public methods.
+* The -c option selects constants.
+* The -i option select just instance variables.
+* The -l option selects just local variables.
+* The -s option modifies the -c and -m and -M options to go up the superclass chain (excluding Object).
+* The --grep REGEX prunes the list to items that match the regex.
+
+###Source Browsing
+
+
+Simply typing show-method method_name will pull the source for the method and display it with syntax highlighting. You can also look up the source for multiple methods at the same time, by typing show-method method1 method2. As a convenience, Pry looks up both instance methods and class methods using this syntax, with priority given to instance methods.
+
+    pry(Pry):1> show-method rep
+
+    From: /Users/john/ruby/projects/pry/lib/pry/pry_instance.rb @ line 191:
+    Number of lines: 6
+
+    def rep(target=TOPLEVEL_BINDING)
+      target = Pry.binding_for(target)
+      result = re(target)
+
+      show_result(result) if should_print?
+    end
+
+##AUTHORS
+
+Pry is primarily the work of John Mair (banisterfiend)

data/pry.gemspec

@@ -2,48 +2,45 @@
 
 Gem::Specification.new do |s|
   s.name = "pry"
-  s.version = "0.9.7.4"
+  s.version = "0.9.8.0pre1"
 
-  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.required_rubygems_version = Gem::Requirement.new("> 1.3.1") if s.respond_to? :required_rubygems_version=
   s.authors = ["John Mair (banisterfiend)"]
-  s.date = "2011-11-05"
+  s.date = "2011-12-17"
   s.description = "An IRB alternative and runtime developer console"
   s.email = "jrmair@gmail.com"
   s.executables = ["pry"]
-  s.files = [".document", ".gemtest", ".gitignore", ".travis.yml", ".yardopts", "CHANGELOG", "CONTRIBUTORS", "Gemfile", "LICENSE", "README.markdown", "Rakefile", "TODO", "bin/pry", "examples/example_basic.rb", "examples/example_command_override.rb", "examples/example_commands.rb", "examples/example_hooks.rb", "examples/example_image_edit.rb", "examples/example_input.rb", "examples/example_input2.rb", "examples/example_output.rb", "examples/example_print.rb", "examples/example_prompt.rb", "examples/helper.rb", "lib/pry.rb", "lib/pry/command_context.rb", "lib/pry/command_processor.rb", "lib/pry/command_set.rb", "lib/pry/commands.rb", "lib/pry/completion.rb", "lib/pry/config.rb", "lib/pry/core_extensions.rb", "lib/pry/custom_completions.rb", "lib/pry/default_commands/basic.rb", "lib/pry/default_commands/context.rb", "lib/pry/default_commands/documentation.rb", "lib/pry/default_commands/easter_eggs.rb", "lib/pry/default_commands/gems.rb", "lib/pry/default_commands/input.rb", "lib/pry/default_commands/introspection.rb", "lib/pry/default_commands/ls.rb", "lib/pry/default_commands/shell.rb", "lib/pry/extended_commands/experimental.rb", "lib/pry/extended_commands/user_command_api.rb", "lib/pry/helpers.rb", "lib/pry/helpers/base_helpers.rb", "lib/pry/helpers/command_helpers.rb", "lib/pry/helpers/options_helpers.rb", "lib/pry/helpers/text.rb", "lib/pry/history.rb", "lib/pry/history_array.rb", "lib/pry/indent.rb", "lib/pry/method.rb", "lib/pry/plugins.rb", "lib/pry/pry_class.rb", "lib/pry/pry_instance.rb", "lib/pry/rbx_method.rb", "lib/pry/rbx_path.rb", "lib/pry/version.rb", "pry.gemspec", "test/helper.rb", "test/test_command_helpers.rb", "test/test_command_processor.rb", "test/test_command_set.rb", "test/test_completion.rb", "test/test_default_commands.rb", "test/test_default_commands/test_context.rb", "test/test_default_commands/test_documentation.rb", "test/test_default_commands/test_gems.rb", "test/test_default_commands/test_input.rb", "test/test_default_commands/test_introspection.rb", "test/test_default_commands/test_ls.rb", "test/test_default_commands/test_shell.rb", "test/test_exception_whitelist.rb", "test/test_history_array.rb", "test/test_indent.rb", "test/test_input_stack.rb", "test/test_method.rb", "test/test_pry.rb", "test/test_pry_history.rb", "test/test_pry_output.rb", "test/test_special_locals.rb", "test/testrc", "test/testrcbad", "wiki/Customizing-pry.md", "wiki/Home.md"]
+  s.files = [".document", ".gemtest", ".gitignore", ".travis.yml", ".yardopts", "CHANGELOG", "CONTRIBUTORS", "Gemfile", "LICENSE", "README.markdown", "Rakefile", "TODO", "bin/pry", "examples/example_basic.rb", "examples/example_command_override.rb", "examples/example_commands.rb", "examples/example_hooks.rb", "examples/example_image_edit.rb", "examples/example_input.rb", "examples/example_input2.rb", "examples/example_output.rb", "examples/example_print.rb", "examples/example_prompt.rb", "examples/helper.rb", "lib/pry.rb", "lib/pry/cli.rb", "lib/pry/command_context.rb", "lib/pry/command_processor.rb", "lib/pry/command_set.rb", "lib/pry/commands.rb", "lib/pry/completion.rb", "lib/pry/config.rb", "lib/pry/core_extensions.rb", "lib/pry/custom_completions.rb", "lib/pry/default_commands/basic.rb", "lib/pry/default_commands/context.rb", "lib/pry/default_commands/documentation.rb", "lib/pry/default_commands/easter_eggs.rb", "lib/pry/default_commands/gems.rb", "lib/pry/default_commands/input.rb", "lib/pry/default_commands/introspection.rb", "lib/pry/default_commands/ls.rb", "lib/pry/default_commands/shell.rb", "lib/pry/extended_commands/experimental.rb", "lib/pry/extended_commands/user_command_api.rb", "lib/pry/helpers.rb", "lib/pry/helpers/base_helpers.rb", "lib/pry/helpers/command_helpers.rb", "lib/pry/helpers/options_helpers.rb", "lib/pry/helpers/text.rb", "lib/pry/history.rb", "lib/pry/history_array.rb", "lib/pry/hooks.rb", "lib/pry/indent.rb", "lib/pry/method.rb", "lib/pry/plugins.rb", "lib/pry/pry_class.rb", "lib/pry/pry_instance.rb", "lib/pry/rbx_method.rb", "lib/pry/rbx_path.rb", "lib/pry/version.rb", "lib/pry/wrapped_module.rb", "man/pry.1", "man/pry.1.html", "man/pry.1.ronn", "pry.gemspec", "test/helper.rb", "test/test_cli.rb", "test/test_command_helpers.rb", "test/test_command_processor.rb", "test/test_command_set.rb", "test/test_completion.rb", "test/test_default_commands.rb", "test/test_default_commands/test_context.rb", "test/test_default_commands/test_documentation.rb", "test/test_default_commands/test_gems.rb", "test/test_default_commands/test_input.rb", "test/test_default_commands/test_introspection.rb", "test/test_default_commands/test_ls.rb", "test/test_default_commands/test_shell.rb", "test/test_exception_whitelist.rb", "test/test_history_array.rb", "test/test_hooks.rb", "test/test_indent.rb", "test/test_input_stack.rb", "test/test_method.rb", "test/test_pry.rb", "test/test_pry_history.rb", "test/test_pry_output.rb", "test/test_special_locals.rb", "test/test_syntax_checking.rb", "test/test_wrapped_module.rb", "test/testrc", "test/testrcbad", "wiki/Customizing-pry.md", "wiki/Home.md"]
   s.homepage = "http://pry.github.com"
   s.require_paths = ["lib"]
-  s.rubygems_version = "1.8.11"
+  s.rubygems_version = "1.8.10"
   s.summary = "An IRB alternative and runtime developer console"
-  s.test_files = ["test/helper.rb", "test/test_command_helpers.rb", "test/test_command_processor.rb", "test/test_command_set.rb", "test/test_completion.rb", "test/test_default_commands.rb", "test/test_default_commands/test_context.rb", "test/test_default_commands/test_documentation.rb", "test/test_default_commands/test_gems.rb", "test/test_default_commands/test_input.rb", "test/test_default_commands/test_introspection.rb", "test/test_default_commands/test_ls.rb", "test/test_default_commands/test_shell.rb", "test/test_exception_whitelist.rb", "test/test_history_array.rb", "test/test_indent.rb", "test/test_input_stack.rb", "test/test_method.rb", "test/test_pry.rb", "test/test_pry_history.rb", "test/test_pry_output.rb", "test/test_special_locals.rb", "test/testrc", "test/testrcbad"]
+  s.test_files = ["test/helper.rb", "test/test_cli.rb", "test/test_command_helpers.rb", "test/test_command_processor.rb", "test/test_command_set.rb", "test/test_completion.rb", "test/test_default_commands.rb", "test/test_default_commands/test_context.rb", "test/test_default_commands/test_documentation.rb", "test/test_default_commands/test_gems.rb", "test/test_default_commands/test_input.rb", "test/test_default_commands/test_introspection.rb", "test/test_default_commands/test_ls.rb", "test/test_default_commands/test_shell.rb", "test/test_exception_whitelist.rb", "test/test_history_array.rb", "test/test_hooks.rb", "test/test_indent.rb", "test/test_input_stack.rb", "test/test_method.rb", "test/test_pry.rb", "test/test_pry_history.rb", "test/test_pry_output.rb", "test/test_special_locals.rb", "test/test_syntax_checking.rb", "test/test_wrapped_module.rb", "test/testrc", "test/testrcbad"]
 
   if s.respond_to? :specification_version then
     s.specification_version = 3
 
     if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
-      s.add_runtime_dependency(%q<ruby_parser>, [">= 2.3.1"])
-      s.add_runtime_dependency(%q<coderay>, ["~> 0.9.8"])
-      s.add_runtime_dependency(%q<slop>, ["~> 2.1.0"])
-      s.add_runtime_dependency(%q<method_source>, ["~> 0.6.7"])
-      s.add_development_dependency(%q<bacon>, ["~> 1.1.0"])
-      s.add_development_dependency(%q<open4>, ["~> 1.0.1"])
+      s.add_runtime_dependency(%q<coderay>, ["~> 0.9"])
+      s.add_runtime_dependency(%q<slop>, ["< 3", ">= 2.4.1"])
+      s.add_runtime_dependency(%q<method_source>, ["~> 0.6"])
+      s.add_development_dependency(%q<bacon>, ["~> 1.1"])
+      s.add_development_dependency(%q<open4>, ["~> 1.3"])
       s.add_development_dependency(%q<rake>, ["~> 0.9"])
     else
-      s.add_dependency(%q<ruby_parser>, [">= 2.3.1"])
-      s.add_dependency(%q<coderay>, ["~> 0.9.8"])
-      s.add_dependency(%q<slop>, ["~> 2.1.0"])
-      s.add_dependency(%q<method_source>, ["~> 0.6.7"])
-      s.add_dependency(%q<bacon>, ["~> 1.1.0"])
-      s.add_dependency(%q<open4>, ["~> 1.0.1"])
+      s.add_dependency(%q<coderay>, ["~> 0.9"])
+      s.add_dependency(%q<slop>, ["< 3", ">= 2.4.1"])
+      s.add_dependency(%q<method_source>, ["~> 0.6"])
+      s.add_dependency(%q<bacon>, ["~> 1.1"])
+      s.add_dependency(%q<open4>, ["~> 1.3"])
       s.add_dependency(%q<rake>, ["~> 0.9"])
     end
   else
-    s.add_dependency(%q<ruby_parser>, [">= 2.3.1"])
-    s.add_dependency(%q<coderay>, ["~> 0.9.8"])
-    s.add_dependency(%q<slop>, ["~> 2.1.0"])
-    s.add_dependency(%q<method_source>, ["~> 0.6.7"])
-    s.add_dependency(%q<bacon>, ["~> 1.1.0"])
-    s.add_dependency(%q<open4>, ["~> 1.0.1"])
+    s.add_dependency(%q<coderay>, ["~> 0.9"])
+    s.add_dependency(%q<slop>, ["< 3", ">= 2.4.1"])
+    s.add_dependency(%q<method_source>, ["~> 0.6"])
+    s.add_dependency(%q<bacon>, ["~> 1.1"])
+    s.add_dependency(%q<open4>, ["~> 1.3"])
     s.add_dependency(%q<rake>, ["~> 0.9"])
   end
 end

data/test/helper.rb

@@ -3,7 +3,7 @@ unless Object.const_defined? 'Pry'
   require 'pry'
 end
 
-puts "Ruby v#{RUBY_VERSION} (#{defined?(RUBY_ENGINE) ? RUBY_ENGINE : "ruby"}), Pry v#{Pry::VERSION}, method_source v#{MethodSource::VERSION}, CodeRay v#{CodeRay::VERSION}"
+puts "Ruby v#{RUBY_VERSION} (#{defined?(RUBY_ENGINE) ? RUBY_ENGINE : "ruby"}), Pry v#{Pry::VERSION}, method_source v#{MethodSource::VERSION}, CodeRay v#{CodeRay::VERSION}, Slop v#{Slop::VERSION}"
 
 require 'bacon'
 require 'open4'
@@ -25,7 +25,8 @@ class << Pry
     Pry.config.history.should_load = false
     Pry.config.history.should_save = false
     Pry.config.auto_indent         = false
-    Pry.config.hooks = { }
+    Pry.config.hooks               = Pry::Hooks.new
+    Pry.config.collision_warning   = false
   end
 end
 
@@ -163,7 +164,7 @@ end
 # Open a temp file and yield it to the block, closing it after
 # @return [String] The path of the temp file
 def temp_file
-  file = Tempfile.new("tmp")
+  file = Tempfile.new('pry')
   yield file
 ensure
   file.close
@@ -179,3 +180,11 @@ CommandTester = Pry::CommandSet.new do
     output.puts arg
   end
 end
+
+# to help with tracking down bugs that cause an infinite loop in the test suite
+if ENV["SET_TRACE_FUNC"]
+  require 'set_trace' if defined?(RUBY_ENGINE) && RUBY_ENGINE =~ /rbx/ # gem install 'rbx-tracer'
+  set_trace_func proc { |event, file, line, id, binding, classname|
+     STDERR.printf "%8s %s:%-2d %10s %8s\n", event, file, line, id, classname
+  }
+end

data/test/test_cli.rb

@@ -0,0 +1,78 @@
+require 'helper'
+
+describe Pry::Hooks do
+  before do
+    Pry::CLI.reset
+  end
+
+  describe "parsing options" do
+    it 'should raise if no options defined' do
+      lambda { Pry::CLI.parse_options(["--nothing"]) }.should.raise Pry::CLI::NoOptionsError
+    end
+  end
+
+  describe "adding options" do
+    it "should be able to add an option" do
+      run = false
+
+      Pry::CLI.add_options do
+        on :optiontest, "A test option" do
+          run = true
+        end
+      end.parse_options(["--optiontest"])
+
+      run.should == true
+    end
+
+    it "should be able to add multiple options" do
+      run = false
+      run2 = false
+
+      Pry::CLI.add_options do
+        on :optiontest, "A test option" do
+          run = true
+        end
+      end.add_options do
+        on :optiontest2, "Another test option" do
+          run2 = true
+        end
+      end.parse_options(["--optiontest", "--optiontest2"])
+
+      run.should == true
+      run2.should == true
+    end
+
+  end
+
+  describe "processing options" do
+    it "should be able to process an option" do
+      run = false
+
+      Pry::CLI.add_options do
+        on :optiontest, "A test option"
+      end.process_options do |opts|
+        run = true if opts.present?(:optiontest)
+      end.parse_options(["--optiontest"])
+
+      run.should == true
+    end
+
+    it "should be able to  process multiple options" do
+      run = false
+      run2 = false
+
+      Pry::CLI.add_options do
+        on :optiontest, "A test option"
+        on :optiontest2, "Another test option"
+      end.process_options do |opts|
+        run = true if opts.present?(:optiontest)
+      end.process_options do |opts|
+        run2 = true if opts.present?(:optiontest2)
+      end.parse_options(["--optiontest", "--optiontest2"])
+
+      run.should == true
+      run2.should == true
+    end
+
+  end
+end

data/test/test_command_set.rb

@@ -50,6 +50,15 @@ describe Pry::CommandSet do
     }.should.raise(Pry::NoCommandError)
   end
 
+  it 'should be able to remove its own commands, by listing name' do
+    @set.command(/^foo1/, 'desc', :listing => 'foo') {}
+    @set.delete 'foo'
+
+    lambda {
+      @set.run_command @ctx, /^foo1/
+    }.should.raise(Pry::NoCommandError)
+  end
+
   it 'should be able to import some commands from other sets' do
     run = false
 
@@ -68,6 +77,19 @@ describe Pry::CommandSet do
     }.should.raise(Pry::NoCommandError)
   end
 
+  it 'should be able to import some commands from other sets using listing name' do
+    run = false
+
+    other_set = Pry::CommandSet.new do
+      command(/^foo1/, 'desc', :listing => 'foo') { run = true }
+    end
+
+    @set.import_from(other_set, 'foo')
+
+    @set.run_command @ctx, /^foo1/
+    run.should == true
+  end
+
   it 'should be able to import a whole set' do
     run = []
 
@@ -108,13 +130,35 @@ describe Pry::CommandSet do
     run.should == true
   end
 
-  it 'should be able to change the descritpions of methods' do
+  it "should be able to alias a method by the command's listing name" do
+    run = false
+    @set.command(/^foo1/, 'stuff', :listing => 'foo') { run = true }
+
+    @set.alias_command 'bar', 'foo'
+    @set.commands['bar'].name.should == 'bar'
+    @set.commands['bar'].description.should == ''
+
+    @set.run_command @ctx, 'bar'
+    run.should == true
+  end
+
+  it 'should be able to change the descriptions of commands' do
     @set.command('foo', 'bar') {}
     @set.desc 'foo', 'baz'
 
     @set.commands['foo'].description.should == 'baz'
   end
 
+  it 'should get the descriptions of commands' do
+    @set.command('foo', 'bar') {}
+    @set.desc('foo').should == 'bar'
+  end
+
+  it 'should get the descriptions of commands, by listing' do
+    @set.command(/^foo1/, 'bar', :listing => 'foo') {}
+    @set.desc('foo').should == 'bar'
+  end
+
   it 'should return Pry::CommandContext::VOID_VALUE for commands by default' do
     @set.command('foo') { 3 }
     @set.run_command(@ctx, 'foo').should == Pry::CommandContext::VOID_VALUE
@@ -226,4 +270,152 @@ describe Pry::CommandSet do
 
     order.should == order.sort
   end
+
+  describe "renaming a command" do
+    it 'should be able to rename and run a command' do
+      run = false
+      @set.command('foo') { run = true }
+      @set.rename_command('bar', 'foo')
+      @set.run_command(@ctx, 'bar')
+      run.should == true
+    end
+
+    it 'should accept listing name when renaming a command' do
+      run = false
+      @set.command('foo', "", :listing => 'love') { run = true }
+      @set.rename_command('bar', 'love')
+      @set.run_command(@ctx, 'bar')
+      run.should == true
+    end
+
+    it 'should raise exception trying to rename non-existent command' do
+      lambda { @set.rename_command('bar', 'foo') }.should.raise ArgumentError
+    end
+
+    it 'should make old command name inaccessible' do
+      @set.command('foo') { }
+      @set.rename_command('bar', 'foo')
+      lambda { @set.run_command(@ctx, 'foo') }.should.raise Pry::NoCommandError
+    end
+
+
+    it 'should be able to pass in options when renaming command' do
+      desc    = "hello"
+      listing = "bing"
+      @set.command('foo') { }
+      @set.rename_command('bar', 'foo', :description => desc, :listing => listing, :keep_retval => true)
+      @set.commands['bar'].description.should           == desc
+      @set.commands['bar'].options[:listing].should     == listing
+      @set.commands['bar'].options[:keep_retval].should == true
+    end
+  end
+
+  describe "command decorators - before_command and after_command" do
+    describe "before_command" do
+      it 'should be called before the original command' do
+        foo = []
+        @set.command('foo') { foo << 1 }
+        @set.before_command('foo') { foo << 2 }
+        @set.run_command(@ctx, 'foo')
+
+        foo.should == [2, 1]
+      end
+
+      it 'should be called before the original command, using listing name' do
+        foo = []
+        @set.command(/^foo1/, '', :listing => 'foo') { foo << 1 }
+        @set.before_command('foo') { foo << 2 }
+        @set.run_command(@ctx, /^foo1/)
+
+        foo.should == [2, 1]
+      end
+
+      it 'should share the context with the original command' do
+        @ctx.target = "test target string"
+        before_val  = nil
+        orig_val    = nil
+        @set.command('foo') { orig_val = target }
+        @set.before_command('foo') { before_val = target }
+        @set.run_command(@ctx, 'foo')
+
+        before_val.should == @ctx.target
+        orig_val.should == @ctx.target
+      end
+
+      it 'should work when applied multiple times' do
+        foo = []
+        @set.command('foo') { foo << 1 }
+        @set.before_command('foo') { foo << 2 }
+        @set.before_command('foo') { foo << 3 }
+        @set.before_command('foo') { foo << 4 }
+        @set.run_command(@ctx, 'foo')
+
+        foo.should == [4, 3, 2, 1]
+      end
+
+    end
+
+    describe "after_command" do
+      it 'should be called after the original command' do
+        foo = []
+        @set.command('foo') { foo << 1 }
+        @set.after_command('foo') { foo << 2 }
+        @set.run_command(@ctx, 'foo')
+
+        foo.should == [1, 2]
+      end
+
+      it 'should be called after the original command, using listing name' do
+        foo = []
+        @set.command(/^foo1/, '', :listing => 'foo') { foo << 1 }
+        @set.after_command('foo') { foo << 2 }
+        @set.run_command(@ctx, /^foo1/)
+
+        foo.should == [1, 2]
+      end
+
+      it 'should share the context with the original command' do
+        @ctx.target = "test target string"
+        after_val  = nil
+        orig_val    = nil
+        @set.command('foo') { orig_val = target }
+        @set.after_command('foo') { after_val = target }
+        @set.run_command(@ctx, 'foo')
+
+        after_val.should == @ctx.target
+        orig_val.should == @ctx.target
+      end
+
+      it 'should determine the return value for the command' do
+        @set.command('foo', 'bar', :keep_retval => true) { 1 }
+        @set.after_command('foo') { 2 }
+        @set.run_command(@ctx, 'foo').should == 2
+      end
+
+      it 'should work when applied multiple times' do
+        foo = []
+        @set.command('foo') { foo << 1 }
+        @set.after_command('foo') { foo << 2 }
+        @set.after_command('foo') { foo << 3 }
+        @set.after_command('foo') { foo << 4 }
+        @set.run_command(@ctx, 'foo')
+
+        foo.should == [1, 2, 3, 4]
+      end
+    end
+
+    describe "before_command and after_command" do
+      it 'should work when combining both before_command and after_command' do
+        foo = []
+        @set.command('foo') { foo << 1 }
+        @set.after_command('foo') { foo << 2 }
+        @set.before_command('foo') { foo << 3 }
+        @set.run_command(@ctx, 'foo')
+
+        foo.should == [3, 1, 2]
+      end
+
+    end
+
+  end
 end