# frozen_string_literal: true require_relative '../../../../lib/declare_schema/model/index_definition' # Beware: It looks out that Rails' sqlite3 driver has a bug in retrieving indexes. # In sqlite3/schema_statements, it skips over any index that starts with sqlite_: # next if row["name"].starts_with?("sqlite_") # but this will skip over any indexes created to support "unique" column constraints. # This gem provides an explicit name for all indexes so it shouldn't be affected by the bug... # unless you manually create any Sqlite tables with UNIQUE constraints. RSpec.describe DeclareSchema::Model::IndexDefinition do before do load File.expand_path('../prepare_testapp.rb', __dir__) class IndexDefinitionTestModel < ActiveRecord::Base fields do name :string, limit: 127, index: true timestamps end end class IndexDefinitionCompoundIndexModel < ActiveRecord::Base fields do fk1_id :integer fk2_id :integer timestamps end end end let(:model_class) { IndexDefinitionTestModel } describe 'instance methods' do let(:model) { model_class.new } subject { declared_class.new(model_class) } it 'has index_definitions' do expect(model_class.index_definitions).to be_kind_of(Array) expect(model_class.index_definitions.map(&:name)).to eq(['on_name']) expect([:name, :fields, :unique].map { |attr| model_class.index_definitions[0].send(attr)}).to eq( ['on_name', ['name'], false] ) end it 'has index_definitions_with_primary_key' do expect(model_class.index_definitions_with_primary_key).to be_kind_of(Array) result = model_class.index_definitions_with_primary_key.sort_by(&:name) expect(result.map(&:name)).to eq(['PRIMARY', 'on_name']) expect([:name, :fields, :unique].map { |attr| result[0].send(attr)}).to eq( ['PRIMARY', ['id'], true] ) expect([:name, :fields, :unique].map { |attr| result[1].send(attr)}).to eq( ['on_name', ['name'], false] ) end end describe 'class methods' do describe 'index_name' do it 'works with a single column' do expect(described_class.index_name('parent_id')).to eq('on_parent_id') end it 'works with many columns' do expect(described_class.index_name(['a', 'b', 'c'])).to eq('on_a_and_b_and_c') end end context 'with a migrated database' do before do ActiveRecord::Base.connection.execute <<~EOS CREATE TABLE index_definition_test_models ( id INTEGER NOT NULL PRIMARY KEY, name #{if defined?(Sqlite3) then 'TEXT' else 'VARCHAR(255)' end} NOT NULL ) EOS ActiveRecord::Base.connection.execute <<~EOS CREATE UNIQUE INDEX index_definition_test_models_on_name ON index_definition_test_models(name) EOS ActiveRecord::Base.connection.execute <<~EOS CREATE TABLE index_definition_compound_index_models ( fk1_id INTEGER NOT NULL, fk2_id INTEGER NOT NULL, PRIMARY KEY (fk1_id, fk2_id) ) EOS ActiveRecord::Base.connection.schema_cache.clear! end describe 'for_model' do subject { described_class.for_model(model_class) } context 'with single-column PK' do it 'returns the indexes for the model' do expect(subject.size).to eq(2), subject.inspect expect([:name, :columns, :unique].map { |attr| subject[0].send(attr) }).to eq( ['index_definition_test_models_on_name', ['name'], true] ) expect([:name, :columns, :unique].map { |attr| subject[1].send(attr) }).to eq( ['PRIMARY', ['id'], true] ) end end context 'with compound-column PK' do let(:model_class) { IndexDefinitionCompoundIndexModel } it 'returns the indexes for the model' do if Rails::VERSION::MAJOR < 5 expect(model_class.connection).to receive(:primary_key).with('index_definition_compound_index_models').and_return(nil) connection_stub = instance_double(ActiveRecord::Base.connection.class, "connection") expect(connection_stub).to receive(:indexes). with('index_definition_compound_index_models'). and_return([DeclareSchema::Model::IndexDefinition.new(model_class, ['fk1_id', 'fk2_id'], name: 'PRIMARY')]) expect(model_class.connection).to receive(:dup).and_return(connection_stub) end expect(subject.size).to eq(1), subject.inspect expect([:name, :columns, :unique].map { |attr| subject[0].send(attr) }).to eq( ['PRIMARY', ['fk1_id', 'fk2_id'], true] ) end end end end end # TODO: fill out remaining tests end