Use memoist_reload flag for all methods

Author: ashu210890

lib/memoist.rb

@@ -63,7 +63,7 @@ def self.extract_reload!(method, args)
     # We don't want to extract out the last argument in this case.
     # Also, if the :memoist_reload flag is passed, we use that
     # irrespective of the arity.
-    if method.arity.negative? && args.length > 0 && args.last.is_a?(Hash)
+    if args.length > 0 && args.last.is_a?(Hash)
       named_args_hash = args.last
       reload = named_args_hash.delete :memoist_reload
       # If no other named args were passed, that means memoist_reload is an
@@ -74,8 +74,10 @@ def self.extract_reload!(method, args)
       end
       return reload
     end
+    return false if method.arity.negative?
 
-    if args.length == method.arity.abs + 1 && (args.last == true || args.last == :reload)
+    # Legacy purposes, works only for methods with no optional/named args
+    if args.length == method.arity + 1 && (args.last == true || args.last == :reload)
       reload = args.pop
     end
     reload

test/memoist_test.rb

@@ -67,9 +67,9 @@ def age
 
     memoize :name, :age
 
-    def sleep(initial_delay, hours = 8, mins = 5)
+    def sleep(hours, mins = 5, seconds: 0)
       @counter.call(:sleep)
-      hours
+      hours * 3600 + mins * 60 + seconds
     end
     memoize :sleep
 
@@ -241,19 +241,25 @@ def test_memoization
 
     3.times { assert_equal 'Josh', @person.name }
     assert_equal 1, @person.name_calls
+
+    3.times { assert_equal 'Josh', @person.name(memoist_reload: true) }
+    assert_equal 4, @person.name_calls
   end
 
   def test_memoize_with_optional_arguments
-    assert_equal 4, @person.sleep(4, 4)
+    assert_equal 15000, @person.sleep(4, 10)
     assert_equal 1, @person.sleep_calls
 
-    3.times { assert_equal 4, @person.sleep(4, 4) }
+    3.times { assert_equal 15000, @person.sleep(4, 10) }
     assert_equal 1, @person.sleep_calls
 
+    3.times { assert_equal 15000, @person.sleep(4, 10, seconds: 0) }
+    assert_equal 2, @person.sleep_calls
+
     # There has been exactly one sleep call due to memoization
     # Now, with reload, we'll have 3 more calls
-    3.times { assert_equal 10, @person.sleep(4, 10, memoist_reload: true) }
-    assert_equal 4, @person.sleep_calls
+    3.times { assert_equal 15000, @person.sleep(4, 10, memoist_reload: true) }
+    assert_equal 5, @person.sleep_calls
   end
 
   def test_memoize_with_options_hash
@@ -295,6 +301,8 @@ def test_reloadable
     assert_equal 2, @calculator.counter
     assert_equal 3, @calculator.counter(true)
     assert_equal 3, @calculator.counter
+    assert_equal 4, @calculator.counter(memoist_reload: true)
+    assert_equal 4, @calculator.counter
   end
 
   def test_flush_cache