josh_cutler-thinking-sphinx 1.3.17

data/features/thinking_sphinx/db/fixtures/posts.rb

@@ -0,0 +1,6 @@
+post = Post.create(
+  :subject => "Hello World", :content => "Um Text", :id => 1, :category_id => 1
+)
+
+post.authors << Author.find(:first)
+post.save

data/features/thinking_sphinx/db/fixtures/robots.rb

@@ -0,0 +1,14 @@
+# Reset the primary key to allow us to create robots with specific internal_ids
+class Robot < ActiveRecord::Base
+  set_primary_key :alternate_primary_key
+end
+
+Robot.create :name => 'Fritz',      :internal_id => 'F0001'
+Robot.create :name => 'Sizzle',     :internal_id => 'S0001'
+Robot.create :name => 'Sizzle Jr.', :internal_id => 'S0002'
+Robot.create :name => 'Expendable', :internal_id => 'E0001'
+
+# Annnnnnnnnnd we're back
+class Robot < ActiveRecord::Base
+  set_primary_key :internal_id
+end

data/features/thinking_sphinx/db/fixtures/tags.rb

@@ -0,0 +1,27 @@
+post = Post.find(1)
+
+[
+  'Nunc in neque. Integer sed odio ac quam pellentesque suscipit. Cras posuere posuere est. Suspendisse est.',
+  'In hac habitasse platea dictumst. Etiam eleifend est ac diam. Ut ac felis sit amet mi bibendum varius.',
+  'Nullam sollicitudin tellus at ipsum. Duis mi eros, blandit sed, condimentum non, mattis vel, orci.',
+  'Praesent auctor mollis leo. Nulla facilisi. Pellentesque habitant morbi tristique senectus et netus et',
+  'malesuada fames ac turpis egestas. Donec non odio. Maecenas varius elit ut ante. Phasellus egestas, quam',
+  'a congue euismod, diam urna gravida risus, at euismod lectus diam id sem. Vestibulum sed dolor et massa',
+  'porta placerat. Etiam eget risus. Sed ornare. Vivamus in sapien. Maecenas non enim nec metus posuere',
+  'vehicula. Donec rhoncus mauris at metus. Curabitur volutpat massa a metus. Aliquam ornare, neque ut',
+  'tristique convallis, libero orci eleifend nibh, ac porta leo felis sed orci. Lorem ipsum dolor sit amet,',
+  'Waffles' # This one is important, whereas the others are just padding
+].each do |text|
+  Tagging.create :taggable => post, :tag => Tag.create(:text => text)
+end
+
+Developer.find(:all).each do |developer|
+  [:country, :city, :state].each do |column|
+    Tagging.create(
+      :taggable => developer,
+      :tag => Tag.find_or_create_by_text(developer.send(column))
+    )
+  end
+end
+
+Tagging.create(:taggable => Developer.find(:all).last)

data/features/thinking_sphinx/db/migrations/create_alphas.rb

@@ -0,0 +1,8 @@
+ActiveRecord::Base.connection.create_table :alphas, :force => true do |t|
+  t.column :name,       :string,  :null => false
+  t.column :value,      :integer, :null => false
+  t.column :active,     :boolean, :null => false, :default => true
+  t.column :cost,       :decimal, :precision => 10, :scale => 6
+  t.column :created_on, :date
+  t.column :created_at, :timestamp
+end

data/features/thinking_sphinx/db/migrations/create_animals.rb

@@ -0,0 +1,5 @@
+ActiveRecord::Base.connection.create_table :animals, :force => true do |t|
+  t.column :name,   :string,  :null => false
+  t.column :type,   :string
+  t.column :delta,  :boolean, :null => false, :default => false
+end

data/features/thinking_sphinx/db/migrations/create_authors.rb

@@ -0,0 +1,3 @@
+ActiveRecord::Base.connection.create_table :authors, :force => true do |t|
+  t.column :name, :string, :null => false
+end

data/features/thinking_sphinx/db/migrations/create_authors_posts.rb

