pry-note 0.2.1 → 0.2.3

data/Rakefile

@@ -6,7 +6,7 @@ direc = File.dirname(__FILE__)
 PROJECT_NAME = "pry-note"
 
 require 'rake/clean'
-require 'rake/gempackagetask'
+require 'rubygems/package_task'
 require "#{PROJECT_NAME}/version"
 
 CLOBBER.include("**/*~", "**/*#*", "**/*.log")
@@ -28,11 +28,21 @@ def apply_spec_defaults(s)
   s.test_files = `git ls-files -- test/*`.split("\n")
 end
 
+desc "Set up and run tests"
+task :default => [:test]
+
+
 desc "run pry with plugin enabled"
 task :pry do
   exec("pry -rubygems -I#{direc}/lib/ -r #{direc}/lib/#{PROJECT_NAME}")
 end
 
+desc "Run tests"
+task :test do
+  sh "bacon -Itest -rubygems -a -q"
+end
+task :spec => :test
+
 desc "Show version"
 task :version do
   puts "Pry-note version: #{PryNote::VERSION}"
@@ -47,7 +57,7 @@ namespace :ruby do
     s.platform = Gem::Platform::RUBY
   end
 
-  Rake::GemPackageTask.new(spec) do |pkg|
+  Gem::PackageTask.new(spec) do |pkg|
     pkg.need_zip = false
     pkg.need_tar = false
   end

data/lib/pry-note.rb

@@ -1,222 +1,27 @@
-Pry.config.notes_file = "./notes.yml"
-
-Pry::Commands.create_command "note" do
-  description "Note stuff."
-
-  banner <<-USAGE
-    Usage: note [OPTIONS]
-    Add notes to classes and methods.
-
-    e.g note -a Pry#repl "this is my note" #=> add a note without opening editor
-    e.g note -a Pry#repl   #=> add a note (with editor) to Pry#repl method
-    e.g note -d Pry#repl:1 #=> delete the 1st note from Pry#repl
-    e.g note -d Pry#repl   #=> delete all notes from Pry#repl
-    e.g note -l            #=> list all notes
-  USAGE
-
-  def options(opt)
-    opt.on :a, :add, "Add a note to a method or class.", :argument => true
-    opt.on :s, :show, "Show any notes associated with the given method or class.", :argument => true
-    opt.on :d, :delete, "Delete notes for a method or class.", :argument => true
-    opt.on "delete-all", "Delete all notes."
-    opt.on :e, :export, "Export notes to a file.", :argument => :optional
-    opt.on :load, "Load notes from a file.", :argument => :optional
-    opt.on :l, :list, "List all notes."
-    opt.on "list-all", "List all notes with content."
-  end
-
-  def setup
-    if !state.initial_setup_complete
-      add_reminders
-      load_notes
-      state.initial_setup_complete = true
-      _pry_.hooks.add_hook(:after_session, :export_notes) { export_notes }
-    end
-  end
-
-  def notes
-    state.notes ||= {}
-  end
-
-  # edit a note in a temporary file and return note content
-  def edit_note(obj_name)
-    temp_file do |f|
-      f.puts("Enter note content here for #{obj_name} (and erase this line)")
-      f.flush
-      f.close(false)
-      invoke_editor(f.path, 1, false)
-      File.read(f.path)
-    end
-  end
-
-  def process
-    if opts.present?(:add)
-      add_note(opts[:a])
-    elsif opts.present?(:show)
-      show_note(opts[:s])
-    elsif opts.present?(:list)
-      list_notes
-    elsif opts.present?(:export)
-      export_notes(opts[:e])
-    elsif opts.present?(:delete)
-      delete_note(opts[:d])
-    elsif opts.present?(:"delete-all")
-      notes.replace({})
-    elsif opts.present?(:"list-all")
-      list_all
-    elsif opts.present?(:load)
-      load_notes(opts[:load])
-    else
-      meth = Pry::Method.from_binding(target)
-      if internal_binding?(target) || !meth
-        obj = target.eval("self")
-        obj_name = obj.is_a?(Module) ? obj.name : obj.class.name
-        add_note(obj_name)
-      else
-        add_note(meth.name_with_owner)
-      end
-    end
-  end
-
-  def retrieve_code_object_safely(name)
-    code_object = retrieve_code_object_from_string(name, target)
-
-    if !code_object
-      raise Pry::CommandError, "No code object found named #{name}"
-    elsif code_object.name.to_s == ""
-      raise Pry::CommandError, "Object #{name} doesn't have a proper name, can't create note"
-    end
-
-    code_object
-  end
-
-  def code_object_name(co)
-    co.is_a?(Pry::Method) ? co.name_with_owner : co.name
-  end
-
-  def add_note(name)
-    co_name = code_object_name(retrieve_code_object_safely(name))
-
-    if args.any?
-      note = args.join(" ")
-    else
-      note = edit_note(co_name)
-    end
-
-    notes[co_name] ||= []
-    notes[co_name] << note
-
-    output.puts "Added note to #{co_name}!"
-  end
-
-  def delete_note(name)
-    name, note_number = name.split(":")
-    co_name = code_object_name(retrieve_code_object_safely(name))
+require 'yaml'
+require 'pry-note/version'
+require 'pry-note/hooks'
+require 'pry-note/commands'
 
