spec/memoit_spec.rb in memoit-0.1.2 vs spec/memoit_spec.rb in memoit-0.2.0

@@ -1,10 +1,14 @@
 require "memoit"
 
 describe Memoit do
   let(:klass) do
     Class.new do
+      memoize_class_method def self.foo
+        rand
+      end
+
       memoize def foo
         rand
       end
 
       memoize def bar(*values)
@@ -38,8 +42,23 @@
     end
 
     it "caches falsy values" do
       expect(instance).to receive(:foo).once
       expect(instance.falsy).to eq(instance.falsy)
+    end
+
+    it "returns the name of the method" do
+      name = nil
+      Class.new do
+        name = memoize def blah
+        end
+      end
+      expect(name).to eq(:blah)
+    end
+  end
+
+  describe ".memoize_class_method" do
+    it "caches result" do
+      expect(klass.foo).to eq(klass.foo)
     end
   end
 end