Added SourceNotFoundError class

2 files changed, 10 insertions, 8 deletions

diff --git a/lib/method_source.rb b/lib/method_source.rb
index aa2b34a..0353144 100644
--- a/lib/method_source.rb
+++ b/lib/method_source.rb
@@ -7,6 +7,9 @@ require "#{direc}/method_source/version"
 require "#{direc}/method_source/source_location"
 
 module MethodSource
+
+  class SourceNotFoundError < StandardError; end
+
   # Determine if a string of code is a valid Ruby expression.
   # @param [String] code The code to validate.
   # @return [Boolean] Whether or not the code is a valid Ruby expression.
@@ -51,8 +54,7 @@ module MethodSource
 
     code
   rescue Errno::ENOENT
-    # source_location[0] of evaled methods would return (eval) if __FILE__ and __LINE__ is not given. File.readlines "(eval)" would raise ENOENT
-    nil
+    raise SourceNotFoundError, "Cannot get source code located at file: #{source_location[0]}"
   end
 
   # @param [Array] source_location The array containing file_name [String], line [Fixnum]
@@ -143,9 +145,9 @@ module MethodSource
       if respond_to?(:source_location)
         source = MethodSource.source_helper(source_location)
 
-        raise "Cannot locate source for this method: #{name}" if !source
+        raise SourceNotFoundError, "Cannot locate source for this method: #{name}" if !source
       else
-        raise "#{self.class}#source not supported by this Ruby version (#{RUBY_VERSION})"
+        raise SourceNotFoundError, "#{self.class}#source not supported by this Ruby version (#{RUBY_VERSION})"
       end
 
       source
@@ -162,9 +164,9 @@ module MethodSource
       if respond_to?(:source_location)
         comment = MethodSource.comment_helper(source_location)
 
-        raise "Cannot locate source for this method: #{name}" if !comment
+        raise SourceNotFoundError, "Cannot locate source for this method: #{name}" if !comment
       else
-        raise "#{self.class}#comment not supported by this Ruby version (#{RUBY_VERSION})"
+        raise SourceNotFoundError, "#{self.class}#comment not supported by this Ruby version (#{RUBY_VERSION})"
       end
 
       comment
diff --git a/test/test.rb b/test/test.rb
index 3fd1597..4743a50 100644
--- a/test/test.rb
+++ b/test/test.rb
@@ -77,12 +77,12 @@ describe MethodSource do
     end
 
     it "should raise error for evaled methods that do not pass __FILE__ and __LINE__ + 1 as its arguments" do
-      lambda { M.instance_method(:name_three).source }.should.raise RuntimeError
+      lambda { M.instance_method(:name_three).source }.should.raise MethodSource::SourceNotFoundError
     end
 
     if !is_rbx?
       it 'should raise for C methods' do
-        lambda { method(:puts).source }.should.raise RuntimeError
+        lambda { method(:puts).source }.should.raise MethodSource::SourceNotFoundError
       end
     end
   end