Set the view_context properly

lib/draper/view_context.rb

@@ -1,7 +1,7 @@
 module Draper
   module ViewContext
     def set_current_view_context
-      Thread.current[:current_view_context] ||= self.view_context
+      Thread.current[:current_view_context] = self.view_context
     end
     
     def self.included(source)

spec/view_context_spec.rb

@@ -22,4 +22,14 @@ describe Draper::ViewContext do
     Thread.current[:current_view_context] = nil
     expect {app_controller.current_view_context}.should raise_exception(Exception)
   end
+  
+  it "sets view_context every time" do
+    app_controller_instance.stub(:view_context) { 'first' }
+    app_controller_instance.set_current_view_context
+    Thread.current[:current_view_context].should == 'first'
+    
+    app_controller_instance.stub(:view_context) { 'second' }
+    app_controller_instance.set_current_view_context
+    Thread.current[:current_view_context].should == 'second'
+  end
 end