eBooru/test/unit/upload_whitelist_test.rb

# frozen_string_literal: true

require "test_helper"

class UploadWhitelistTest < ActiveSupport::TestCase
  context "A upload whitelist" do
    setup do
      user = create(:privileged_user)
      CurrentUser.user = user

      @whitelist = create(:upload_whitelist, pattern: "*.e621.net/data/*", note: "e621")
    end

    should "match" do
      assert_equal([true, nil], UploadWhitelist.is_whitelisted?(Addressable::URI.parse("https://static1.e621.net/data/123.png")))
      assert_equal([false, "123.com not in whitelist"], UploadWhitelist.is_whitelisted?(Addressable::URI.parse("https://123.com/what.png")))
    end

    should "bypass for admins" do
      CurrentUser.user.level = 50
      Danbooru.config.stubs(:bypass_upload_whitelist?).returns(true)
      assert_equal([true, "bypassed"], UploadWhitelist.is_whitelisted?(Addressable::URI.parse("https://123.com/what.png")))
    end
  end
end