Fix handling of inline comments in export/public expressions

fredrikekre · Jul 26, 2024 · 49e6c9d · 49e6c9d
1 parent c19f5b5
commit 49e6c9d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/runestone.jl b/src/runestone.jl
@@ -986,7 +986,7 @@ function spaces_in_export_public(ctx::Context, node::Node)
             elseif kind(kid) in KSet"Comment NewlineWs"
                 any_changes && push!(kids′, kid)
                 accept_node!(ctx, kid)
-                state = :expect_identifier
+                state = :expect_space
             else
                 @assert false
             end

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -850,6 +850,8 @@ end
         @test format_string("$(verb) $(a)$(sp),\n$(b)") == "$(verb) $(a),\n    $(b)"
         @test format_string("$(verb) \n$(a)$(sp),\n$(b)") == "$(verb)\n    $(a),\n    $(b)"
         @test format_string("$(verb) $(a)$(sp),\n# b\n$(b)") == "$(verb) $(a),\n    # b\n    $(b)"
+        # Inline comments
+        @test format_string("export a$(sp),$(sp)#= b, =#$(sp)c") == "export a, #= b, =# c"
     end
     # Interpolated identifiers (currently only expected in K"quote" and K"macrocall")
     @test format_string(":(export \$a)") == ":(export \$a)"