eBooru/test/unit/session_loader_test.rb

# frozen_string_literal: true

require "test_helper"

class SessionLoaderTest < ActiveSupport::TestCase
  context "SessionLoader" do
    setup do
      @request = mock
      @request.stubs(:host).returns("danbooru")
      @request.stubs(:remote_ip).returns("127.0.0.1")
      @request.stubs(:authorization).returns(nil)
      cookie_jar = mock
      cookie_jar.stubs(:encrypted).returns({})
      @request.stubs(:cookie_jar).returns(cookie_jar)
      @request.stubs(:parameters).returns({})
      @request.stubs(:session).returns({})
    end

    context ".safe_mode?" do
      should "return true if the config has safe mode enabled" do
        Danbooru.config.stubs(:safe_mode?).returns(true)
        SessionLoader.new(@request).load

        assert_equal(true, CurrentUser.safe_mode?)
      end

      should "return false if the config has safe mode disabled" do
        Danbooru.config.stubs(:safe_mode?).returns(false)
        SessionLoader.new(@request).load

        assert_equal(false, CurrentUser.safe_mode?)
      end

      should "return true if the user has enabled the safe mode account setting" do
        @user = create(:user, enable_safe_mode: true)
        @request.stubs(:session).returns(user_id: @user.id, ph: @user.password_token)
        SessionLoader.new(@request).load

        assert_equal(true, CurrentUser.safe_mode?)
      end
    end
  end
end