view_assigns.rb

module RSpec
  module Rails
    module ViewAssigns
      extend ActiveSupport::Concern

      module InstanceMethods
        # :call-seq:
        #   assign(:widget, stub_model(Widget))
        #
        # Assigns a value to an instance variable in the scope of the
        # view being rendered.
        def assign(key, value)
          _encapsulated_assigns[key] = value
        end

        if ::Rails::VERSION::STRING == "3.0.0"
          def _assigns
            super.merge(_encapsulated_assigns)
          end
          def view_assigns
            _assigns
          end
        else # >= 3.0.1
          def view_assigns
            super.merge(_encapsulated_assigns)
          end
        end

      private

        def _encapsulated_assigns
          @_encapsulated_assigns ||= {}
        end
      end
    end
  end
end