eson-dsl 0.7.0

data/test/search/query_test.rb

@@ -0,0 +1,311 @@
+require './test/test_config'
+
+context "Queries" do
+  setup do
+    create_index :index => "default"
+  end
+  
+  context "#term" do
+    query_name "test/search/queries/term"
+
+    setup do
+      example("queries/term")
+    end
+  end
+  
+  context "#term with parameters" do
+    query_name "test/search/queries/term"
+
+    setup do
+      example("queries/term_with_parameters")
+    end
+  end
+
+  context "#wildcard" do
+    query_name "test/search/queries/wildcard"
+
+    setup do
+      example("queries/wildcard")
+    end
+  end
+
+  context "#wildcard short style" do
+    query_name "test/search/queries/wildcard_short"
+
+    setup do
+      example("queries/wildcard_short")
+    end
+  end
+
+  context "#prefix" do
+    query_name "test/search/queries/prefix"
+
+    setup do
+      example("queries/prefix")
+    end
+  end
+
+  context "#prefix short style" do
+    query_name "test/search/queries/prefix_short"
+
+    setup do
+      example("queries/prefix_short")
+    end
+  end
+
+  context "#match_all" do
+    query_name "test/search/queries/match_all"
+
+    setup do
+      example("queries/match_all")
+    end
+  end
+
+  context "#terms" do
+    query_name "test/search/queries/terms"
+
+    setup do
+      example("queries/terms")
+    end
+  end
+
+  context "#ids" do
+    query_name "test/search/queries/ids"
+
+    setup do
+      example("queries/ids")
+    end
+  end
+
+  context '#range query' do
+    query_name "test/search/queries/range"
+
+    setup do
+      example("queries/range")
+    end
+  end
+
+  context "#constant_score query" do
+    query_name "test/search/queries/constant_score"
+
+    setup do
+      example("queries/constant_score")
+    end
+  end
+
+  context "#constant_store query with filter" do
+    query_name "test/search/queries/constant_score_filter"
+
+    setup do
+      example("queries/constant_score_with_filter")
+    end
+  end
+
+  context "#text query" do
+    query_name "test/search/queries/text"
+
+    setup do
+      example("queries/text")
+    end
+  end
+
+  context "#bool query" do
+    query_name "test/search/queries/bool"
+
+    setup do
+      example("queries/bool")
+    end
+  end
+
+
+  context "#boosting query" do
+    query_name "test/search/queries/boosting"
+
+    setup do
+      example("queries/boosting")
+    end
+  end
+
+  context "#custom_score query" do
+    query_name "test/search/queries/custom_score"
+
+    setup do
+      example("queries/custom_score")
+    end
+  end
+
+  context "#dis_max query" do
+    query_name "test/search/queries/dis_max"
+
+    setup do
+      example("queries/dis_max")
+    end
+  end
+
+  context "#field query" do
+    query_name "test/search/queries/field"
+
+    setup do
+      example("queries/field")
+    end
+  end
+
+  context "#flt query" do
+    query_name "test/search/queries/flt"
+
+    setup do
+      example("queries/flt")
+    end
+
+  end
+
+  context "#flt_field query" do
+    query_name "test/search/queries/flt_field"
+
+    setup do
+      example("queries/flt_field")
+    end
+  end
+
+  context "#fuzzy simple query" do
+    query_name "test/search/queries/fuzzy_simple"
+
+    setup do
+      example("queries/fuzzy")
+    end
+  end
+
+  context "#fuzzy complex query" do
+    query_name "test/search/queries/fuzzy_complex"
+
+    setup do
+      example("queries/fuzzy_complex")
+    end
+
+  end
+
+  context "#has_child query" do
+    query_name "test/search/queries/has_child"
+    set :type, "blog_tag"
+    set :index, "has_child_query"
+
+    setup do
+      example("queries/has_child")
+    end
+  end
+
+  context "complex #has_child query" do
+    query_name "test/search/queries/has_child_complex"
+    set :type, "blog_tag"
+    set :index, "has_child_query"
+
+    setup do
+      example("queries/has_child_complex")
+    end
+  end
+
+  context "#mlt query" do
+    query_name "test/search/queries/mlt"
+    set :type, "bar"
+
+    setup do
+      example("queries/mlt")
+    end
+  end
+
+  context "#mlt_field query" do
+    query_name "test/search/queries/mlt_field"
+
+    setup do
+      example("queries/mlt_field")
+    end
+  end
+
+  context "#query_string" do
+    query_name "test/search/queries/query_string"
+
+    setup do
+      example("queries/query_string")
+    end
+  end
+
+  context "#query_string" do
+    query_name "test/search/queries/query_string_without_default_field"
+
+    setup do
+      example("queries/query_string_without_default_field")
+    end
+  end
+
+  context "#query_string with multiple fields" do
+    query_name "test/search/queries/query_string_with_multiple_fields"
+
+    setup do
+      example("queries/query_string_with_multiple_fields")
+    end
+  end
+
+  context "#top_children query" do
+    query_name "test/search/queries/top_children"
+
+    setup do
+      example("queries/top_children")
+    end
+  end
+
+  context "query with filter and facets" do
+    query_name "test/search/queries/filters_and_facets"
+
+    setup do
+      example("queries/filters_and_facets")
+    end
+  end
+
+  context "nested queries" do
+    query_name "test/search/queries/nested"
+
+    setup do
+      example("queries/nested")
+    end
+  end
+
+  context "span_or query" do
+    query_name "test/search/queries/span_or"
+
+    setup do
+      example("queries/span_or")
+    end
+  end
+
+  context "span_near query" do
+    query_name "test/search/queries/span_near"
+
+    setup do
+      example("queries/span_near")
+    end
+  end
+
+  context "span_first query" do
+    query_name "test/search/queries/span_first"
+
+    setup do
+      example("queries/span_first")
+    end
+  end
+
+  context "span_not query" do
+    query_name "test/search/queries/span_not"
+
+    setup do
+      example("queries/span_not")
+    end
+  end
+
+  context "span_term query" do
+    query_name "test/search/queries/span_term"
+
+    setup do
+      example("queries/span_term")
+    end
+  end
+end

