Anjing Webshells

Website : rimsha.abasa.com
: / var / canvas / spec / factories /
Filename : pseudonym_session_factory.rb
back
# frozen_string_literal: true

#
# Copyright (C) 2015 - present Instructure, Inc.
#
# This file is part of Canvas.
#
# Canvas is free software: you can redistribute it and/or modify it under
# the terms of the GNU Affero General Public License as published by the Free
# Software Foundation, version 3 of the License.
#
# Canvas is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
# A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
# details.
#
# You should have received a copy of the GNU Affero General Public License along
# with this program. If not, see <http://www.gnu.org/licenses/>.

module Factories
  def user_session(user, pseudonym = nil)
    if caller.grep(%r{onceler/recorder.*record!}).present?
      raise "don't stub sessions in a `before(:once)` block; do it in a `before(:each)` so the stubbing works for all examples and not just the first one"
    end

    unless pseudonym
      pseudonym = double(
        "Pseudonym",
        record: user,
        user_id: user.id,
        user:,
        login_count: 1,
        global_account_id: "10000000000001",
        sis_user_id: "U001",
        shard: Shard.default,
        works_for_account?: true,
        suspended?: false
      )
      # at least one thing cares about the id of the pseudonym... using the
      # object_id should make it unique (but obviously things will fail if
      # it tries to load it from the db.)
      allow(pseudonym).to receive_messages(id: pseudonym.object_id,
                                           unique_id: "unique_id",
                                           global_id: 10_000_000_000_001)
    end

    session = double("PseudonymSession",
                     record: pseudonym)
    allow(session).to receive(:[]).with(anything).and_return(nil)

    @session_stubbed = true
    allow(PseudonymSession).to receive(:find).and_wrap_original do |original|
      next original.call unless @session_stubbed

      session
    end
    allow(PseudonymSession).to receive(:find_with_validation).and_wrap_original do |original|
      next original.call unless @session_stubbed

      session
    end
    session
  end

  def remove_user_session
    @session_stubbed = false if @session_stubbed
  end
end