spec/mem_spec.rb in mem-0.0.2 vs spec/mem_spec.rb in mem-0.1.0

@@ -1,29 +1,58 @@
 require "spec_helper"
 
 describe Mem do
   describe "#memoize" do
-    let(:klass) do
-      Class.new do
-        extend Mem
+    let(:object) do
+      klass.new
+    end
 
-        def foo(&block)
-          bar(&block)
-        end
-        memoize :foo
+    context "with normal method name" do
+      let(:klass) do
+        Class.new do
+          include Mem
 
-        def bar
-          yield
+          def a(value, &block)
+            b(value, &block)
+          end
+          memoize :a
+
+          def b(value)
+            [value, yield]
+          end
         end
       end
-    end
 
-    let(:object) do
-      klass.new
+      it "memoizes the method call" do
+        expect(object).to receive(:b).once.and_call_original
+        expect(object.a(1) { 2 }).to eq [1, 2]
+        expect(object.a(3) { 4 }).to eq [1, 2]
+      end
     end
 
-    it "memoizes the result of specified method call" do
-      expect(object).to receive(:bar).once.and_call_original
-      expect(object.foo { "foo" }).to eq "foo"
+    context "with normal method name" do
+      let(:klass) do
+        Class.new do
+          include Mem
+
+          def a!(value, &block)
+            b(value, &block)
+          end
+          memoize :a!
+
+          def b(value)
+            [value, yield]
+          end
+        end
+      end
+
+      it "memoizes the method call" do
+        expect(object).to receive(:b).once.and_call_original
+        expect(object.a!(1) { 2 }).to eq [1, 2]
+        expect(object.a!(3) { 4 }).to eq [1, 2]
+        object.should have_memoized(:a!)
+        object.memoized(:a!).should == [1, 2]
+        object.memoized_table.should == { a!: [1, 2] }
+      end
     end
   end
 end