diff options
author | eileencodes <eileencodes@gmail.com> | 2022-10-19 13:50:30 -0400 |
---|---|---|
committer | git <svn-admin@ruby-lang.org> | 2022-10-31 04:58:38 +0000 |
commit | 350d0aa023223f560d812ef61396a2c5359f09a6 (patch) | |
tree | 3b6eec1c6726ca419f8c9c411bbab43f30b3169f | |
parent | 3391c51effcd61f9a718adf59740661d99f57b5b (diff) | |
download | ruby-350d0aa023223f560d812ef61396a2c5359f09a6.tar.gz |
[ruby/error_highlight] Support nodes in `spot`
Fixes a bug where `spot` was using the wrong local variable.
We want to use error highlight with code that has been eval'd,
specifically ERB templates. We can recover the compiled source code of
the ERB template but we need an API to pass the node into error
highlight's `spot`.
Required Ruby PR: https://github.com/ruby/ruby/pull/6593
https://github.com/ruby/error_highlight/commit/0b1b650a59
Co-authored-by: Aaron Patterson <tenderlove@ruby-lang.org>
-rw-r--r-- | lib/error_highlight/base.rb | 3 | ||||
-rw-r--r-- | test/error_highlight/test_error_highlight.rb | 32 |
2 files changed, 33 insertions, 2 deletions
diff --git a/lib/error_highlight/base.rb b/lib/error_highlight/base.rb index 4c115cc828..062871ee16 100644 --- a/lib/error_highlight/base.rb +++ b/lib/error_highlight/base.rb @@ -59,8 +59,7 @@ module ErrorHighlight Spotter.new(node, **opts).spot when RubyVM::AbstractSyntaxTree::Node - # Just for compatibility - Spotter.new(node, **opts).spot + Spotter.new(obj, **opts).spot else raise TypeError, "Exception is expected" diff --git a/test/error_highlight/test_error_highlight.rb b/test/error_highlight/test_error_highlight.rb index c4a998092b..d5f57a88bc 100644 --- a/test/error_highlight/test_error_highlight.rb +++ b/test/error_highlight/test_error_highlight.rb @@ -1257,4 +1257,36 @@ undefined method `foo' for nil:NilClass assert_equal(22, spot[:last_column]) assert_equal(" raise_name_error\n", spot[:snippet]) end + + def test_spot_with_node + omit unless RubyVM::AbstractSyntaxTree.respond_to?(:node_id_for_backtrace_location) + + begin + raise_name_error + rescue NameError => exc + end + + bl = exc.backtrace_locations.first + expected_spot = ErrorHighlight.spot(exc, backtrace_location: bl) + ast = RubyVM::AbstractSyntaxTree.parse_file(__FILE__, keep_script_lines: true) + node_id = RubyVM::AbstractSyntaxTree.node_id_for_backtrace_location(bl) + node = find_node_by_id(ast, node_id) + actual_spot = ErrorHighlight.spot(node) + + assert_equal expected_spot, actual_spot + end + + private + + def find_node_by_id(node, node_id) + return node if node.node_id == node_id + + node.children.each do |child| + next unless child.is_a?(RubyVM::AbstractSyntaxTree::Node) + found = find_node_by_id(child, node_id) + return found if found + end + + return false + end end |