-    if !notes[co_name]
-      output.puts "No notes to delete for #{co_name}!"
-    elsif note_number
-      notes[co_name].delete_at(note_number.to_i - 1)
-      notes.delete(co_name) if notes[co_name].empty?
-      output.puts "Deleted note #{note_number} for #{co_name}!"
-    else
-      notes.delete(co_name)
-      output.puts "Deleted all notes for #{text.bold(co_name)}!"
-    end
-  end
-
-  def show_note(name)
-    code_object = retrieve_code_object_safely(name)
-
-    co_name = code_object_name(code_object)
-
-    if !notes.has_key?(co_name)
-      output.puts "No notes saved for #{text.bold(co_name)}"
-      return
-    end
-
-    output.puts text.bold("#{co_name}:\n--")
-
-    notes[code_object_name(code_object)].each_with_index do |note, index|
-      output.puts "\nNote #{text.bold((index + 1).to_s)}: #{note}"
-    end
-  end
-
-  def export_notes(file_name=nil)
-    require 'yaml'
-    file_name ||= Pry.config.notes_file
+Pry.config.notes_file = "./notes.yml"
 
-    expanded_path = File.expand_path(file_name)
-    File.open(expanded_path, "w") { |f| f.puts YAML.dump(notes) }
-    output.puts "Exported notes to #{expanded_path}!"
-  end
+module PryNote
+  def self.notes() @notes ||= {}; end
+  def self.notes=(o) @notes = o; end
 
-  def load_notes(file_name=nil)
-    require 'yaml'
+  def self.load_notes(file_name=nil)
+    return if !file_name && !Pry.config.notes_file
     file_name ||= Pry.config.notes_file
-
     expanded_path = File.expand_path(file_name)
     if File.exists?(expanded_path)
-      notes.replace YAML.load File.read(expanded_path)
-    end
-  end
-
-  def list_all
-    if notes.any?
-      output.puts text.bold("Showing all available notes:\n\n")
-      notes.each do |key, content|
-        begin
-          show_note(key)
-          output.puts "\n"
-        rescue
-        end
-      end
-
-      output.puts "\nTo view notes for an item type, e.g: `note -s Klass#method`"
-    else
-      output.puts "No notes available."
+      PryNote.notes = YAML.load File.read(expanded_path)
     end
   end
 
