Memoize can be a on it's own line

This allows defining methods like:

memoize
def foo
  "string"
end

This would previous not memoize the method.

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
@@ -480,6 +493,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 }
 
@@ -516,5 +535,7 @@ class H
 
       include_examples "works correctly"
     end
+
+
   end
 end