Fix end defs comments

Fixes #295
fables-tales · May 31, 2021 · f26e3b5 · f26e3b5
1 parent 3cd7b85
commit f26e3b5
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 1 deletion.
diff --git a/fixtures/small/defs_end_comment_actual.rb b/fixtures/small/defs_end_comment_actual.rb
@@ -0,0 +1,8 @@
+class A
+  def self.m
+    operation_name
+    # @@formatted_op_names.compute_if_absent(operation_name) do
+    #   operation_name.to_s.split('_').collect(&:capitalize).join
+    # end
+  end
+end
diff --git a/fixtures/small/defs_end_comment_expected.rb b/fixtures/small/defs_end_comment_expected.rb
@@ -0,0 +1,8 @@
+class A
+  def self.m
+    operation_name
+    # @@formatted_op_names.compute_if_absent(operation_name) do
+    #   operation_name.to_s.split('_').collect(&:capitalize).join
+    # end
+  end
+end
diff --git a/librubyfmt/src/format.rs b/librubyfmt/src/format.rs
@@ -1991,7 +1991,7 @@ pub fn format_defs(ps: &mut dyn ConcreteParserState, defs: Defs) {
         }),
     );
 
-    ps.wind_line_forward();
+    ps.wind_dumping_comments();
     ps.emit_end();
 
     if ps.at_start_of_line() {