diff --git a/test/spoom/deadcode/remover_test.rb b/test/spoom/deadcode/remover_test.rb index 4e38b71b..59f4750e 100644 --- a/test/spoom/deadcode/remover_test.rb +++ b/test/spoom/deadcode/remover_test.rb @@ -965,6 +965,29 @@ def baz; end RB end + def test_removes_node_rbs_comment + res = remove(<<~RB, "bar") + class Foo + def foo; end + + #: -> void + def bar + something + end + + def baz; end + end + RB + + assert_equal(<<~RB, res) + class Foo + def foo; end + + def baz; end + end + RB + end + def test_removes_singleton_class_if_needed res = remove(<<~RB, "foo") class Foo