-  def list_notes
-    if notes.any?
-      output.puts text.bold("Showing all available notes:\n\n")
-      notes.each do |key, content|
-        if retrieve_code_object_from_string(key, target)
-          output.puts "#{text.bold(key)} has #{content.count} notes"
-        end
-      end
-
-      output.puts "\nTo view notes for an item type, e.g: `note -s Klass#method`"
-    else
-      output.puts "No notes available."
-    end
-  end
-
-  def add_reminders
-    me = self
-    reminder = proc do
-      begin
-        code_object = retrieve_code_object_from_string(args.first.to_s, target)
-        if me.notes.keys.include?(me.code_object_name(code_object))
-          co_name = me.code_object_name(code_object)
-          output.puts "\n\n#{text.bold("Notes:")}\n--\n\n"
-
-          me.notes[me.code_object_name(code_object)].each_with_index do |note, index|
-            clipped_note = note.lines.count < 3  ? note : note.lines.to_a[0..2].join +
-              text.bold("<...clipped...>") + " Use `note -s #{co_name}` to view unelided notes."
-            amended_note = clipped_note.lines.each_with_index.map do |line, idx|
-              idx > 0 ? "#{' ' * ((index + 1).to_s.size + 2)}#{line}" : line
-            end.join
-            output.puts "#{text.bold((index + 1).to_s)}. #{amended_note}"
-          end
-
-        end
-      rescue
-      end
-    end
-
-    _pry_.commands.after_command("show-source", &reminder)
-    _pry_.commands.after_command("show-doc", &reminder)
+  def self.export_notes(file_name=nil)
+    return if !file_name && !Pry.config.notes_file
+    file_name ||= Pry.config.notes_file
+    expanded_path = File.expand_path(file_name)
+    File.open(expanded_path, "w") { |f| f.puts YAML.dump(PryNote.notes) }
   end
 end

data/lib/pry-note/commands.rb

