rails--rails/activestorage/test/models/strict_loading_test.rb

# frozen_string_literal: true

require "test_helper"
require "database/setup"

class ActiveStorage::StrictLoadingTest < ActiveSupport::TestCase
  class Admin < User
    has_one_attached :image, strict_loading: true
    has_many_attached :images, strict_loading: true
  end

  setup do
    Admin.create!(name: "David")
  end

  test "has_one_attached raises if strict loading and lazy loading" do
    assert_raises ActiveRecord::StrictLoadingViolationError do
      admins = Admin.all
      admins.as_json(include: :image)
    end
  end

  test "has_many_attached raises if strict loading and lazy loading" do
    assert_raises ActiveRecord::StrictLoadingViolationError do
      admins = Admin.all
      admins.as_json(include: :images)
    end
  end
end