# typed: true
# frozen_string_literal: true
require "test_helper"
class ERBDocumentTest < Minitest::Test
def setup
@global_state = RubyLsp::GlobalState.new
end
def test_erb_file_is_properly_parsed
source = +<<~ERB
- <%= foo %>
-
- <%= bar %>
-
- <%== baz %>
-
- <%- quz %>
-
ERB
document = RubyLsp::ERBDocument.new(
source: source,
version: 1,
uri: URI("file:///foo.erb"),
global_state: @global_state,
)
document.parse!
refute_predicate(document, :syntax_error?)
assert_equal(
" \n foo \n bar \n baz \n quz \n \n",
document.parse_result.source.source,
)
end
def test_erb_file_parses_in_eval_context
source = +<<~ERB
<%= yield :head %>
<%= yield %>
ERB
document = RubyLsp::ERBDocument.new(
source: source,
version: 1,
uri: URI("file:///foo.erb"),
global_state: @global_state,
)
document.parse!
refute_predicate(document, :syntax_error?)
assert_equal(
" \n \n yield :head \n \n \n yield \n \n \n",
document.parse_result.source.source,
)
end
def test_erb_document_handles_windows_newlines
document = RubyLsp::ERBDocument.new(
source: "<%=\r\nbar %>",
version: 1,
uri: URI("file:///foo.erb"),
global_state: @global_state,
)
document.parse!
refute_predicate(document, :syntax_error?)
assert_equal(" \r\nbar ", document.parse_result.source.source)
end
def test_erb_syntax_error_does_not_cause_crash
[
"<%=",
"<%",
"<%-",
"<%#",
"<%= foo %>\n<%= bar",
"<%= foo %\n<%= bar %>",
].each do |source|
document = RubyLsp::ERBDocument.new(
source: source,
version: 1,
uri: URI("file:///foo.erb"),
global_state: @global_state,
)
document.parse!
end
end
def test_failing_to_parse_indicates_syntax_error
source = +<<~ERB
ERB
document = RubyLsp::ERBDocument.new(
source: source,
version: 1,
uri: URI("file:///foo.erb"),
global_state: @global_state,
)
assert_predicate(document, :syntax_error?)
end
def test_locate
source = <<~ERB
<% Post.all.each do |post| %>
<%= post.title %>
<% end %>
ERB
document = RubyLsp::ERBDocument.new(
source: source,
version: 1,
uri: URI("file:///foo/bar.erb"),
global_state: @global_state,
)
# Locate the `Post` class
node_context = document.locate_node({ line: 0, character: 3 })
assert_instance_of(Prism::ConstantReadNode, node_context.node)
assert_equal(
"Post",
node_context.node #: as Prism::ConstantReadNode
.location.slice,
)
# Locate the `each` call from block
node_context = document.locate_node({ line: 0, character: 17 })
assert_instance_of(Prism::BlockNode, node_context.node)
assert_equal(
:each,
node_context.call_node #: as !nil
.name,
)
# Locate the `title` invocation
node_context = document.locate_node({ line: 1, character: 15 })
assert_equal(
"title",
node_context.node #: as Prism::CallNode
.message,
)
end
def test_cache_set_and_get
document = RubyLsp::ERBDocument.new(
source: +"",
version: 1,
uri: URI("file:///foo/bar.erb"),
global_state: @global_state,
)
value = [1, 2, 3]
assert_equal(value, document.cache_set("textDocument/semanticHighlighting", value))
assert_equal(value, document.cache_get("textDocument/semanticHighlighting"))
end
def test_keeps_track_of_virtual_host_language_source
source = +<<~ERB
ERB
document = RubyLsp::ERBDocument.new(
source: source,
version: 1,
uri: URI("file:///foo.erb"),
global_state: @global_state,
)
assert_equal(<<~HTML, document.host_language_source)
HTML
end
def test_erb_is_parsed_as_a_partial_script
source = +<<~ERB
ERB
document = RubyLsp::ERBDocument.new(
source: source,
version: 1,
uri: URI("file:///foo.erb"),
global_state: @global_state,
)
document.parse!
refute_predicate(document, :syntax_error?)
end
end