Update component spec

zetachang · zetachang · commit d5f3e159d145 · 2015-02-22T20:27:44.000+08:00
diff --git a/spec/component_spec.rb b/spec/component_spec.rb
@@ -87,18 +87,17 @@ def render
       renderToDocument(Foo)
     end
 
-    it "should allow block for life cycle helpers" do
-      proc_a = Proc.new {}
-      proc_b = Proc.new {}
+    it "should allow block for life cycle callback" do
       Foo.class_eval do
-        before_mount(&proc_a)
-        after_mount(&proc_b)
+        define_state(:foo)
+        
+        before_mount do
+          self.foo = "bar"
+        end
       end
 
-      expect(proc_a).to receive(:call)
-      expect(proc_b).to receive(:call)
-
-      renderToDocument(Foo)
+      element = renderToDocument(Foo)
+      expect(element.state.foo).to be("bar")
     end
   end
 