@@ -0,0 +1,242 @@
+Pry::Commands.create_command "note" do
+  description "Note stuff."
+
+  banner <<-USAGE
+    Usage: note [OPTIONS]
+    Add notes to classes and methods.
+
+    e.g note -a Pry#repl "this is my note" #=> add a note without opening editor
+    e.g note -a Pry#repl   #=> add a note (with editor) to Pry#repl method
+    e.g note -d Pry#repl:1 #=> delete the 1st note from Pry#repl
+    e.g note -d Pry#repl   #=> delete all notes from Pry#repl
+    e.g note -l            #=> list all notes
+  USAGE
+
+  def subcommands(cmd)
+    cmd.on :add do |opt|
+      opt.on :m, "message", "Provide the note inline (without opening an editor).", :argument => true
+    end
+
+    cmd.on :show  do |opt|
+      opt.on :v, :verbose, "Show all notes together with source code."
+    end
+
+    cmd.on :list do |opt|
+      opt.on :v, :verbose, "List all notes and content with source code."
+    end
+    cmd.on :export
+    cmd.on :load
+    cmd.on :delete do |opt|
+      opt.on :all, "Delete all notes."
+    end
+
+    cmd.on :edit do |opt|
+      opt.on :m, "message", "Update the note inline (without opening an editor).", :argument => true
+    end
+  end
+
+  def notes() PryNote.notes ||= {} end
+  def notes=(o) PryNote.notes = o; end
+
+  # edit a note in a temporary file and return note content
+  def edit_note(obj_name, initial_content=nil)
+    initial_content ||= "Enter note content here for #{obj_name} (and erase this line)"
+    temp_file do |f|
+      f.puts(initial_content)
+      f.flush
+      f.close(false)
+      invoke_editor(f.path, 1, false)
+      File.read(f.path)
+    end
+  end
+
+  def process
+    if opts.command?(:add)
+      cmd_opts = opts[:add]
+      add_note(opts.arguments.first, cmd_opts[:message])
+    elsif opts.command?(:show)
+      cmd_opts = opts[:show]
+      stagger_output create_note_output(opts.arguments.first, cmd_opts[:verbose])
+    elsif opts.command?(:list)
+      cmd_opts = opts[:list]
+      if cmd_opts.present?(:verbose)
+        list_all
+      else
+        list_notes
+      end
+    elsif opts.command?(:edit)
+      cmd_opts = opts[:edit]
+      reedit_note(opts.arguments.first, cmd_opts[:message])
+    elsif opts.command?(:export)
+      f = opts.arguments.first
+      PryNote.export_notes(f)
+      output.puts "Exported notes to #{f}"
+    elsif opts.command?(:delete)
+      cmd_opts = opts[:delete]
+      if cmd_opts.present?(:all)
+        notes.replace({})
+        output.puts "Deleted all notes!"
+      else
+        delete_note(opts.arguments.first)
+      end
+    elsif opts.command?(:load)
+      PryNote.load_notes(opts.arguments.first)
+    else
+      output.puts opts.to_s
+    end
+  end
+
+  def retrieve_code_object_safely(name)
+    code_object = retrieve_code_object_from_string(name, target)
+
+    if !code_object
+      raise Pry::CommandError, "No code object found named #{name}"
+    elsif code_object.name.to_s == ""
+      raise Pry::CommandError, "Object #{name} doesn't have a proper name, can't create note"
+    end
+
+    code_object
+  end
+
+  def default_object_name
+    meth = Pry::Method.from_binding(target)
+    if internal_binding?(target) || !meth
+      obj = target.eval("self")
+      obj_name = obj.is_a?(Module) ? obj.name : obj.class.name
+      obj_name
+    else
+      meth.name_with_owner
+    end
+  end
+
+  def code_object_name(co)
+    co.is_a?(Pry::Method) ? co.name_with_owner : co.name
+  end
+
+  def add_note(name, message=nil)
+    name ||= default_object_name
+    co_name = code_object_name(retrieve_code_object_safely(name))
+
+    if message
+      note = message
+    else
+      note = edit_note(co_name)
+    end
+
+    notes[co_name] ||= []
+    notes[co_name] << note
+
+    output.puts "Added note to #{co_name}!"
+  end
+
+  def reedit_note(name, message=nil)
+    name, note_number_s = name.split(/:(\d+)$/)
+    co_name = code_object_name(retrieve_code_object_safely(name))
+    raise Pry::CommandError, "No notes to edit!" if !notes[co_name]
+
+    total_notes = notes[co_name].count
+    note_number = note_number_s.to_i
+
+    out = ""
+    if !notes[co_name]
+      out << "No notes to edit for #{co_name}!\n"
+    elsif !note_number_s
+      raise Pry::CommandError, "Must specify a note number. Allowable range is 1-#{total_notes}."
+    elsif note_number < 1 || note_number > total_notes
+      raise Pry::CommandError,  "Invalid note number (#{note_number}). Allowable range is 1-#{total_notes}."
+    else
+      if message
+        new_content = message
+      else
+        old_content = notes[co_name][note_number.to_i - 1]
+        new_content = edit_note(co_name, old_content.to_s)
+      end
+
+      notes[co_name][note_number.to_i - 1] = new_content
+      out << "Updated note #{note_number} for #{co_name}!\n"
+    end
+  end
+
+  def delete_note(name)
+    name, note_number = name.split(/:(\d+)$/)
+    co_name = code_object_name(retrieve_code_object_safely(name))
+
+    out = ""
+    if !notes[co_name]
+      out << "No notes to delete for #{co_name}!\n"
+    elsif note_number
+      notes[co_name].delete_at(note_number.to_i - 1)
+      notes.delete(co_name) if notes[co_name].empty?
+      out << "Deleted note #{note_number} for #{co_name}!\n"
+    else
+      notes.delete(co_name)
+      out << "Deleted all notes for #{text.bold(co_name)}!\n"
+    end
+
+    stagger_output out
+  end
+
+  def create_note_output(name, verbose=false)
+    name ||= default_object_name
+    name, _ = name.split(/:(\d+)$/)
+    code_object = retrieve_code_object_safely(name)
+    co_name = code_object_name(code_object)
+
+    raise Pry::CommandError, "Please specify the name of a method or class." if !name
+
+    if !notes.has_key?(co_name)
+      raise Pry::CommandError, "No notes saved for #{text.bold(co_name)}"
+    end
+
+    out = ""
+    out << text.bold("#{co_name}:\n--\n")
+
+    if verbose
+      out << Pry::Code.new(code_object.source, code_object.source_line).with_line_numbers.to_s + "\n"
+    end
+    notes[code_object_name(code_object)].each_with_index do |note, index|
+      out << "\nNote #{text.bold((index + 1).to_s)}: #{note}"
+    end
+
+    out
+  end
+
+  def list_all
+    if notes.any?
+      out = ""
+      out << text.bold("Showing all available notes:\n\n")
+      notes.each do |key, content|
+        begin
+          out << create_note_output(key, true) << "\n"
+        rescue
+        end
+      end
+
+     else
+      out << "No notes available.\n"
+    end
+
+    stagger_output out
+  end
+
+  def list_notes
+    if notes.any?
+      out = ""
+      out << text.bold("Showing all available notes:\n\n")
+      notes.each do |key, content|
+        begin
+          if retrieve_code_object_from_string(key, target)
+            out << "#{text.bold(key)} has #{content.count} notes\n"
+          end
+        rescue
+        end
+      end
+
+      out << "\nTo view notes for a specific item, e.g: `note show Klass#method`\n"
+    else
+      out << "No notes available.\n"
+    end
+
+    stagger_output out
+  end
+end