data/test/test_config.rb

@@ -0,0 +1,157 @@
+begin
+  # Require the preresolved locked set of gems.
+  require File.expand_path('../../.bundle/environment', __FILE__)
+rescue LoadError
+  # Fallback on doing the resolve at runtime.
+  require 'rubygems'
+  require 'bundler'
+  Bundler.setup
+end
+
+Bundler.require(:test, :default)
+
+require 'eson-dsl'
+require 'faraday'
+require 'elasticsearch-node/external'
+
+module Node
+  module External
+    def self.instance
+      @node ||= begin 
+        node = ElasticSearch::Node::External.new("gateway.type" => "none") do
+          def client
+            conn = Faraday.new(:url => "http://#{self.ip}:#{self.port}") do |builder|
+              #builder.response :raise_error
+              builder.request :json
+              builder.adapter :net_http
+            end
+          end
+        end
+        at_exit do
+          node.close
+        end
+        node
+      end
+    end
+  end
+  
+end
+Node::External.instance
+
+require 'riot'
+
+#Riot.dots
+
+class EsonQueries < Riot::ContextMiddleware
+  register
+
+  def call(context) 
+    middleware.call(context)
+
+    return unless context.option(:query_name) 
+
+    type = context.option(:type) || "bar"
+    index = context.option(:index) || "default"
+    file = context.option(:query_name)
+
+    Dir["#{file}*.documents"].each do |f|
+
+      f.match(/\A[^.]+(?:\.([^.]*))?.documents\Z/)
+      type = $1 if $1
+
+      context.setup(true) do
+        create_index :index => index
+
+        File.open(f) do |p|
+          begin
+            p.each_line do |l|
+              index_doc :index => index,
+                        :source => l,
+                        :type => type
+            end
+          rescue Exception => e
+            puts e.inspect
+            raise e
+          end
+        end
+      end
+
+    end
+
+    Dir["#{file}*.mapping"].each do |f|
+      context.setup(true) do
+        f.match(/\A[^.]+(?:\.([^.]*))?.mapping\Z/)
+        type = $1 if $1
+
+        delete_mapping :index => index, :type => type
+
+        create_index :index => index
+        put_mapping :index => index,
+                    :type => type,
+                    :source => File.read(f)
+      end
+    end
+
+    if File.exists?("#{file}.rb") 
+      context.asserts(:to_query_hash).equals(eval(File.read("#{file}.rb")))
+    end
+
+    if File.exists?("#{file}.json")
+      context.asserts("json format") do
+        MultiJson.decode(MultiJson.encode(topic.to_query_hash))
+      end.equals(MultiJson.decode(File.read("#{file}.json").strip))
+    end
+
+    context.asserts("search status") do
+      begin
+        result = search :index => index, :type => type, :query => topic
+        result.status
+      rescue Exception => e
+        puts e.inspect
+        raise e
+      end
+    end.equals(200)
+
+  end
+end
+
+class Riot::Context
+  def query_name(name)
+    set :query_name, name 
+  end
+end
+
+class Riot::Situation
+  def node
+    Node::External.instance
+  end
+
+  def create_index(opts)
+    node.client.post "/#{opts[:index]}"
+  end
+
+  def index_doc(opts)
+    node.client.post "/#{opts[:index]}/#{opts[:type]}", opts[:source]
+  end
+
+  def delete_mapping(opts)
+    node.client.delete "/#{opts[:index]}/#{opts[:type]}"
+  end
+
+  def put_mapping(opts)
+    node.client.put "/#{opts[:index]}/#{opts[:type]}/_mapping", opts[:source]
+  end
+
+  def search(opts)
+    source = opts[:query].to_query_hash
+    if opts[:type]
+      node.client.post "/#{opts[:index]}/#{opts[:type]}/_search", source
+    else
+      node.client.post "/#{opts[:index]}/_search", source
+    end
+  end
+  
+  def example(file)
+    eval(File.read(File.expand_path(File.join('examples', "#{file}.rb"))))
+  end
+end