@@ -0,0 +1,6 @@
+ActiveRecord::Base.connection.create_table(
+  :authors_posts, :force => true, :id => false
+) do |t|
+  t.column :author_id, :integer, :null => false
+  t.column :post_id,   :integer, :null => false
+end

data/features/thinking_sphinx/db/migrations/create_betas.rb

@@ -0,0 +1,5 @@
+ActiveRecord::Base.connection.create_table :betas, :force => true do |t|
+  t.column :name, :string,  :null => false
+  t.column :value, :integer, :null => false
+  t.column :delta, :boolean, :null => false, :default => false
+end

data/features/thinking_sphinx/db/migrations/create_boxes.rb

@@ -0,0 +1,5 @@
+ActiveRecord::Base.connection.create_table :boxes, :force => true do |t|
+  t.column :width,  :integer, :null => false
+  t.column :length, :integer, :null => false
+  t.column :depth,  :integer
+end

data/features/thinking_sphinx/db/migrations/create_categories.rb

@@ -0,0 +1,3 @@
+ActiveRecord::Base.connection.create_table :categories, :force => true do |t|
+  t.column :name, :string
+end

data/features/thinking_sphinx/db/migrations/create_comments.rb

@@ -0,0 +1,10 @@
+ActiveRecord::Base.connection.create_table :comments, :force => true do |t|
+  t.column :name,     :string,  :null => false
+  t.column :email,    :string
+  t.column :url,      :string
+  t.column :content,  :text
+  t.column :post_id,  :integer, :null => false
+  t.column :category_id, :integer, :null => false
+  t.column :created_at, :datetime
+  t.column :updated_at, :datetime
+end

data/features/thinking_sphinx/db/migrations/create_developers.rb

@@ -0,0 +1,7 @@
+ActiveRecord::Base.connection.create_table :developers, :force => true do |t|
+  t.column :name,     :string,  :null => false
+  t.column :city,     :string
+  t.column :state,    :string
+  t.column :country,  :string
+  t.column :age,      :integer
+end

data/features/thinking_sphinx/db/migrations/create_extensible_betas.rb

@@ -0,0 +1,5 @@
+ActiveRecord::Base.connection.create_table :extensible_betas, :force => true do |t|
+  t.column  :name, :string,  :null => false
+  t.column :delta, :boolean, :null => false, :default => false
+  t.column :changed_by_generic, :boolean, :null => false, :default => false
+end

data/features/thinking_sphinx/db/migrations/create_gammas.rb

@@ -0,0 +1,3 @@
+ActiveRecord::Base.connection.create_table :gammas, :force => true do |t|
+  t.column :name, :string, :null => false
+end

data/features/thinking_sphinx/db/migrations/create_genres.rb

@@ -0,0 +1,3 @@
+ActiveRecord::Base.connection.create_table :genres, :force => true do |t|
+  t.column :name, :string
+end

data/features/thinking_sphinx/db/migrations/create_music.rb

@@ -0,0 +1,6 @@
+ActiveRecord::Base.connection.create_table :music, :force => true do |t|
+  t.column :artist,   :string
+  t.column :album,    :string
+  t.column :track,    :string
+  t.column :genre_id, :integer
+end

data/features/thinking_sphinx/db/migrations/create_people.rb

@@ -0,0 +1,13 @@
+ActiveRecord::Base.connection.create_table :people, :force => true do |t|
+  t.column :first_name,     :string
+  t.column :middle_initial, :string
+  t.column :last_name,      :string
+  t.column :gender,         :string
+  t.column :street_address, :string
+  t.column :city,           :string
+  t.column :state,          :string
+  t.column :postcode,       :string
+  t.column :email,          :string
+  t.column :birthday,       :datetime
+  t.column :delta, :boolean, :null => false, :default => false
+end

data/features/thinking_sphinx/db/migrations/create_posts.rb

@@ -0,0 +1,5 @@
+ActiveRecord::Base.connection.create_table :posts, :force => true do |t|
+  t.column :subject, :string,  :null => false
+  t.column :content, :text
+  t.column :category_id, :integer, :null => false
+end