data/lib/pry-note/hooks.rb

@@ -0,0 +1,31 @@
+reminder = proc do
+  begin
+    co = Pry::Helpers::CommandHelpers.retrieve_code_object_from_string(args.first.to_s, target)
+    co_name = co.is_a?(Pry::Method) ? co.name_with_owner : co.name
+    if PryNote.notes.keys.include?(co_name)
+      output.puts "\n\n#{text.bold("Notes:")}\n--\n\n"
+
+      PryNote.notes[co_name].each_with_index do |note, index|
+        clipped_note = note.lines.count < 3  ? note : note.lines.to_a[0..2].join +
+          text.bold("<...clipped...>") + " Use `note show #{co_name}` to view unelided notes."
+        amended_note = clipped_note.lines.each_with_index.map do |line, idx|
+          idx > 0 ? "#{' ' * ((index + 1).to_s.size + 2)}#{line}" : line
+        end.join
+        output.puts "#{text.bold((index + 1).to_s)}. #{amended_note}"
+      end
+
+    end
+  rescue
+  end
+end
+
+Pry.commands.after_command("show-source", &reminder)
+Pry.commands.after_command("show-doc", &reminder)
+
+Pry.config.hooks.add_hook(:when_started, :load_notes) do
+  PryNote.load_notes if PryNote.notes.empty?
+end
+
+Pry.config.hooks.add_hook(:after_session, :export_notes) do
+  PryNote.export_notes
+end

data/lib/pry-note/version.rb

@@ -1,3 +1,3 @@
 module PryNote
-  VERSION = "0.2.1"
+  VERSION = "0.2.3"
 end

data/pry-note.gemspec

@@ -0,0 +1,30 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = "pry-note"
+  s.version = "0.2.3"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["John Mair (banisterfiend)"]
+  s.date = "2012-12-01"
+  s.description = "Ease refactoring and exploration by attaching notes to methods and classes in Pry"
+  s.email = "jrmair@gmail.com"
+  s.files = ["Rakefile", "lib/pry-note.rb", "lib/pry-note/commands.rb", "lib/pry-note/hooks.rb", "lib/pry-note/version.rb", "pry-note.gemspec", "test/helper.rb", "test/test_pry_note.rb"]
+  s.homepage = "https://github.com/banister"
+  s.require_paths = ["lib"]
+  s.rubygems_version = "1.8.23"
+  s.summary = "Ease refactoring and exploration by attaching notes to methods and classes in Pry"
+  s.test_files = ["test/helper.rb", "test/test_pry_note.rb"]
+
+  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_development_dependency(%q<rake>, ["~> 0.9"])
+    else
+      s.add_dependency(%q<rake>, ["~> 0.9"])
+    end
+  else
+    s.add_dependency(%q<rake>, ["~> 0.9"])
+  end
+end

