tire 0.1.0

data/test/integration/index_mapping_test.rb

@@ -0,0 +1,44 @@
+require 'test_helper'
+
+module Tire
+
+  class IndexMappingIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Default mapping" do
+      teardown { Tire.index('mapped-index').delete }
+
+      should "create and return the default mapping" do
+
+        index = Tire.index 'mapped-index' do
+          create
+          store :article, :title => 'One'
+          refresh
+        end
+        sleep 1.5
+
+        assert_equal 'string', index.mapping['article']['properties']['title']['type'],  index.mapping.inspect
+        assert_nil             index.mapping['article']['properties']['title']['boost'], index.mapping.inspect
+      end
+    end
+
+    context "Creating index with mapping" do
+      teardown { Tire.index('mapped-index').delete; sleep 1 }
+    
+      should "create the specified mapping" do
+    
+        index = Tire.index 'mapped-index' do
+          create :mappings => { :article => { :properties => { :title => { :type => 'string', :boost => 2.0, :store => 'yes' } } } }
+        end
+        sleep 1
+
+        # p index.mapping    
+        assert_equal 2.0,   index.mapping['article']['properties']['title']['boost'],  index.mapping.inspect
+        assert_equal 'yes', index.mapping['article']['properties']['title']['store'],  index.mapping.inspect
+    
+      end
+    end
+
+  end
+
+end

data/test/integration/index_store_test.rb

@@ -0,0 +1,68 @@
+require 'test_helper'
+
+module Tire
+
+  class IndexStoreIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Storing the documents in index" do
+
+      teardown { Tire.index('articles-test-ids').delete }
+
+      should "store hashes under their IDs" do
+
+        Tire.index 'articles-test-ids' do
+          delete
+          create
+
+          store :id => 1, :title => 'One'
+          store :id => 2, :title => 'Two'
+          store :id => 3, :title => 'Three'
+          store :id => 4, :title => 'Four'
+          store :id => 4, :title => 'Four'
+
+          refresh
+        end
+
+        s = Tire.search('articles-test-ids') { query { string '*' } }
+
+        assert_equal 4, s.results.count
+
+        document = Tire.index('articles-test-ids').retrieve :document, 4
+        assert_equal 'Four', document.title
+        assert_equal 2,      document._version.to_i
+
+      end
+
+    end
+
+    context "Removing documents from the index" do
+
+      teardown { Tire.index('articles-test-remove').delete }
+
+      setup do
+        Tire.index 'articles-test-remove' do
+          delete
+          create
+          store :id => 1, :title => 'One'
+          store :id => 2, :title => 'Two'
+          refresh
+        end
+      end
+
+      should "remove document from the index" do
+
+        assert_equal 2, Tire.search('articles-test-remove') { query { string '*' } }.results.count
+
+        assert_nothing_raised do
+          assert Tire.index('articles-test-remove').remove 1
+          assert ! Tire.index('articles-test-remove').remove(1)
+        end
+
+      end
+
+    end
+
+  end
+
+end

data/test/integration/persistent_model_test.rb

@@ -0,0 +1,35 @@
+require 'test_helper'
+
+module Tire
+
+  class PersistentModelIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    def setup
+      super
+      PersistentArticle.index.delete
+    end
+
+    def teardown
+      super
+      PersistentArticle.index.delete
+    end
+
+    context "PersistentModel" do
+
+      should "save documents into index and find them by IDs" do
+        one = PersistentArticle.create :id => 1, :title => 'One'
+        two = PersistentArticle.create :id => 2, :title => 'Two'
+
+        PersistentArticle.index.refresh
+        sleep(1.5)
+
+        results = PersistentArticle.find [1, 2]
+
+        assert_equal 2, results.size
+        
+      end
+    end
+
+  end
+end

data/test/integration/query_string_test.rb

