handle modules with a single method

rubys · rubys · commit e0b0f6a9647c · 2021-01-18T18:29:50.000-05:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -3,6 +3,7 @@
 * Support static method calls with blocks in es2015+
 * Auto-bind instance methods referenced as properties within a class
 * handle begin, if, and case as expressions
+* handle modules with exactly one method
 
 # 3.6.1 / 2020-12-31
 
diff --git a/lib/ruby2js/converter/class.rb b/lib/ruby2js/converter/class.rb
@@ -219,8 +219,9 @@ class Converter
 
         # collapse sequence to a single assignment
         if \
-          @ast.type != :class_extend and
-          (methods > 1 or (methods == 1 and body[start].type == :prop))
+          @ast.type == :class_module or
+          (@ast.type == :class and 
+            (methods > 1 or (methods == 1 and body[start].type == :prop)))
         then
           pairs = body[start...start+methods].map do |node|
             if node.type == :method
diff --git a/spec/transliteration_spec.rb b/spec/transliteration_spec.rb
@@ -824,6 +824,8 @@ def to_js( string, opts={} )
         must_equal 'A = function() {var B = 1; return {B: B}}()'
       to_js( 'module A; def b; return 1; end; end' ).
         must_equal 'var A = {get b() {return 1}}'
+      to_js( 'module A; def b(); return 1; end; end' ).
+        must_equal 'var A = {b: function() {return 1}}'
       to_js( 'module A; class B; def initialize; @c=1; end; end; end' ).
         must_equal 'A = function() {function B() {this._c = 1}; return {B: B}}()'
     end