data/test/helper.rb

@@ -0,0 +1,26 @@
+require 'pry/test/helper'
+
+unless Object.const_defined? 'PryNote'
+  $:.unshift File.expand_path '../../lib', __FILE__
+  require 'pry-note'
+end
+
+# Ensure file is deleted before and after block
+def cleanup_file(file_name)
+  f = File.expand_path(file_name)
+  File.unlink(f) if File.exists?(f)
+  yield
+ensure
+  File.unlink(f) if File.exists?(f)
+end
+
+# Return any raised exceptino objects inside the block
+def capture_exception
+  ex = nil
+  begin
+    yield
+  rescue Exception => e
+    ex = e
+  end
+  ex
+end

data/test/test_pry_note.rb

@@ -0,0 +1,240 @@
+require 'helper'
+
+# useful test class
+class PryNote::TestClass
+  def ping
+    binding
+  end
+end
+
+describe PryNote do
+  before do
+    Pad.obj = PryNote::TestClass.new
+    @t = pry_tester
+    PryNote.notes = {}
+    Pry.config.notes_file = nil
+  end
+
+  after do
+    Pad.clear
+  end
+
+  describe "note add" do
+    describe "opens an editor when -m flag not provided" do
+      it 'should open the editor' do
+        used_editor = nil
+        Pry.config.editor = proc { used_editor = true; nil }
+        @t.process_command "note add PryNote::TestClass"
+        used_editor.should == true
+      end
+
+      it 'should save the note' do
+        Pry.config.editor = proc { nil }
+        @t.process_command "note add PryNote::TestClass"
+        PryNote.notes["PryNote::TestClass"].count.should == 1
+      end
+
+      it 'should put default note content in file' do
+        Pry.config.editor = proc { nil }
+        @t.process_command "note add PryNote::TestClass"
+        PryNote.notes["PryNote::TestClass"].first.should =~ /Enter note content here/
+      end
+    end
+
+    describe "explicit object" do
+      it 'should add a new note for a method (bound method)' do
+        @t.process_command "note add Pad.obj.ping -m 'my note'"
+        @t.last_output.should =~ /Added note to PryNote::TestClass#ping/
+        PryNote.notes["PryNote::TestClass#ping"].first.should =~ /my note/
+      end
+
+      it 'should add a new note for a method (unbound method)' do
+        @t.process_command "note add PryNote::TestClass#ping -m 'my note'"
+        @t.last_output.should =~ /Added note to PryNote::TestClass#ping/
+        PryNote.notes["PryNote::TestClass#ping"].first.should =~ /my note/
+      end
+
+      it 'should add a new note for a class' do
+        @t.process_command "note add PryNote::TestClass -m 'my note'"
+        @t.last_output.should =~ /Added note to PryNote::TestClass/
+        PryNote.notes["PryNote::TestClass"].first.should =~ /my note/
+      end
+    end
+
+    describe "implicit object" do
+      it 'should add a new note for class of object implicitly (without specifying object)' do
+        @t.process_command "cd 0"
+        @t.process_command "note add -m 'my note'"
+        @t.last_output.should =~ /Added note to Fixnum/
+        PryNote.notes["Fixnum"].first.should =~ /my note/
+      end
+
+      it 'should add a new note for a method implicitly (without specifying object)' do
+        o = PryNote::TestClass.new
+        t = pry_tester(o.ping)
+        t.process_command "note add -m 'my note'"
+        t.last_output.should =~ /Added note to PryNote::TestClass#ping/
+        PryNote.notes["PryNote::TestClass#ping"].first.should =~ /my note/
+      end
+    end
+
+    describe "multiple notes can be added" do
+      it 'should add multiple notes' do
+        @t.process_command "note add PryNote::TestClass -m 'my note1'"
+        @t.process_command "note add PryNote::TestClass -m 'my note2'"
+        PryNote.notes["PryNote::TestClass"].count.should == 2
+        PryNote.notes["PryNote::TestClass"].first.should =~ /my note1/
+        PryNote.notes["PryNote::TestClass"].last.should =~ /my note2/
+      end
+    end
+  end
+
+  describe "note delete" do
+    it 'should delete all notes for an object' do
+      @t.process_command "note add PryNote::TestClass -m 'my note'"
+      PryNote.notes["PryNote::TestClass"].count.should == 1
+      @t.process_command "note delete PryNote::TestClass"
+      @t.last_output.should =~ /Deleted all notes for PryNote::TestClass/
+      PryNote.notes["PryNote::TestClass"].should == nil
+    end
+
+    it 'should NOT delete notes for unspecified object' do
+      @t.process_command "note add PryNote::TestClass -m 'my note'"
+      @t.process_command "note add PryNote::TestClass#ping -m 'my note'"
+      @t.process_command "note delete PryNote::TestClass"
+      PryNote.notes["PryNote::TestClass#ping"].count.should == 1
+    end
+
+    it 'should delete all notes for all objects' do
+      @t.process_command "note add PryNote::TestClass -m 'my note'"
+      @t.process_command "note add PryNote::TestClass#ping -m 'my note'"
+      PryNote.notes.keys.count.should == 2
+      @t.process_command "note delete --all"
+      @t.last_output.should =~ /Deleted all notes/
+      PryNote.notes.empty?.should == true
+    end
+
+    describe "deleting specific notes for an object" do
+      it 'should delete first note for an object' do
+        @t.process_command "note add PryNote::TestClass -m 'my note1'"
+        @t.process_command "note add PryNote::TestClass -m 'my note2'"
+        PryNote.notes["PryNote::TestClass"].count.should == 2
+        @t.process_command "note delete PryNote::TestClass:1"
+        @t.last_output.should =~ /Deleted note 1 for PryNote::TestClass/
+        PryNote.notes["PryNote::TestClass"].count.should == 1
+        PryNote.notes["PryNote::TestClass"].first.should =~ /my note2/
+      end
+
+      it 'should delete middle note for an object' do
+        @t.process_command "note add PryNote::TestClass -m 'my note1'"
+        @t.process_command "note add PryNote::TestClass -m 'my note2'"
+        @t.process_command "note add PryNote::TestClass -m 'my note3'"
+        PryNote.notes["PryNote::TestClass"].count.should == 3
+        @t.process_command "note delete PryNote::TestClass:2"
+        @t.last_output.should =~ /Deleted note 2 for PryNote::TestClass/
+        PryNote.notes["PryNote::TestClass"].count.should == 2
+        PryNote.notes["PryNote::TestClass"].first.should =~ /my note1/
+        PryNote.notes["PryNote::TestClass"].last.should =~ /my note3/
+      end
+    end
+  end
+
+  describe "note edit" do
+    describe "errors" do
+      it 'should error when not given a note number' do
+        @t.process_command "note add PryNote::TestClass -m 'my note1'"
+
+        capture_exception do
+          @t.process_command "note edit PryNote::TestClass -m 'bing'"
+        end.message.should =~ /Must specify a note number/
+      end
+
+      it 'should error when given out of range note number' do
+        @t.process_command "note add PryNote::TestClass -m 'my note1'"
+
+        capture_exception do
+          @t.process_command "note edit PryNote::TestClass:2 -m 'bing'"
+        end.message.should =~ /Invalid note number/
+      end
+
+      it 'should error when editing object with no notes' do
+        capture_exception do
+          @t.process_command "note edit PryNote::TestClass:2 -m 'bing'"
+        end.message.should =~ /No notes to edit/
+      end
+    end
+
+    describe "-m switch" do
+      it 'should amend the content of a note' do
+        @t.process_command "note add PryNote::TestClass -m 'my note1'"
+        @t.process_command "note edit PryNote::TestClass:1 -m 'bing'"
+        PryNote.notes["PryNote::TestClass"].count.should == 1
+        PryNote.notes["PryNote::TestClass"].first.should =~ /bing/
+      end
+    end
+  end
+
+  describe "note show" do
+    it 'should display method source when -v flag is used' do
+      @t.process_command "note add PryNote::TestClass -m 'my note1'"
+      @t.process_command "note show PryNote::TestClass -v"
+      @t.last_output.should =~ /ping/
+    end
+
+    it 'should just display number of notes by default' do
+      @t.process_command "note add PryNote::TestClass -m 'my note1'"
+      @t.process_command "note add PryNote::TestClass -m 'my note2'"
+      @t.process_command "note show PryNote::TestClass"
+      @t.last_output.should =~ /2/
+      @t.last_output.should.not =~ /ping/
+    end
+
+    it 'should ignore :number suffix (as used in edit and delete)' do
+      @t.process_command "note add PryNote::TestClass -m 'my note2'"
+      @t.process_command "note show PryNote::TestClass:99"
+      @t.last_output.should =~ /1/
+    end
+
+    it 'should implicitly display notes for current object (class)' do
+      @t.process_command "note add PryNote::TestClass -m 'my note1'"
+      @t.process_command "note add PryNote::TestClass -m 'my note2'"
+      @t.process_command "cd PryNote::TestClass"
+      @t.process_command "note show -v"
+      @t.last_output.should =~ /ping/
+    end
+
+    it 'should implicitly display notes for current object (method)' do
+      t = pry_tester(Pad.obj.ping)
+      t.process_command "note add PryNote::TestClass#ping -m 'my note1'"
+      t.process_command "note add PryNote::TestClass#ping -m 'my note2'"
+      t.process_command "note show -v"
+      t.last_output.should =~ /binding/
+    end
+  end
+
+  describe "note export" do
+    it 'should export to Pry.config.notes_file by default' do
+      cleanup_file("bing.yml") do
+        Pry.config.notes_file = "bing.yml"
+        @t.process_command "note add PryNote::TestClass -m 'my note1'"
+        @t.process_command "note add PryNote::TestClass -m 'my note2'"
+        @t.process_command "note export"
+
+        o = YAML.load(File.read("bing.yml"))
+        o["PryNote::TestClass"].should == ['my note1', 'my note2']
+
+        Pry.config.notes_file = nil
+      end
+    end
+
+    it 'should export to specified file' do
+      cleanup_file("blah.yml") do
+        @t.process_command "note add PryNote::TestClass -m 'my note1'"
+        @t.process_command "note add PryNote::TestClass -m 'my note2'"
+        @t.process_command "note export blah.yml"
+        o = YAML.load(File.read("blah.yml"))
+        o["PryNote::TestClass"].should == ['my note1', 'my note2']
+      end
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pry-note
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-11-22 00:00:00.000000000 Z
+date: 2012-12-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -36,7 +36,12 @@ extra_rdoc_files: []
 files:
 - Rakefile
 - lib/pry-note.rb
+- lib/pry-note/commands.rb
+- lib/pry-note/hooks.rb
 - lib/pry-note/version.rb
+- pry-note.gemspec
+- test/helper.rb
+- test/test_pry_note.rb
 homepage: https://github.com/banister
 licenses: []
 post_install_message: 
@@ -62,4 +67,6 @@ signing_key:
 specification_version: 3
 summary: Ease refactoring and exploration by attaching notes to methods and classes
   in Pry
-test_files: []
+test_files:
+- test/helper.rb
+- test/test_pry_note.rb