data/features/thinking_sphinx/db/migrations/create_robots.rb

@@ -0,0 +1,4 @@
+ActiveRecord::Base.connection.create_table :robots, :primary_key => :alternate_primary_key, :force => true do |t|
+  t.column :name,                   :string,  :null => false
+  t.column :internal_id,            :string,  :null => false
+end

data/features/thinking_sphinx/db/migrations/create_taggings.rb

@@ -0,0 +1,5 @@
+ActiveRecord::Base.connection.create_table :taggings, :force => true do |t|
+  t.column :tag_id,         :integer
+  t.column :taggable_id,    :integer, :null => false
+  t.column :taggable_type,  :string,  :null => false
+end

data/features/thinking_sphinx/db/migrations/create_tags.rb

@@ -0,0 +1,4 @@
+ActiveRecord::Base.connection.create_table :tags, :force => true do |t|
+  # t.column :post_id, :integer, :null => false
+  t.column :text,       :text,  :null => true
+end

data/features/thinking_sphinx/models/alpha.rb

@@ -0,0 +1,22 @@
+class Alpha < ActiveRecord::Base
+  define_index do
+    indexes :name, :sortable => true
+    
+    has value, created_at, created_on
+    has cost, :facet => true
+    has active
+    
+    set_property :field_weights => {"name" => 10}
+  end
+  
+  define_index 'alternative' do
+    indexes :name, :as => :alternative_name, :sortable => true
+    
+    has value, created_at, created_on
+    has cost, :facet => true
+    
+    set_property :field_weights => {'alternative_name' => 10}
+    
+    where "value > 3"
+  end
+end

data/features/thinking_sphinx/models/animal.rb

@@ -0,0 +1,5 @@
+class Animal < ActiveRecord::Base
+  define_index do
+    indexes name, :sortable => true
+  end
+end

data/features/thinking_sphinx/models/author.rb

@@ -0,0 +1,3 @@
+class Author < ActiveRecord::Base
+  #
+end

data/features/thinking_sphinx/models/beta.rb

@@ -0,0 +1,8 @@
+class Beta < ActiveRecord::Base
+  define_index do
+    indexes :name, :sortable => true, :facet => true
+    has value
+    
+    set_property :delta => true
+  end
+end

data/features/thinking_sphinx/models/box.rb

@@ -0,0 +1,8 @@
+class Box < ActiveRecord::Base
+  define_index do
+    indexes width, :as => :width_field
+    
+    has width, length, depth
+    has [width, length, depth], :as => :dimensions
+  end
+end

data/features/thinking_sphinx/models/cat.rb

@@ -0,0 +1,3 @@
+class Cat < Animal
+  #
+end

data/features/thinking_sphinx/models/category.rb

@@ -0,0 +1,4 @@
+class Category < ActiveRecord::Base
+  has_many :posts
+  has_many :comments
+end

data/features/thinking_sphinx/models/comment.rb

@@ -0,0 +1,10 @@
+class Comment < ActiveRecord::Base
+  define_index do 
+    indexes :content
+    
+    has category.name, :facet => true, :as => :category_name, :type => :string
+  end
+  
+  belongs_to :post
+  belongs_to :category
+end

data/features/thinking_sphinx/models/developer.rb

@@ -0,0 +1,16 @@
+require 'features/thinking_sphinx/models/tag'
+require 'features/thinking_sphinx/models/tagging'
+
+class Developer < ActiveRecord::Base
+  has_many :taggings, :as => :taggable
+  has_many :tags, :through => :taggings
+  
+  define_index do
+    indexes country,                      :facet => true
+    indexes state,                        :facet => true
+    indexes tags.text,  :as => :tags,     :facet => true
+    has age,                              :facet => true
+    has tags(:id),      :as => :tag_ids,  :facet => true
+    facet city
+  end
+end

data/features/thinking_sphinx/models/dog.rb

@@ -0,0 +1,3 @@
+class Dog < Animal
+  #
+end

