Sha256: 119a774cf66fe6264f0648b0621d8824f8cd043616a17afd11dfe940459e6697
Contents?: true
Size: 1.99 KB
Versions: 23
Compression:
Stored size: 1.99 KB
Contents
require 'rails_helper'
require 'sorbet-rails/model_rbi_formatter'
RSpec.describe SorbetRails::ModelRbiFormatter do
before(:all) do
SorbetRails::Utils.rails_eager_load_all!
end
it 'does not throw an error when given an abstract class' do
formatter = SorbetRails::ModelRbiFormatter.new(Potion, Set.new(['Potion', 'Wizard']))
expect_match_file(
formatter.generate_rbi,
'expected_potion.rbi',
)
end
it 'generates correct rbi file for Wizard' do
class_set = Set.new(['Wizard', 'Wand', 'SpellBook', 'School', 'Subject'])
if Object.const_defined?('ActiveStorage::Attachment')
class_set << 'ActiveStorage::Attachment'
end
if Object.const_defined?('ActiveStorage::Blob')
class_set << 'ActiveStorage::Blob'
end
formatter = SorbetRails::ModelRbiFormatter.new(Wizard, class_set)
expect_match_file(
formatter.generate_rbi,
'expected_wizard.rbi',
)
end
it 'generates strict belongs_to and generate overridden methods' do
formatter = SorbetRails::ModelRbiFormatter.new(Wand, Set.new(['Wizard', 'Wand', 'SpellBook']))
expect_match_file(
formatter.generate_rbi,
'expected_wand.rbi',
)
end
context 'there is a hidden model' do
it 'fallbacks to use ActiveRecord::Relation' do
class_set = Set.new(['Wizard', 'Wand'])
if Object.const_defined?('ActiveStorage::Attachment')
class_set << 'ActiveStorage::Attachment'
end
if Object.const_defined?('ActiveStorage::Blob')
class_set << 'ActiveStorage::Blob'
end
formatter = SorbetRails::ModelRbiFormatter.new(Wizard, class_set)
expect_match_file(
formatter.generate_rbi,
'expected_wizard_wo_spellbook.rbi',
)
end
end
it 'creates Model::RelationType alias at runtime' do
ActiveRecord::Base.descendants.each do |model|
expect(model.const_defined?("RelationType")).to be(true)
expect(model.const_get("RelationType")).to be_a(T::Private::Types::TypeAlias)
end
end
end
Version data entries
23 entries across 23 versions & 1 rubygems