Memoize can be a on its own line (#52)

Author: Antsiscool

lib/memery.rb

@@ -38,6 +38,10 @@ def included(base = nil, &block)
 
   module ClassMethods
     def memoize(*method_names, condition: nil, ttl: nil)
+      if method_names.empty?
+        @_memery_memoize_next_method = { condition: condition, ttl: ttl }
+        return
+      end
       prepend_memery_module!
       method_names.each do |method_name|
         define_memoized_method!(method_name, condition: condition, ttl: ttl)
@@ -52,6 +56,15 @@ def memoized?(method_name)
       @_memery_module.private_method_defined?(method_name)
     end
 
+    def method_added(name)
+      super
+
+      return unless @_memery_memoize_next_method
+
+      memoize(name, **@_memery_memoize_next_method)
+      @_memery_memoize_next_method = nil
+    end
+
     private
 
     def prepend_memery_module!

spec/memery_spec.rb

@@ -14,6 +14,12 @@ class A
     m_private
   end
 
+  memoize
+  def m_different_line
+    CALLS << :m_different_line
+    :m_different_line
+  end
+
   def not_memoized; end
 
   memoize def m_nil
@@ -80,6 +86,13 @@ module M
     :m
   end
 
+  memoize
+
+  def m_different_line
+    CALLS << :m_different_line
+    :m_different_line
+  end
+
   def not_memoized; end
 
   private
@@ -170,6 +183,14 @@ class H
     end
   end
 
+  context "methods without args memoize on new line" do
+    specify do
+      values = [ a.m_different_line, a.m_nil, a.m_different_line, a.m_nil ]
+      expect(values).to eq([:m_different_line, nil, :m_different_line, nil])
+      expect(CALLS).to eq([:m_different_line, nil])
+    end
+  end
+
   context "flushing cache" do
     specify do
       values = [ a.m, a.m ]
@@ -294,6 +315,12 @@ class H
       expect(CALLS).to eq([:m])
     end
 
+    specify do
+      values = [c.m_different_line, c.m_different_line, c.m_different_line]
+      expect(values).to eq([:m_different_line, :m_different_line, :m_different_line])
+      expect(CALLS).to eq([:m_different_line])
+    end
+
     context "memoization in class" do
       specify do
         values = [c.m_class, c.m_class, c.m_class]
@@ -480,6 +507,12 @@ class H
         it { is_expected.to be true }
       end
 
+      context "memoize is on a different line" do
+        let(:method_name) { :m_different_line }
+
+        it { is_expected.to be true }
+      end
+
       context "private memoized method" do
         let(:method_name) { :m_private }