data/features/thinking_sphinx/models/extensible_beta.rb

@@ -0,0 +1,9 @@
+require 'features/support/lib/generic_delta_handler'
+
+class ExtensibleBeta < ActiveRecord::Base
+  define_index do
+    indexes :name, :sortable => true
+    
+    set_property :delta => "GenericDeltaHandler"
+  end
+end

data/features/thinking_sphinx/models/fox.rb

@@ -0,0 +1,5 @@
+class Fox < Animal
+  define_index do
+    indexes :name
+  end
+end

data/features/thinking_sphinx/models/gamma.rb

@@ -0,0 +1,5 @@
+class Gamma < ActiveRecord::Base
+  define_index do
+    indexes :name, :sortable => true
+  end
+end

data/features/thinking_sphinx/models/genre.rb

@@ -0,0 +1,3 @@
+class Genre < ActiveRecord::Base
+  #
+end

data/features/thinking_sphinx/models/medium.rb

@@ -0,0 +1,5 @@
+class Medium < ActiveRecord::Base
+  self.abstract_class = true
+  
+  belongs_to :genre
+end

data/features/thinking_sphinx/models/music.rb

@@ -0,0 +1,8 @@
+class Music < Medium
+  set_table_name 'music'
+  
+  define_index do
+    indexes artist, track, album
+    indexes genre(:name), :as => :genre
+  end
+end

data/features/thinking_sphinx/models/person.rb

@@ -0,0 +1,23 @@
+class Person < ActiveRecord::Base
+  define_index do
+    indexes first_name, last_name, :sortable => true
+    
+    has [first_name, middle_initial, last_name], :as => :name_sort
+    has birthday
+    has gender, :facet => true
+    
+    set_property :min_infix_len => 1
+    set_property :enable_star => true
+  end
+  
+  sphinx_scope(:with_first_name) { |name|
+    { :conditions => {:first_name => name} }
+  }
+  sphinx_scope(:with_last_name) { |name|
+    { :conditions => {:last_name => name} }
+  }
+  sphinx_scope(:with_id) { |id|
+    { :with => {:sphinx_internal_id => id} }
+  }
+  sphinx_scope(:ids_only) { {:ids_only => true} }
+end

data/features/thinking_sphinx/models/post.rb

@@ -0,0 +1,21 @@
+class Post < ActiveRecord::Base
+  has_many :comments, :dependent => :destroy
+  has_many :taggings, :as => :taggable
+  has_many :tags, :through => :taggings
+  belongs_to :category
+  has_and_belongs_to_many :authors
+  
+  define_index do
+    indexes subject
+    indexes content
+    indexes tags.text, :as => :tags
+    indexes comments.content, :as => :comments
+    indexes authors.name, :as => :authors
+    
+    has comments(:id), :as => :comment_ids, :source => :ranged_query,
+      :facet => true
+    has category.name, :facet => true, :as => :category_name, :type => :string
+    has 'COUNT(DISTINCT comments.id)', :as => :comments_count, :type => :integer
+    has comments.created_at, :as => :comments_created_at
+  end
+end

data/features/thinking_sphinx/models/robot.rb

@@ -0,0 +1,12 @@
+class Robot < ActiveRecord::Base
+  set_primary_key :internal_id
+  set_sphinx_primary_key :alternate_primary_key
+  
+  define_index do
+    indexes :name
+  end
+  
+  def id
+    internal_id
+  end
+end

data/features/thinking_sphinx/models/tag.rb

@@ -0,0 +1,3 @@
+class Tag < ActiveRecord::Base
+  belongs_to :post
+end

data/features/thinking_sphinx/models/tagging.rb

@@ -0,0 +1,4 @@
+class Tagging < ActiveRecord::Base
+  belongs_to :tag
+  belongs_to :taggable, :polymorphic => true
+end

data/lib/cucumber/thinking_sphinx/external_world.rb

