module RSpec::Rails::ViewAssigns

def _assigns
  super.merge(_encapsulated_assigns)
end

def _encapsulated_assigns
  @_encapsulated_assigns ||= {}
end

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