@@ -0,0 +1,43 @@
+require 'test_helper'
+
+module Tire
+
+  class QueryStringIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Searching for query string" do
+
+      should "find article by title" do
+        q = 'title:one'
+        assert_equal 1, search(q).results.count
+        assert_equal 'One', search(q).results.first[:title]
+      end
+
+      should "find articles by title with boosting" do
+        q = 'title:one^100 OR title:two'
+        assert_equal 2, search(q).results.count
+        assert_equal 'One', search(q).results.first[:title]
+      end
+
+      should "find articles by tags" do
+        q = 'tags:ruby AND tags:python'
+        assert_equal 1, search(q).results.count
+        assert_equal 'Two', search(q).results.first[:title]
+      end
+
+      should "find any article with tags" do
+        q = 'tags:ruby OR tags:python OR tags:java'
+        assert_equal 4, search(q).results.count
+      end
+
+    end
+
+    private
+
+    def search(q)
+      Tire.search('articles-test') { query { string q } }
+    end
+
+  end
+
+end

data/test/integration/results_test.rb

@@ -0,0 +1,28 @@
+require 'test_helper'
+
+module Tire
+
+  class ResultsIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Query results" do
+
+      should "allow easy access to returned documents" do
+        q = 'title:one'
+        s = Tire.search('articles-test') { query { string q } }
+        assert_equal 'One',  s.results.first.title
+        assert_equal 'ruby', s.results.first.tags[0]
+      end
+
+      should "allow easy access to returned documents with limited fields" do
+        q = 'title:one'
+        s = Tire.search('articles-test') { query { string q }.fields :title }
+        assert_equal 'One', s.results.first.title
+        assert_nil s.results.first.tags
+      end
+
+    end
+
+  end
+
+end

data/test/integration/sort_test.rb

@@ -0,0 +1,36 @@
+require 'test_helper'
+
+module Tire
+
+  class SortIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Sort" do
+
+      should "sort by title" do
+        q = '*'
+        s = Tire.search('articles-test') do
+          query { string q }
+          sort  { title }
+        end
+
+        assert_equal 5, s.results.count
+        assert_equal 'Five', s.results.first[:title]
+      end
+
+      should "sort by title, descending" do
+        q = '*'
+        s = Tire.search('articles-test') do
+          query { string q }
+          sort  { title :desc }
+        end
+
+        assert_equal 5, s.results.count
+        assert_equal 'Two', s.results.first[:title]
+      end
+
+    end
+
+  end
+
+end

data/test/models/active_model_article.rb

@@ -0,0 +1,31 @@
+# Example ActiveModel class
+
+require 'rubygems'
+require 'active_model'
+
+class ActiveModelArticle
+
+  extend  ActiveModel::Naming
+  include ActiveModel::AttributeMethods
+  include ActiveModel::Serialization
+  include ActiveModel::Serializers::JSON
+
+  include Tire::Model::Search
+
+  attr_reader :attributes
+
+  def initialize(attributes = {})
+    @attributes = attributes
+  end
+
+  def id;         attributes['id'] || attributes['_id']; end
+  def id=(value); attributes['id'] = value;              end
+
+  def method_missing(name, *args, &block)
+    attributes[name.to_sym] || attributes[name.to_s] || super
+  end
+
+  def persisted?; true; end
+  def save;       true; end
+
+end

data/test/models/active_model_article_with_callbacks.rb

@@ -0,0 +1,49 @@
+# Example ActiveModel class with callbacks
+
+require 'rubygems'
+require 'active_model'
+
+class ActiveModelArticleWithCallbacks
+
+  include ActiveModel::AttributeMethods
+  include ActiveModel::Validations
+  include ActiveModel::Serialization
+  include ActiveModel::Serializers::JSON
+  include ActiveModel::Naming
+
+  extend  ActiveModel::Callbacks
+  define_model_callbacks :save, :destroy
+
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+
+  attr_reader :attributes
+
+  def initialize(attributes = {})
+    @attributes = attributes
+  end
+
+  def method_missing(id, *args, &block)
+    attributes[id.to_sym] || attributes[id.to_s] || super
+  end
+
+  def persisted?
+    true
+  end
+
+  def save
+    _run_save_callbacks do
+      STDERR.puts "[Saving ...]"
+    end
+  end
+
+  def destroy
+    _run_destroy_callbacks do
+      STDERR.puts "[Destroying ...]"
+      @destroyed = true
+    end
+  end
+
+  def destroyed?; !!@destroyed; end
+
+end

data/test/models/active_model_article_with_custom_index_name.rb

