module RSpec
  module Rails
    module ViewAssigns
      # Assigns a value to an instance variable in the scope of the
      # view being rendered.
      #
      # @example
      #
      #     assign(:widget, stub_model(Widget))
      def assign(key, value)
        _encapsulated_assigns[key] = value
      end

      def view_assigns
        begin
          # TODO: _assigns was deprecated in favor of view_assigns after
          # Rails-3.0.0 was released. Since we are not able to predict when
          # the _assigns/view_assigns patch will be released (I thought it
          # would have been in 3.0.1, but 3.0.1 bypassed this change for a
          # security fix), this bit ensures that we do the right thing without
          # knowing anything about the Rails version we are dealing with.
          #
          # Once that change _is_ released, this can be changed to something
          # that checks for the Rails version when the module is being
          # interpreted, as it was before commit dd0095.
          super.merge(_encapsulated_assigns)
        rescue
          _assigns
        end
      end

      # @api private
      def _assigns
        super.merge(_encapsulated_assigns)
      end

      private

      def _encapsulated_assigns
        @_encapsulated_assigns ||= {}
      end

    end
  end
end