diff options
Diffstat (limited to 'lib/method_source.rb')
| -rw-r--r-- | lib/method_source.rb | 14 |
1 files changed, 8 insertions, 6 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 |