@@ -0,0 +1,8 @@
+require 'thinking_sphinx/test'
+
+class Cucumber::ThinkingSphinx::ExternalWorld
+  def initialize(suppress_delta_output = true)
+    ::ThinkingSphinx::Test.init
+    ::ThinkingSphinx::Test.start_with_autostop
+  end
+end

data/lib/cucumber/thinking_sphinx/internal_world.rb

@@ -0,0 +1,126 @@
+require 'cucumber/thinking_sphinx/sql_logger'
+
+module Cucumber
+  module ThinkingSphinx
+    class InternalWorld
+      attr_accessor :temporary_directory, :migrations_directory,
+        :models_directory, :fixtures_directory, :database_file
+      attr_accessor :adapter, :database, :username,
+        :password, :host
+      
+      def initialize
+        @temporary_directory  = "#{Dir.pwd}/tmp"
+        @migrations_directory = "features/thinking_sphinx/db/migrations"
+        @models_directory     = "features/thinking_sphinx/models"
+        @fixtures_directory   = "features/thinking_sphinx/db/fixtures"
+        @database_file        = "features/thinking_sphinx/database.yml"
+        
+        @adapter  = ENV['DATABASE'] || 'mysql'
+        @database = 'thinking_sphinx'
+        @username = 'thinking_sphinx'
+        # @password = 'thinking_sphinx'
+        @host     = 'localhost'
+      end
+      
+      def setup
+        make_temporary_directory
+        
+        configure_cleanup
+        configure_thinking_sphinx
+        configure_active_record
+        
+        prepare_data
+        setup_sphinx
+        
+        self
+      end
+      
+      def configure_database
+        ActiveRecord::Base.establish_connection database_settings
+        self
+      end
+      
+      private
+      
+      def config
+        @config ||= ::ThinkingSphinx::Configuration.instance
+      end
+      
+      def make_temporary_directory
+        FileUtils.mkdir_p temporary_directory
+        Dir["#{temporary_directory}/*"].each do |file|
+          FileUtils.rm_rf file
+        end
+      end
+      
+      def configure_thinking_sphinx
+        config.config_file        = "#{temporary_directory}/sphinx.conf"
+        config.searchd_log_file   = "#{temporary_directory}/searchd.log"
+        config.query_log_file     = "#{temporary_directory}/searchd.query.log"
+        config.pid_file           = "#{temporary_directory}/searchd.pid"
+        config.searchd_file_path  = "#{temporary_directory}/indexes/"
+        
+        ::ThinkingSphinx.suppress_delta_output = true
+      end
+      
+      def configure_cleanup
+        Kernel.at_exit do
+          ::ThinkingSphinx::Configuration.instance.controller.stop
+          sleep(0.5) # Ensure Sphinx has shut down completely
+          ActiveRecord::Base.logger.close
+        end
+      end
+      
+      def yaml_database_settings
+        return {} unless File.exist?(@database_file)
+        
+        YAML.load open(@database_file)
+      end
+      
+      def database_settings
+        {
+          'adapter'  => @adapter,
+          'database' => @database,
+          'username' => @username,
+          'password' => @password,
+          'host'     => @host
+        }.merge yaml_database_settings
+      end
+      
+      def configure_active_record
+        ActiveRecord::Base.logger = Logger.new(
+          open("#{temporary_directory}/active_record.log", "a")
+        )
+        
+        ActiveRecord::Base.connection.class.send(
+          :include, Cucumber::ThinkingSphinx::SqlLogger
+        )
+      end
+      
+      def prepare_data
+        ::ThinkingSphinx.deltas_enabled = false
+        
+        load_files migrations_directory
+        load_files models_directory
+        load_files fixtures_directory
+        
+        ::ThinkingSphinx.deltas_enabled = true
+      end
+      
+      def load_files(path)
+        files = Dir["#{path}/*.rb"].sort!
+        files.each do |file|
+          require file.gsub(/\.rb$/, '')
+        end
+      end
+      
+      def setup_sphinx
+        FileUtils.mkdir_p config.searchd_file_path
+        
+        config.build
+        config.controller.index
+        config.controller.start
+      end
+    end
+  end
+end