2010-12-15 22:34:19 +00:00
|
|
|
module Shoulda # :nodoc:
|
|
|
|
module Matchers
|
|
|
|
module ActiveRecord # :nodoc:
|
|
|
|
|
|
|
|
# Ensures that there are DB indices on the given columns or tuples of
|
|
|
|
# columns.
|
|
|
|
#
|
|
|
|
# Options:
|
|
|
|
# * <tt>unique</tt> - whether or not the index has a unique
|
|
|
|
# constraint. Use <tt>true</tt> to explicitly test for a unique
|
|
|
|
# constraint. Use <tt>false</tt> to explicitly test for a non-unique
|
|
|
|
# constraint. Use <tt>nil</tt> if you don't care whether the index is
|
|
|
|
# unique or not. Default = <tt>nil</tt>
|
|
|
|
#
|
|
|
|
# Examples:
|
|
|
|
#
|
|
|
|
# it { should have_db_index(:age) }
|
|
|
|
# it { should have_db_index([:commentable_type, :commentable_id]) }
|
|
|
|
# it { should have_db_index(:ssn).unique(true) }
|
|
|
|
#
|
|
|
|
def have_db_index(columns)
|
2012-03-30 15:36:04 +00:00
|
|
|
HaveDbIndexMatcher.new(columns)
|
2010-12-15 22:34:19 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
class HaveDbIndexMatcher # :nodoc:
|
2012-03-30 15:36:04 +00:00
|
|
|
def initialize(columns)
|
2010-12-15 22:34:19 +00:00
|
|
|
@columns = normalize_columns_to_array(columns)
|
|
|
|
end
|
|
|
|
|
|
|
|
def unique(unique)
|
|
|
|
@unique = unique
|
|
|
|
self
|
|
|
|
end
|
|
|
|
|
|
|
|
def matches?(subject)
|
|
|
|
@subject = subject
|
|
|
|
index_exists? && correct_unique?
|
|
|
|
end
|
|
|
|
|
|
|
|
def failure_message
|
|
|
|
"Expected #{expectation} (#{@missing})"
|
|
|
|
end
|
|
|
|
|
|
|
|
def negative_failure_message
|
|
|
|
"Did not expect #{expectation}"
|
|
|
|
end
|
|
|
|
|
|
|
|
def description
|
|
|
|
"have a #{index_type} index on columns #{@columns.join(' and ')}"
|
|
|
|
end
|
|
|
|
|
|
|
|
protected
|
|
|
|
|
|
|
|
def index_exists?
|
|
|
|
! matched_index.nil?
|
|
|
|
end
|
|
|
|
|
|
|
|
def correct_unique?
|
|
|
|
return true if @unique.nil?
|
2011-09-18 14:40:47 +00:00
|
|
|
if !!matched_index.unique == @unique
|
2010-12-15 22:34:19 +00:00
|
|
|
true
|
|
|
|
else
|
|
|
|
@missing = "#{table_name} has an index named #{matched_index.name} " <<
|
2011-09-18 14:40:47 +00:00
|
|
|
"of unique #{!!matched_index.unique}, not #{@unique}."
|
2010-12-15 22:34:19 +00:00
|
|
|
false
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def matched_index
|
|
|
|
indexes.detect { |each| each.columns == @columns }
|
|
|
|
end
|
|
|
|
|
|
|
|
def model_class
|
|
|
|
@subject.class
|
|
|
|
end
|
|
|
|
|
|
|
|
def table_name
|
|
|
|
model_class.table_name
|
|
|
|
end
|
|
|
|
|
|
|
|
def indexes
|
|
|
|
::ActiveRecord::Base.connection.indexes(table_name)
|
|
|
|
end
|
|
|
|
|
|
|
|
def expectation
|
2012-03-30 15:22:31 +00:00
|
|
|
"#{model_class.name} to #{description}"
|
2010-12-15 22:34:19 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def index_type
|
|
|
|
case @unique
|
|
|
|
when nil
|
|
|
|
''
|
|
|
|
when false
|
|
|
|
'non-unique'
|
|
|
|
else
|
|
|
|
'unique'
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def normalize_columns_to_array(columns)
|
2012-04-10 01:03:19 +00:00
|
|
|
Array.wrap(columns).map(&:to_s)
|
2010-12-15 22:34:19 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|