Fix state getter

zetachang · zetachang · commit 1fbaf50583e5 · 2015-02-22T20:39:56.000+08:00
diff --git a/lib/react/component.rb b/lib/react/component.rb
@@ -38,6 +38,11 @@ def mounted?
       `#{@native}.isMounted()`
     end
 
+    def state
+      raise "No native ReactComponent associated" unless @native
+      Native(`#{@native}.state`)
+    end
+
     def set_state(state, &block)
       raise "No native ReactComponent associated" unless @native
       %x{
@@ -141,7 +146,7 @@ def define_state(*states)
           # getter
           define_method("#{name}") do
             return unless @native
-            `#{@native}.state[#{name}]`
+            self.state[name]
           end
           # setter
           define_method("#{name}=") do |new_state|
diff --git a/spec/component_spec.rb b/spec/component_spec.rb
@@ -193,7 +193,7 @@ def render
       expect(element.getDOMNode.textContent).to eq("10")
     end
 
-    it "should support original `set_state` method" do
+    it "should support original `setState` as `set_state` method" do
       Foo.class_eval do
         before_mount do
           self.set_state(foo: "bar")
@@ -203,6 +203,36 @@ def render
       element = renderToDocument(Foo)
       expect(element.state.foo).to be("bar")
     end
+
+    it "should support originl `state` method" do
+      Foo.class_eval do
+        before_mount do
+          self.set_state(foo: "bar")
+        end
+
+        def render
+          div { self.state[:foo] }
+        end
+      end
+
+      expect(React.render_to_static_markup(React.create_element(Foo))).to eq("<div>bar</div>")
+    end
+
+    it "should transform state getter to Ruby object" do
+      Foo.class_eval do
+        define_state :foo
+
+        before_mount do
+          self.foo = {a: 10}
+        end
+
+        def render
+          div { self.foo[:a] }
+        end
+      end
+
+      expect(React.render_to_static_markup(React.create_element(Foo))).to eq("<div>10</div>")
+    end
   end
 
   describe "Props" do
