Merge pull request #24 from matthewrudy/return-symbol-from-memoize

Make memoize return a :symbol
matthewrudy · Aug 13, 2014 · d4f5590 · d4f5590
2 parents 2ab0dcc + 4c98b82
commit d4f5590
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/lib/memoist.rb b/lib/memoist.rb
@@ -193,6 +193,8 @@ def #{method_name}(*args)
         end
       end
     end
+    # return a chainable method_name symbol if we can
+    method_names.length == 1 ? method_names.first : method_names
   end
 
   class AlreadyMemoizedError < RuntimeError; end

diff --git a/test/memoist_test.rb b/test/memoist_test.rb
@@ -312,6 +312,15 @@ def test_memoization_with_a_subclass
     assert_equal 1, student.name_calls
   end
 
+  def test_memoization_is_chainable
+    klass = Class.new do
+      def foo; "bar"; end
+    end
+    klass.extend Memoist
+    chainable = klass.memoize :foo
+    assert_equal :foo, chainable
+  end
+
   def test_protected_method_memoization
     person = Person.new