@@ -0,0 +1,5 @@
+# Example ActiveModel class with custom index name
+
+class ActiveModelArticleWithCustomIndexName < ActiveModelArticle
+  index_name 'custom-index-name'
+end

data/test/models/active_record_article.rb

@@ -0,0 +1,12 @@
+require 'rubygems'
+require 'active_record'
+
+class ActiveRecordArticle < ActiveRecord::Base
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+
+  mapping do
+    indexes :title,      :type => 'string', :boost => 10, :analyzer => 'snowball'
+    indexes :created_at, :type => 'date'
+  end
+end

data/test/models/article.rb

@@ -0,0 +1,15 @@
+# Example non-ActiveModel custom wrapper for result item
+
+class Article
+  attr_reader :id, :title, :body
+
+  def initialize(attributes={})
+    attributes.each { |k,v| instance_variable_set(:"@#{k}", v) }
+  end
+
+  def to_json
+    { :id => @id, :title => @title, :body => @body }.to_json
+  end
+
+  alias :to_indexed_json :to_json
+end

data/test/models/persistent_article.rb

@@ -0,0 +1,11 @@
+# Example class with ElasticSearch persistence
+
+class PersistentArticle
+
+  include Tire::Model::Persistence
+
+  property :title
+  property :published_on
+  property :tags
+
+end

data/test/models/persistent_articles_with_custom_index_name.rb

@@ -0,0 +1,10 @@
+# Example class with ElasticSearch persistence and custom index name
+
+class PersistentArticleWithCustomIndexName
+
+  include Tire::Model::Persistence
+
+  property :title
+
+  index_name 'custom-index-name'
+end

data/test/models/supermodel_article.rb

@@ -0,0 +1,22 @@
+# Example ActiveModel class for testing :searchable mode
+
+require 'rubygems'
+require 'supermodel'
+
+class SupermodelArticle < SuperModel::Base
+  include SuperModel::RandomID
+
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+
+  mapping do
+    indexes :title,      :type => 'string', :boost => 15, :analyzer => 'czech'
+  end
+
+  alias :persisted? :exists?
+
+  def destroyed?
+    !self.class.find(self.id) rescue true
+  end
+
+end

data/test/models/validated_model.rb

@@ -0,0 +1,11 @@
+# Example ActiveModel with validations
+
+class ValidatedModel
+
+  include Tire::Model::Persistence
+
+  property :name
+
+  validates_presence_of :name
+
+end

data/test/test_helper.rb

@@ -0,0 +1,52 @@
+require 'rubygems'
+require 'test/unit'
+require 'shoulda'
+require 'mocha'
+require 'turn' unless ENV["TM_FILEPATH"]
+require 'pathname'
+
+require 'tire'
+
+Dir[File.dirname(__FILE__) + '/models/**/*.rb'].each { |m| require m }
+
+class Test::Unit::TestCase
+
+  def mock_response(body, code=200)
+    stub(:body => body, :code => code)
+  end
+
+  def fixtures_path
+    Pathname( File.expand_path( 'fixtures', File.dirname(__FILE__) ) )
+  end
+
+  def fixture_file(path)
+    File.read File.expand_path( path, fixtures_path )
+  end
+
+end
+
+module Test::Integration
+  URL = "http://localhost:9200"
+
+  def setup
+    begin
+      ::RestClient.get URL
+    rescue Errno::ECONNREFUSED
+      abort "\n\n#{'-'*87}\n[ABORTED] You have to run ElasticSearch on #{URL} for integration tests\n#{'-'*87}\n\n"
+    end
+
+    ::RestClient.delete "#{URL}/articles-test"     rescue nil
+    ::RestClient.post   "#{URL}/articles-test", ''
+    fixtures_path.join('articles').entries.each do |f|
+      filename = f.to_s
+      next if filename =~ /^\./
+      ::RestClient.put "#{URL}/articles-test/article/#{File.basename(filename, '.*')}",
+                       fixtures_path.join('articles').join(f).read
+    end
+    ::RestClient.post "#{URL}/articles-test/_refresh", ''
+  end
+
+  def teardown
+    ::RestClient.delete "#{URL}/articles-test"  rescue nil
+  end
+end