Merge pull request #260 from cassar/edge

Fix for issue #258

Author: catmando

lib/react/component/api.rb

@@ -11,6 +11,7 @@ def mounted?
 
       def force_update!
         `#{self}.native.forceUpdate()`
+        self
       end
 
       def set_props(prop, &block)

spec/react/component_spec.rb

@@ -116,7 +116,7 @@ def render
           def self.get_error
             @@error
           end
-          
+
           def self.get_info
             @@info
           end
@@ -136,6 +136,48 @@ def render
     end
   end
 
+  describe 'Misc Methods' do
+    it 'has a force_update! method' do
+      mount 'Foo' do
+        class Foo < Hyperloop::Component
+          class << self
+            attr_accessor :render_counter
+            attr_accessor :instance
+          end
+          before_mount do
+            Foo.render_counter = 0
+            Foo.instance = self
+          end
+          def render
+            Foo.render_counter += 1
+            DIV { "I have been rendered #{Foo.render_counter} times" }
+          end
+        end
+      end
+      expect_evaluate_ruby do
+        Foo.instance.force_update!
+        Foo.render_counter
+      end.to eq(2)
+    end
+
+    it 'has its force_update! method return itself' do
+      mount 'Foo' do
+        class Foo < Hyperloop::Component
+          class << self
+            attr_accessor :instance
+          end
+          before_mount do
+            Foo.instance = self
+          end
+          def render
+            DIV { "I have been rendered" }
+          end
+        end
+      end
+      expect_evaluate_ruby('Foo.instance == Foo.instance.force_update!').to be_truthy
+    end
+  end
+
   describe 'New style setter & getter' do
     before(:each) do
       on_client do
@@ -843,7 +885,6 @@ def render
 
     it 'only overrides `p` in render context' do
       mount 'Foo' do
-        
         class Foo
           include React::Component